def test_remove_avatar(self): world_map = InfiniteMap() manager = DummyAvatarManager() game_state = GameState(world_map, manager) avatar1 = DummyAvatar(1, Location(0, 0)) avatar2 = DummyAvatar(2, Location(1, 1)) avatar2.marked = True manager.add_avatar_directly(avatar1) world_map.get_cell(Location(0, 0)).avatar = avatar1 manager.add_avatar_directly(avatar2) world_map.get_cell(Location(1, 1)).avatar = avatar2 game_state.remove_avatar(1) self.assertNotIn(1, manager.avatars_by_id) self.assertEqual(world_map.get_cell(Location(0, 0)).avatar, None) self.assertTrue(manager.avatars_by_id[2].marked) self.assertTrue(world_map.get_cell(Location(1, 1)).avatar.marked)
def test_remove_avatar(self): world_map = InfiniteMap() manager = DummyAvatarManager() game_state = GameState(world_map, manager) avatar1 = DummyAvatar(1, Location(0, 0)) avatar2 = DummyAvatar(2, Location(1, 1)) avatar2.marked = True manager.add_avatar_directly(avatar1) world_map.get_cell(Location(0, 0)).avatar = avatar1 manager.add_avatar_directly(avatar2) world_map.get_cell(Location(1, 1)).avatar = avatar2 game_state.remove_avatar(1) self.assertNotIn(1, manager.avatars_by_id) self.assertEqual(world_map.get_cell((0, 0)).avatar, None) self.assertTrue(manager.avatars_by_id[2].marked) self.assertTrue(world_map.get_cell(Location(1, 1)).avatar.marked)
def test_remove_non_existant_avatar(self): state = GameState(None, DummyAvatarManager()) state.remove_avatar(10)
def test_remove_non_existant_avatar(self): state = GameState(None, DummyAvatarManager()) state.remove_avatar(10)