def test_avoid_dragon_by_jumping_back(self):
map = TreasureMap()
map.start = 0
map.treasure = 3
map.roads = [(0, 1), (1, 2), (2, 3)]
map.dragons = [3]
self.assertEqual(states_shortest_path(map), [0, 1, 0, 1, 2, 3])
def test_dragon_finished_sneezing_at_start_of_journey(self):
map = TreasureMap()
map.start = 0
map.treasure = 3
map.roads = [(0, 1), (1, 2), (2, 3)]
map.dragons = [0]
self.assertEqual(states_shortest_path(map), [0, 1, 2, 3])