77 lines
1.3 KiB
Python
77 lines
1.3 KiB
Python
import unittest
|
|
|
|
from solutions.day_10 import Solution
|
|
|
|
|
|
class Day10TestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.solution = Solution()
|
|
self.data_1 = self.puzzle_input = self.solution.parse_input(
|
|
"""
|
|
16
|
|
10
|
|
15
|
|
5
|
|
1
|
|
11
|
|
7
|
|
19
|
|
6
|
|
12
|
|
4
|
|
""".strip()
|
|
)
|
|
self.data_2 = self.puzzle_input = self.solution.parse_input(
|
|
"""
|
|
28
|
|
33
|
|
18
|
|
42
|
|
31
|
|
14
|
|
46
|
|
20
|
|
48
|
|
47
|
|
24
|
|
23
|
|
49
|
|
45
|
|
19
|
|
38
|
|
39
|
|
11
|
|
1
|
|
32
|
|
25
|
|
35
|
|
8
|
|
17
|
|
7
|
|
9
|
|
4
|
|
2
|
|
34
|
|
10
|
|
3
|
|
""".strip()
|
|
)
|
|
|
|
def test_parse_puzzle_input(self):
|
|
data = """
|
|
16
|
|
1
|
|
15
|
|
""".strip()
|
|
assert self.solution.parse_input(data) == [16, 1, 15]
|
|
|
|
def test_solve_first_part(self):
|
|
assert self.solution.solve(self.data_1) == 7 * 5
|
|
assert self.solution.solve(self.data_2) == 22 * 10
|
|
|
|
def test_solve_second_part(self):
|
|
assert self.solution.solve_again(self.data_1) == 8
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|