def test_get_living_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 = "baz" unit_generator = UnitGenerator() unit = unit_generator.generate(unit_name=unit_name) dungeon.add_unit(unit) # test typical use case actual = dungeon.get_living_unit_by_name(unit.get_name()) self.assertIsInstance(actual, Unit) unit.kill() actual = dungeon.get_living_unit_by_name(unit.get_name()) self.assertIsNone(actual)