def test_get_living_players(self): dungeon_model = self._get_dungeon_model() with LogCapture(): logger = logging.getLogger() dungeon = Dungeon(dungeon=dungeon_model, announcer={}, log=logger) unit_generator = UnitGenerator() unit = unit_generator.generate(is_player=True) dungeon.add_unit(unit) dead_unit = unit_generator.generate() dead_unit.kill() dungeon.add_unit(dead_unit) # test typical use case actual = dungeon.get_living_players() self.assertEqual(len(actual), 1) dead_players = dungeon.get_dead_players() self.assertEqual(len(dead_players), 0)