diff options
| -rw-r--r-- | tests/test_vba_battle.py | 11 |
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() |