from prog import *

inp = get_input(sample = True)

size = 5

result_1 = get_result(inp, sample = True)
expected_1 = 127

print(f"{result_1 = } {result_1 == expected_1}")

result_2 = get_result(inp, part = 2, sample = True)
expected_2 = 62
print(f"{result_2 = } {result_2 == expected_2}")