diff --git a/squirrelbattle/enums.py b/squirrelbattle/enums.py index 42bd643..b61f236 100644 --- a/squirrelbattle/enums.py +++ b/squirrelbattle/enums.py @@ -53,7 +53,8 @@ class KeyValues(Enum): DANCE = auto() @staticmethod - def translate_key(key: str, settings: Settings) -> Optional["KeyValues"]: + def translate_key(key: str, settings: Settings) \ + -> Optional["KeyValues"]: # noqa: C901 """ Translates the raw string key into an enum value that we can use. """ @@ -91,4 +92,3 @@ class KeyValues(Enum): return KeyValues.LAUNCH elif key == settings.KEY_DANCE: return KeyValues.DANCE - return None diff --git a/squirrelbattle/tests/game_test.py b/squirrelbattle/tests/game_test.py index a3c3a87..895af12 100644 --- a/squirrelbattle/tests/game_test.py +++ b/squirrelbattle/tests/game_test.py @@ -260,12 +260,10 @@ class TestGame(unittest.TestCase): self.game.handle_key_pressed(KeyValues.DANCE) self.assertEqual(rabbit.confused, 1) string = rabbit.hit(self.game.player) - self.assertEqual(string, - "{name} is confused, it can not hit {opponent}." - .format(name=_(rabbit.translated_name.capitalize() - ), opponent=_( - self.game.player.translated_name - ))) + self.assertEqual( + string, _("{name} is confused, it can not hit {opponent}.") + .format(name=rabbit.translated_name.capitalize(), + opponent=self.game.player.translated_name)) rabbit.confused = 0 self.game.player.charisma = 0 self.game.handle_key_pressed(KeyValues.DANCE)