def test_movement(self):
input_list = [
'#########', '#G..G..G#', '#.......#', '#.......#', '#G..E..G#',
'#.......#', '#.......#', '#G..G..G#', '#########'
]
output_list = [
'#########', '#.......#', '#..GGG..#', '#..GEG..#', '#G..G...#',
'#......G#', '#.......#', '#.......#', '#########'
]
map_data = day15.parse_input(input_list)
day15.combat(map_data, max_rounds=3)
map_output = day15.print_map(map_data)
self.assertEqual('\n'.join(output_list) + '\n', map_output)
def test_battle6(self):
input_list = [
'#########', '#G......#', '#.E.#...#', '#..##..G#', '#...##..#',
'#...#...#', '#.G...G.#', '#.....G.#', '#########'
]
map_data = day15.parse_input(input_list)
final_sum, _ = day15.combat(map_data)
self.assertEqual(18740, final_sum)
def test_battle5(self):
input_list = [
'#######', '#.E...#', '#.#..G#', '#.###.#', '#E#G#G#', '#...#G#',
'#######'
]
map_data = day15.parse_input(input_list)
final_sum, _ = day15.combat(map_data)
self.assertEqual(28944, final_sum)
def test_battle4(self):
input_list = [
'#######', '#E.G#.#', '#.#G..#', '#G.#.G#', '#G..#.#', '#...E.#',
'#######'
]
map_data = day15.parse_input(input_list)
final_sum, _ = day15.combat(map_data)
self.assertEqual(27755, final_sum)
def test_battle3(self):
input_list = [
'#######', '#E..EG#', '#.#G.E#', '#E.##E#', '#G..#.#', '#..E#.#',
'#######'
]
map_data = day15.parse_input(input_list)
final_sum, _ = day15.combat(map_data)
self.assertEqual(39514, final_sum)
def test_battle2(self):
input_list = [
'#######', '#G..#E#', '#E#E.E#', '#G.##.#', '#...#E#', '#...E.#',
'#######'
]
map_data = day15.parse_input(input_list)
final_sum, _ = day15.combat(map_data)
self.assertEqual(36334, final_sum)
def test_battle1(self):
input_list = [
'#######', '#.G...#', '#...EG#', '#.#.#G#', '#..G#E#', '#.....#',
'#######'
]
map_data = day15.parse_input(input_list)
final_sum, _ = day15.combat(map_data)
self.assertEqual(27730, final_sum)