diff --git a/squirrelbattle/tests/game_test.py b/squirrelbattle/tests/game_test.py index 40a2331..f8a6e93 100644 --- a/squirrelbattle/tests/game_test.py +++ b/squirrelbattle/tests/game_test.py @@ -596,3 +596,8 @@ class TestGame(unittest.TestCase): self.assertEqual(self.game.state, GameMode.CREDITS) self.game.display_actions(DisplayActions.MOUSE, 21, 21) self.game.display_actions(DisplayActions.REFRESH) + + self.game.state = GameMode.CREDITS + self.game.handle_key_pressed(KeyValues.ENTER) + + self.assertEqual(self.game.state, GameMode.MAINMENU)