def test_pickups_not_added_when_at_target(self):
world_map.TARGET_NUM_PICKUPS_PER_AVATAR = 1
grid = self._generate_grid()
grid[Location(0, 1)].pickup = MockPickup()
map = WorldMap(grid)
map.update(1)
self.assertEqual(len(list(map.pickup_cells())), 1)
self.assertIn(grid[Location(0, 1)], map.pickup_cells())
def test_pickups_added(self):
self.settings['TARGET_NUM_PICKUPS_PER_AVATAR'] = 1
self.settings['PICKUP_SPAWN_CHANCE'] = 1
world_map = WorldMap(self._generate_grid(), self.settings)
world_map.update(1)
self.assertEqual(len(list(world_map.pickup_cells())), 1)
world_map.update(2)
self.assertEqual(len(list(world_map.pickup_cells())), 2)
def test_pickups_added(self):
world_map.TARGET_NUM_PICKUPS_PER_AVATAR = 1
world_map.PICKUP_SPAWN_CHANCE = 1
map = WorldMap(self._generate_grid())
map.update(1)
self.assertEqual(len(list(map.pickup_cells())), 1)
map.update(2)
self.assertEqual(len(list(map.pickup_cells())), 2)
def test_not_enough_pickup_space(self):
world_map.TARGET_NUM_PICKUPS_PER_AVATAR = 1
grid = self._generate_grid(1, 1)
grid[Location(0, 0)].generates_score = True
map = WorldMap(grid)
map.update(1)
self.assertEqual(len(list(map.pickup_cells())), 0)
def test_pickup_spawn_chance(self):
world_map.TARGET_NUM_PICKUPS_PER_AVATAR = 5
world_map.PICKUP_SPAWN_CHANCE = 0
grid = self._generate_grid()
map = WorldMap(grid)
map.update(1)
self.assertEqual(len(list(map.pickup_cells())), 0)
def test_pickup_spawn_chance(self):
self.settings['TARGET_NUM_PICKUPS_PER_AVATAR'] = 5
self.settings['PICKUP_SPAWN_CHANCE'] = 0
grid = self._generate_grid()
world_map = WorldMap(grid, self.settings)
world_map.update(1)
self.assertEqual(len(list(world_map.pickup_cells())), 0)
def test_not_enough_pickup_space(self):
world_map.TARGET_NUM_PICKUPS_PER_AVATAR = 1
grid = self._generate_grid(1)
grid = [[MockCell(avatar='avatar')]]
map = WorldMap(grid)
map.reconstruct_interactive_state(1)
self.assertEqual(len(list(map.pickup_cells())), 0)
def test_pickup_cells(self):
cells = self._generate_cells()
cells[0]['pickup'] = {'health_restored': 5}
cells[8]['pickup'] = {'health_restored': 2}
map = WorldMap(cells)
self.assertLocationsEqual(map.pickup_cells(),
(Location(-1, -1), Location(1, 1)))
def test_not_enough_pickup_space(self):
world_map.TARGET_NUM_PICKUPS_PER_AVATAR = 1
grid = self._generate_grid(1)
grid = [[MockCell(avatar='avatar')]]
map = WorldMap(grid)
map.reconstruct_interactive_state(1)
self.assertEqual(len(list(map.pickup_cells())), 0)
def test_pickups_not_added_when_at_target(self):
world_map.TARGET_NUM_PICKUPS_PER_AVATAR = 1
grid = self._generate_grid()
grid[0][1].pickup = True
map = WorldMap(grid)
map.reconstruct_interactive_state(1)
self.assertEqual(len(list(map.pickup_cells())), 1)
self.assertIn(grid[0][1], map.pickup_cells())
def test_pickups_not_added_when_at_target(self):
world_map.TARGET_NUM_PICKUPS_PER_AVATAR = 1
grid = self._generate_grid()
grid[0][1].pickup = True
map = WorldMap(grid)
map.reconstruct_interactive_state(1)
self.assertEqual(len(list(map.pickup_cells())), 1)
self.assertIn(grid[0][1], map.pickup_cells())
def test_pickup_cells(self):
pickup_cell1 = MockCell(pickup=MockPickup())
pickup_cell2 = MockCell(pickup=MockPickup())
no_pickup_cell = MockCell()
grid = self._grid_from_list([[pickup_cell1, no_pickup_cell], [no_pickup_cell, pickup_cell2]])
map = WorldMap(grid)
cells = list(map.pickup_cells())
self.assertIn(pickup_cell1, cells)
self.assertIn(pickup_cell2, cells)
self.assertEqual(len(cells), 2, "Non-pickup cells present")
def test_get_all_pickup_locations(self):
pickup_cell = MockCell()
pickup_cell.interactable = HealthPickup(pickup_cell)
grid = self._grid_from_list([[pickup_cell, MockCell()],
[MockCell(), MockCell()]])
world_map = WorldMap(grid, self.settings)
pickup_list = list(world_map.pickup_cells())
self.assertEqual(len(pickup_list), 1)
self.assertTrue(isinstance(pickup_list[0].interactable, ALL_PICKUPS))
def test_pickup_cells(self):
pickup_cell1 = MockCell(pickup='pickup1')
pickup_cell2 = MockCell(pickup='pickup')
no_pickup_cell = MockCell()
grid = [[pickup_cell1, no_pickup_cell], [no_pickup_cell, pickup_cell2]]
map = WorldMap(grid)
cells = list(map.pickup_cells())
self.assertIn(pickup_cell1, cells)
self.assertIn(pickup_cell2, cells)
self.assertEqual(len(cells), 2, "Non-pickup cells present")
def test_pickup_cells(self):
pickup_cell1 = MockCell(pickup='pickup1')
pickup_cell2 = MockCell(pickup='pickup')
no_pickup_cell = MockCell()
grid = [[pickup_cell1, no_pickup_cell], [no_pickup_cell, pickup_cell2]]
map = WorldMap(grid)
cells = list(map.pickup_cells())
self.assertIn(pickup_cell1, cells)
self.assertIn(pickup_cell2, cells)
self.assertEqual(len(cells), 2, "Non-pickup cells present")
def test_pickup_cells(self):
cells = self._generate_cells()
cells[0]['pickup'] = {'health_restored': 5}
cells[8]['pickup'] = {'health_restored': 2}
map = WorldMap(cells)
self.assertLocationsEqual(map.pickup_cells(), (Location(-1, -1), Location(1, 1)))
