def test_get_dead_unit_by_name(self): dungeon_model = self._get_dungeon_model() with LogCapture(): logger = logging.getLogger() dungeon = Dungeon(dungeon=dungeon_model, announcer={}, log=logger) unit_name = "Oliver Queen" unit_generator = UnitGenerator() unit = unit_generator.generate(unit_name=unit_name) dungeon.add_unit(unit) unit.kill() # test typical use case actual = dungeon.get_dead_unit_by_name(unit.get_name()) self.assertIsInstance(actual, Unit) unit.hp = 1 actual_alive = dungeon.get_living_unit_by_name(unit.get_name()) self.assertIsInstance(actual_alive, Unit)