From ba6eb203b4afef6179d30a13f6f4037349d79272 Mon Sep 17 00:00:00 2001 From: Bryan Bishop Date: Mon, 11 Nov 2013 21:48:29 -0600 Subject: a test for detecting the yes/no prompt --- tests/test_vba_battle.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/test_vba_battle.py b/tests/test_vba_battle.py index 8b02cef..8299bea 100644 --- a/tests/test_vba_battle.py +++ b/tests/test_vba_battle.py @@ -88,5 +88,30 @@ class BattleTests(unittest.TestCase): self.assertTrue(self.battle.is_player_turn()) + def test_is_battle_switch_prompt(self): + self.battle.skip_start_text() + self.battle.skip_until_input_required() + + # press "FIGHT" + self.vba.press(["a"], after=20) + + # press the first move ("SCRATCH") + self.vba.press(["a"], after=20) + + # set enemy hp to very low + self.cry.lower_enemy_hp() + + # 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()) + + # but it's not mandatory + self.assertFalse(self.battle.is_mandatory_switch()) + if __name__ == "__main__": unittest.main() -- cgit v1.2.3