diff --git a/test/base/TestFill.py b/test/base/TestFill.py index 152bd8c3..d231fa15 100644 --- a/test/base/TestFill.py +++ b/test/base/TestFill.py @@ -30,7 +30,9 @@ def generate_locations(count: int, player_id: int, address: int = None, region: locations = [] for i in range(count): name = "player" + str(player_id) + "_location" + str(i) - locations.append(Location(player_id, name, address, region)) + location = Location(player_id, name, address, region) + locations.append(location) + region.locations.append(location) return locations @@ -103,4 +105,4 @@ class TestBase(unittest.TestCase): fill_restrictive(multi_world, multi_world.state, locations, items) self.assertEqual(loc0.item, item1) - self.assertEqual(loc1.item, item0) + self.assertEqual(loc1.item, item0) \ No newline at end of file