summaryrefslogtreecommitdiff
path: root/tests/test_vba_battle.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_vba_battle.py')
-rw-r--r--tests/test_vba_battle.py11
1 files changed, 7 insertions, 4 deletions
diff --git a/tests/test_vba_battle.py b/tests/test_vba_battle.py
index 6d3b504..acf9d7d 100644
--- a/tests/test_vba_battle.py
+++ b/tests/test_vba_battle.py
@@ -39,13 +39,16 @@ class BattleTests(unittest.TestCase):
def setUp(self):
# reset to whatever the bootstrapper created
self.vba.state = self.bootstrap_state
+ self.battle = Battle(emulator=self.cry)
- def test_battle_is_player_turn(self):
- self.cry.vba.state = self.bootstrap_state
+ def test_is_in_battle(self):
+ self.assertTrue(self.battle.is_in_battle())
- battle = Battle(emulator=self.cry)
+ def test_is_player_turn(self):
+ self.battle.skip_start_text()
- self.assertTrue(battle.is_player_turn())
+ # the initial state should be the player's turn
+ self.assertTrue(self.battle.is_player_turn())
if __name__ == "__main__":
unittest.main()