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_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")