def test_intra_room_shunts(self):
expected_shunted = load_cave(
get_test_file_path("samples/d23/unfolded/shunted.txt"))
cave = load_cave(
get_test_file_path("samples/d23/unfolded/to_shunt.txt"))
auto_shunt(cave)
self.assertEqual(get_cave_hash(cave), get_cave_hash(expected_shunted))
self.assertEqual(cave.total_cost, 2001)
def test_nothing_to_shunt(self):
cave = load_cave(
get_test_file_path("samples/d23/unfolded/mid_example.txt"))
shunted = load_cave(
get_test_file_path("samples/d23/unfolded/mid_example.txt"))
auto_shunt(shunted)
self.assertEqual(get_cave_hash(cave), get_cave_hash(shunted))
self.assertEqual(cave.total_cost, shunted.total_cost)
def test_corridor_shunt(self):
expected_shunted = load_cave(
get_test_file_path("samples/d23/unfolded/corridor_shunted.txt"))
cave = load_cave(
get_test_file_path("samples/d23/unfolded/to_corridor_shunt.txt"))
auto_shunt(cave)
self.assertEqual(get_cave_hash(cave), get_cave_hash(expected_shunted))
cost = 1 + 2000 + 4000 + 9000
self.assertEqual(cave.total_cost, cost)
def test_corridor(self):
cave = load_cave(
get_test_file_path("samples/d23/example_corridor.txt"))
self.assertEqual(get_cave_hash(cave), ".....D.D.A.A.BBCC..")
def test_folded_complete(self):
cave = load_cave(
get_test_file_path("samples/d23/unfolded/complete.txt"))
self.assertEqual(get_cave_hash(cave), "...........AAAABBBBCCCCDDDD")