def test_example_two(self):
example_input = """
#######
#G..#E#
#E#E.E#
#G.##.#
#...#E#
#...E.#
#######
"""
s = Solver(from_str=example_input)
one, units, grid_str = s.solve_part_one()
round_str = """
#######
#...#E#
#E#...#
#.E##.#
#E..#E#
#.....#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((5, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((1, 2), units)["HP"], 197)
self.assertEqual(Solver.unit_by_loc((2, 3), units)["HP"], 185)
self.assertEqual(Solver.unit_by_loc((1, 4), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 200)
self.assertEqual(one, 36334)
def test_example_six(self):
example_input = """
#########
#G......#
#.E.#...#
#..##..G#
#...##..#
#...#...#
#.G...G.#
#.....G.#
#########
"""
s = Solver(from_str=example_input)
one, units, grid_str = s.solve_part_one()
self.assertEqual(one, 18740)
round_str = """
#########
#.G.....#
#G.G#...#
#.G##...#
#...##..#
#.G.#...#
#.......#
#.......#
#########
""".strip().replace(" ", "")
self.assertEqual(Solver.unit_by_loc((2, 1), units)["HP"], 137)
self.assertEqual(Solver.unit_by_loc((1, 2), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((3, 2), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((2, 3), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((2, 5), units)["HP"], 200)
self.assertEqual(grid_str, round_str)
two, units, grid_str = s.solve_part_two()
self.assertEqual(two, 1140)
round_str = """
#########
#.......#
#.E.#...#
#..##...#
#...##..#
#...#...#
#.......#
#.......#
#########
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((2, 2), units)["HP"], 38)
def test_example_four(self):
example_input = """
#######
#E.G#.#
#.#G..#
#G.#.G#
#G..#.#
#...E.#
#######
"""
s = Solver(from_str=example_input)
one, units, grid_str = s.solve_part_one()
self.assertEqual(one, 27755)
round_str = """
#######
#G.G#.#
#.#G..#
#..#..#
#...#G#
#...G.#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((1, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((3, 1), units)["HP"], 98)
self.assertEqual(Solver.unit_by_loc((3, 2), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 95)
self.assertEqual(Solver.unit_by_loc((4, 5), units)["HP"], 200)
two, units, grid_str = s.solve_part_two()
self.assertEqual(two, 3478)
round_str = """
#######
#.E.#.#
#.#E..#
#..#..#
#...#.#
#.....#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((2, 1), units)["HP"], 8)
self.assertEqual(Solver.unit_by_loc((3, 2), units)["HP"], 86)
def test_example_five(self):
example_input = """
#######
#.E...#
#.#..G#
#.###.#
#E#G#G#
#...#G#
#######
"""
s = Solver(from_str=example_input)
one, units, grid_str = s.solve_part_one()
self.assertEqual(one, 28944)
round_str = """
#######
#.....#
#.#G..#
#.###.#
#.#.#.#
#G.G#G#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((3, 2), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((1, 5), units)["HP"], 98)
self.assertEqual(Solver.unit_by_loc((3, 5), units)["HP"], 38)
self.assertEqual(Solver.unit_by_loc((5, 5), units)["HP"], 200)
two, units, grid_str = s.solve_part_two()
self.assertEqual(two, 6474)
round_str = """
#######
#...E.#
#.#..E#
#.###.#
#.#.#.#
#...#.#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((4, 1), units)["HP"], 14)
self.assertEqual(Solver.unit_by_loc((5, 2), units)["HP"], 152)
def test_example_one(self):
example_input = """
#######
#.G...#
#...EG#
#.#.#G#
#..G#E#
#.....#
#######
"""
s = Solver(from_str=example_input)
one, units, grid_str = s.solve_part_one()
self.assertEqual(one, 27730)
round_str = """
#######
#G....#
#.G...#
#.#.#G#
#...#.#
#....G#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((1, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((2, 2), units)["HP"], 131)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 59)
self.assertEqual(Solver.unit_by_loc((5, 5), units)["HP"], 200)
two, units, grid_str = s.solve_part_two()
round_str = """
#######
#..E..#
#...E.#
#.#.#.#
#...#.#
#.....#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((3, 1), units)["HP"], 158)
self.assertEqual(Solver.unit_by_loc((4, 2), units)["HP"], 14)
self.assertEqual(two, 4988)
def test_example_three(self):
example_input = """
#######
#E..EG#
#.#G.E#
#E.##E#
#G..#.#
#..E#.#
#######
"""
s = Solver(from_str=example_input)
one, units, grid_str = s.solve_part_one()
self.assertEqual(one, 39514)
round_str = """
#######
#.E.E.#
#.#E..#
#E.##.#
#.E.#.#
#...#.#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((2, 1), units)["HP"], 164)
self.assertEqual(Solver.unit_by_loc((4, 1), units)["HP"], 197)
self.assertEqual(Solver.unit_by_loc((3, 2), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((1, 3), units)["HP"], 98)
self.assertEqual(Solver.unit_by_loc((2, 4), units)["HP"], 200)
two, units, grid_str = s.solve_part_two()
self.assertEqual(two, 31284)
round_str = """
#######
#.E.E.#
#.#E..#
#E.##E#
#.E.#.#
#...#.#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((2, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((4, 1), units)["HP"], 23)
self.assertEqual(Solver.unit_by_loc((3, 2), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((1, 3), units)["HP"], 125)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((2, 4), units)["HP"], 200)
def test_sample_combat(self):
example_input = """
#######
#.G...#
#...EG#
#.#.#G#
#..G#E#
#.....#
#######
"""
s = Solver(from_str=example_input)
one, units, grid_str = s.solve_part_one(last_round=1)
round_str = """
#######
#..G..#
#...EG#
#.#G#G#
#...#E#
#.....#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((3, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((4, 2), units)["HP"], 197)
self.assertEqual(Solver.unit_by_loc((5, 2), units)["HP"], 197)
self.assertEqual(Solver.unit_by_loc((3, 3), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 197)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 197)
one, units, grid_str = s.solve_part_one(last_round=2)
round_str = """
#######
#...G.#
#..GEG#
#.#.#G#
#...#E#
#.....#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((4, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((3, 2), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((4, 2), units)["HP"], 188)
self.assertEqual(Solver.unit_by_loc((5, 2), units)["HP"], 194)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 194)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 194)
one, units, grid_str = s.solve_part_one(last_round=23)
round_str = """
#######
#...G.#
#..G.G#
#.#.#G#
#...#E#
#.....#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((4, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((3, 2), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((5, 2), units)["HP"], 131)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 131)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 131)
one, units, grid_str = s.solve_part_one(last_round=24)
round_str = """
#######
#..G..#
#...G.#
#.#G#G#
#...#E#
#.....#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((3, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((4, 2), units)["HP"], 131)
self.assertEqual(Solver.unit_by_loc((3, 3), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 128)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 128)
one, units, grid_str = s.solve_part_one(last_round=25)
round_str = """
#######
#.G...#
#..G..#
#.#.#G#
#..G#E#
#.....#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((2, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((3, 2), units)["HP"], 131)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 125)
self.assertEqual(Solver.unit_by_loc((3, 4), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 125)
one, units, grid_str = s.solve_part_one(last_round=26)
round_str = """
#######
#G....#
#.G...#
#.#.#G#
#...#E#
#..G..#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((1, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((2, 2), units)["HP"], 131)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 122)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 122)
self.assertEqual(Solver.unit_by_loc((3, 5), units)["HP"], 200)
one, units, grid_str = s.solve_part_one(last_round=27)
round_str = """
#######
#G....#
#.G...#
#.#.#G#
#...#E#
#...G.#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((1, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((2, 2), units)["HP"], 131)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 119)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 119)
self.assertEqual(Solver.unit_by_loc((4, 5), units)["HP"], 200)
one, units, grid_str = s.solve_part_one(last_round=28)
round_str = """
#######
#G....#
#.G...#
#.#.#G#
#...#E#
#....G#
#######
""".strip().replace(" ", "")
self.assertEqual(grid_str, round_str)
self.assertEqual(Solver.unit_by_loc((1, 1), units)["HP"], 200)
self.assertEqual(Solver.unit_by_loc((2, 2), units)["HP"], 131)
self.assertEqual(Solver.unit_by_loc((5, 3), units)["HP"], 116)
self.assertEqual(Solver.unit_by_loc((5, 4), units)["HP"], 113)
self.assertEqual(Solver.unit_by_loc((5, 5), units)["HP"], 200)