from solution import solution test_cases = [ [[2], 2], [[3], 5], [[4], 14], ] for i, o in test_cases: print('input', *i, 'output', o) print('solution', solution(*i)) print()