def test_get_unit_by_user_id(self): dungeon_model = self._get_dungeon_model() with LogCapture(): logger = logging.getLogger() dungeon = Dungeon(dungeon=dungeon_model, announcer={}, log=logger) user_id = 42 unit_generator = UnitGenerator() unit = unit_generator.generate(user_id=user_id) dungeon.add_unit(unit) actual = dungeon.get_unit_by_user_id(user_id) self.assertIsInstance(actual, Unit)