from prog import * inp_1 = get_input(sample = True, number = 1) result_1_1 = get_result(inp_1) expected_1_1 = 35 print(f"{result_1_1 = } {result_1_1 == expected_1_1}") inp_2 = get_input(sample = True, number = 2) result_1_2 = get_result(inp_2) expected_1_2 = 220 print(f"{result_1_2 = } {result_1_2 == expected_1_2}")