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)
