from solution import solution test_cases = { (0,): 0, (1,): 0, (2,): 3, (3,): 0, (4,): 11, (6,): 3*3*3 + 3*2*2 + 2, } for i, o in test_cases.items(): print('input', *i, '\toutput', o, '\tsolution', solution(*i))