fix the is_switch_prompt test

1 files changed, 3 insertions, 3 deletions

diff --git a/tests/test_vba_battle.py b/tests/test_vba_battle.py
index 8299bea..a0338e5 100644
--- a/tests/test_vba_battle.py
+++ b/tests/test_vba_battle.py
@@ -104,12 +104,12 @@ class BattleTests(unittest.TestCase):
         # attack the enemy and kill it
         self.battle.skip_until_input_required()
 
-        import time
-        time.sleep(1)
-
         # yes/no menu is present, should be detected
         self.assertTrue(self.battle.is_switch_prompt())
 
+        # and input should be required
+        self.assertTrue(self.is_input_required())
+
         # but it's not mandatory
         self.assertFalse(self.battle.is_mandatory_switch())