write another quick battle test

1 files changed, 5 insertions, 0 deletions

diff --git a/tests/test_vba_battle.py b/tests/test_vba_battle.py
index acf9d7d..12b991c 100644
--- a/tests/test_vba_battle.py
+++ b/tests/test_vba_battle.py
@@ -46,9 +46,14 @@ class BattleTests(unittest.TestCase):
 
     def test_is_player_turn(self):
         self.battle.skip_start_text()
+        self.battle.skip_until_input_required()
 
         # the initial state should be the player's turn
         self.assertTrue(self.battle.is_player_turn())
 
+    def test_is_mandatory_switch_initial(self):
+        # should not be asking for a switch so soon in the battle
+        self.assertFalse(self.battle.is_mandatory_switch())
+
 if __name__ == "__main__":
     unittest.main()