-
Notifications
You must be signed in to change notification settings - Fork 1
/
test_aoc202310.py
70 lines (49 loc) · 1.66 KB
/
test_aoc202310.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
"""Tests for AoC 10, 2023: Pipe Maze."""
# Standard library imports
import pathlib
# Third party imports
import aoc202310
import pytest
PUZZLE_DIR = pathlib.Path(__file__).parent
@pytest.fixture
def example1():
puzzle_input = (PUZZLE_DIR / "example1.txt").read_text().rstrip()
return aoc202310.parse_data(puzzle_input)
@pytest.fixture
def example2():
puzzle_input = (PUZZLE_DIR / "example2.txt").read_text().rstrip()
return aoc202310.parse_data(puzzle_input)
@pytest.fixture
def example3():
puzzle_input = (PUZZLE_DIR / "example3.txt").read_text().rstrip()
return aoc202310.parse_data(puzzle_input)
@pytest.fixture
def example4():
puzzle_input = (PUZZLE_DIR / "example4.txt").read_text().rstrip()
return aoc202310.parse_data(puzzle_input)
def test_parse_example1(example1):
"""Test that input is parsed properly."""
start, grid = example1
assert start == (1, 1)
assert grid == {
(1, 1): {(1, 2), (2, 1)},
(1, 2): {(1, 1), (1, 3)},
(1, 3): {(1, 2), (2, 3)},
(2, 1): {(1, 1), (3, 1)},
(2, 3): {(1, 3), (3, 3)},
(3, 1): {(2, 1), (3, 2)},
(3, 2): {(3, 1), (3, 3)},
(3, 3): {(3, 2), (2, 3)},
}
def test_part1_example1(example1):
"""Test part 1 on example input."""
assert aoc202310.part1(example1) == 4
def test_part1_example2(example2):
"""Test part 1 on example input."""
assert aoc202310.part1(example2) == 8
def test_part2_example3(example3):
"""Test part 2 on example input."""
assert aoc202310.part2(example3) == 8
def test_part2_example4(example4):
"""Test part 2 on example input."""
assert aoc202310.part2(example4) == 10