def test_example3_part2():
answer, lines = get_puzzle_tests(3, 3)
result = part_two(lines)
expected_answer = int(answer[1])
assert result == expected_answer
def test_example2():
answer, lines = get_puzzle_tests(6, 2)
assert int(answer[0]) == part_two(lines)
def test_example3():
answer, lines = get_puzzle_tests(3, 3)
result = part_one(lines)
expected_answer = int(answer[0])
assert result == expected_answer
def test_example1():
answer, lines = get_puzzle_tests(6, 1)
assert int(answer[0]) == part_one(lines)