56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from solutions.day_14 import Solution
|
||
|
|
|
||
|
|
|
||
|
|
class Day14TestCase(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.solution = Solution()
|
||
|
|
|
||
|
|
def test_parse_puzzle_input(self):
|
||
|
|
data = """
|
||
|
|
mask = XXXXXXXXXXXXXXXXXXXXXXXXXXXXX1XXXX0X
|
||
|
|
mem[8] = 11
|
||
|
|
mem[7] = 101
|
||
|
|
mem[8] = 0
|
||
|
|
""".strip()
|
||
|
|
expected = [
|
||
|
|
("XXXXXXXXXXXXXXXXXXXXXXXXXXXXX1XXXX0X", [(8, 11), (7, 101), (8, 0)])
|
||
|
|
]
|
||
|
|
|
||
|
|
assert len(self.solution.parse_input(data)) == 1
|
||
|
|
assert (
|
||
|
|
self.solution.parse_input(data)[0][0]
|
||
|
|
== "XXXXXXXXXXXXXXXXXXXXXXXXXXXXX1XXXX0X"
|
||
|
|
)
|
||
|
|
assert len(self.solution.parse_input(data)[0][1]) == 3
|
||
|
|
assert (8, 11) in self.solution.parse_input(data)[0][1]
|
||
|
|
assert (7, 101) in self.solution.parse_input(data)[0][1]
|
||
|
|
assert (8, 0) in self.solution.parse_input(data)[0][1]
|
||
|
|
|
||
|
|
def test_solve_first_part(self):
|
||
|
|
data = """
|
||
|
|
mask = XXXXXXXXXXXXXXXXXXXXXXXXXXXXX1XXXX0X
|
||
|
|
mem[8] = 11
|
||
|
|
mem[7] = 101
|
||
|
|
mem[8] = 0
|
||
|
|
""".strip()
|
||
|
|
|
||
|
|
puzzle_input = self.solution.parse_input(data)
|
||
|
|
assert self.solution.solve(puzzle_input) == 165
|
||
|
|
|
||
|
|
def test_solve_second_part(self):
|
||
|
|
data = """
|
||
|
|
mask = 000000000000000000000000000000X1001X
|
||
|
|
mem[42] = 100
|
||
|
|
mask = 00000000000000000000000000000000X0XX
|
||
|
|
mem[26] = 1
|
||
|
|
""".strip()
|
||
|
|
|
||
|
|
puzzle_input = self.solution.parse_input(data)
|
||
|
|
assert self.solution.solve_again(puzzle_input) == 208
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|