144 lines
3.5 KiB
Python
144 lines
3.5 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from solutions.day_11 import Solution
|
||
|
|
|
||
|
|
|
||
|
|
class Day11TestCase(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.solution = Solution()
|
||
|
|
self.puzzle_input = self.solution.parse_input(
|
||
|
|
"""
|
||
|
|
L.LL.LL.LL
|
||
|
|
LLLLLLL.LL
|
||
|
|
L.L.L..L..
|
||
|
|
LLLL.LL.LL
|
||
|
|
L.LL.LL.LL
|
||
|
|
L.LLLLL.LL
|
||
|
|
..L.L.....
|
||
|
|
LLLLLLLLLL
|
||
|
|
L.LLLLLL.L
|
||
|
|
L.LLLLL.LL
|
||
|
|
""".strip()
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_parse_puzzle_input(self):
|
||
|
|
data = """
|
||
|
|
L.LL.LL.LL
|
||
|
|
LLLLLLL.LL
|
||
|
|
""".strip()
|
||
|
|
expected = [
|
||
|
|
["L", ".", "L", "L", ".", "L", "L", ".", "L", "L"],
|
||
|
|
["L", "L", "L", "L", "L", "L", "L", ".", "L", "L"],
|
||
|
|
]
|
||
|
|
|
||
|
|
assert self.solution.parse_input(data) == expected
|
||
|
|
|
||
|
|
def test_adjacent(self):
|
||
|
|
data = [[1, 2, 3, 4], [11, 12, 13, 14], [21, 22, 23, 24], [31, 32, 33, 34]]
|
||
|
|
|
||
|
|
assert self.solution.get_adjacent(data, 1, 1) == [
|
||
|
|
((-1, -1), 1),
|
||
|
|
((0, -1), 2),
|
||
|
|
((1, -1), 3),
|
||
|
|
((-1, 0), 11),
|
||
|
|
((1, 0), 13),
|
||
|
|
((-1, 1), 21),
|
||
|
|
((0, 1), 22),
|
||
|
|
((1, 1), 23),
|
||
|
|
]
|
||
|
|
assert self.solution.get_adjacent(data, 0, 0) == [
|
||
|
|
((1, 0), 2),
|
||
|
|
((0, 1), 11),
|
||
|
|
((1, 1), 12),
|
||
|
|
]
|
||
|
|
assert self.solution.get_adjacent(data, 3, 3) == [
|
||
|
|
((-1, -1), 23),
|
||
|
|
((0, -1), 24),
|
||
|
|
((-1, 0), 33),
|
||
|
|
]
|
||
|
|
assert self.solution.get_adjacent(data, 3, 0) == [
|
||
|
|
((0, -1), 21),
|
||
|
|
((1, -1), 22),
|
||
|
|
((1, 0), 32),
|
||
|
|
]
|
||
|
|
assert self.solution.get_adjacent(data, 0, 3) == [
|
||
|
|
((-1, 0), 3),
|
||
|
|
((-1, 1), 13),
|
||
|
|
((0, 1), 14),
|
||
|
|
]
|
||
|
|
|
||
|
|
def test_occupy_if_empty(self):
|
||
|
|
data = ["L", "L", "L", "L", "L", "L"]
|
||
|
|
data_no = ["#", "L", "L", "L", "L", "L"]
|
||
|
|
|
||
|
|
assert self.solution.occupy_if_empty("L", data) == "#"
|
||
|
|
assert self.solution.occupy_if_empty("#", data) == "#"
|
||
|
|
assert self.solution.occupy_if_empty("L", data_no) == "L"
|
||
|
|
|
||
|
|
def test_empty_if_occupied(self):
|
||
|
|
data = ["L", "L", "L", "L", "L", "L"]
|
||
|
|
data_no = ["L", "L", "#", "#", "#", "#"]
|
||
|
|
data_no2 = ["L", "#", "#", "#", "#", "#"]
|
||
|
|
|
||
|
|
assert self.solution.empty_if_occupied("L", data) == "L"
|
||
|
|
assert self.solution.empty_if_occupied("#", data_no) == "L"
|
||
|
|
assert self.solution.empty_if_occupied("#", data_no2) == "L"
|
||
|
|
assert self.solution.empty_if_occupied("#", data) == "#"
|
||
|
|
|
||
|
|
def test_tick(self):
|
||
|
|
data_1 = self.solution.parse_input(
|
||
|
|
"""
|
||
|
|
L.LL.LL.LL
|
||
|
|
LLLLLLL.LL
|
||
|
|
L.L.L..L..
|
||
|
|
LLLL.LL.LL
|
||
|
|
L.LL.LL.LL
|
||
|
|
L.LLLLL.LL
|
||
|
|
..L.L.....
|
||
|
|
LLLLLLLLLL
|
||
|
|
L.LLLLLL.L
|
||
|
|
L.LLLLL.LL
|
||
|
|
""".strip()
|
||
|
|
)
|
||
|
|
data_2 = self.solution.parse_input(
|
||
|
|
"""
|
||
|
|
#.##.##.##
|
||
|
|
#######.##
|
||
|
|
#.#.#..#..
|
||
|
|
####.##.##
|
||
|
|
#.##.##.##
|
||
|
|
#.#####.##
|
||
|
|
..#.#.....
|
||
|
|
##########
|
||
|
|
#.######.#
|
||
|
|
#.#####.##
|
||
|
|
""".strip()
|
||
|
|
)
|
||
|
|
data_3 = self.solution.parse_input(
|
||
|
|
"""
|
||
|
|
#.LL.LL.L#
|
||
|
|
#LLLLLL.LL
|
||
|
|
L.L.L..L..
|
||
|
|
LLLL.LL.LL
|
||
|
|
L.LL.LL.LL
|
||
|
|
L.LLLLL.LL
|
||
|
|
..L.L.....
|
||
|
|
LLLLLLLLL#
|
||
|
|
#.LLLLLL.L
|
||
|
|
#.LLLLL.L#
|
||
|
|
""".strip()
|
||
|
|
)
|
||
|
|
|
||
|
|
assert self.solution.tick(data_1, 10, 10, tolerance=5, in_view=True) == data_2
|
||
|
|
assert self.solution.tick(data_2, 10, 10, tolerance=5, in_view=True) == data_3
|
||
|
|
|
||
|
|
def test_solve_first_part(self):
|
||
|
|
assert self.solution.solve(self.puzzle_input) == 37
|
||
|
|
|
||
|
|
def test_solve_second_part(self):
|
||
|
|
assert self.solution.solve_again(self.puzzle_input) == 26
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|