Add test locations to region

This commit is contained in:
Brad Humphrey 2021-12-20 07:59:36 -07:00 committed by Fabian Dill
parent 39869bcdc5
commit 461961c3be
1 changed files with 4 additions and 2 deletions

View File

@ -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)