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