test_day18.py

from aoc.day18 import parse_input, part1, part2

TEST_INPUT = """5,4
4,2
4,5
3,0
2,1
6,3
2,4
1,5
0,6
3,3
2,6
5,1
1,2
5,5
2,5
6,5
1,4
0,4
6,4
1,1
6,1
1,0
0,5
1,6
2,0
""".splitlines()


def test_part_1():
    assert part1(parse_input(TEST_INPUT[:12]), 7, 7) == 22


def test_part_2():
    assert part2(TEST_INPUT, 7, 7) == "6,1"