move tests into test_vba.py

They didn't belong in pokemontools/vba/vba.py in the first place.

1 files changed, 30 insertions, 0 deletions

diff --git a/tests/test_vba.py b/tests/test_vba.py
index 56a71e3..12dd51b 100644
--- a/tests/test_vba.py
+++ b/tests/test_vba.py
@@ -102,5 +102,35 @@ class VbaTests(unittest.TestCase):
         player_action = self.get_wram_value("PlayerAction")
         self.assertEqual(player_action, 1) # 1 = standing
 
+class TestEmulator(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.cry = crystal()
+
+        # advance it forward past the intro sequences
+        cls.cry.vba.step(count=3500)
+
+    def test_PlaceString(self):
+        self.cry.call(0, 0x1078)
+
+        # where to draw the text
+        self.cry.registers["hl"] = 0xc4a0
+
+        # what text to read from
+        self.cry.registers["de"] = 0x1276
+
+        self.cry.vba.step(count=10)
+
+        text = self.cry.get_text()
+
+        self.assertTrue("TRAINER" in text)
+
+    def test_keyboard_planner(self):
+        button_sequence = keyboard.plan_typing("an")
+        expected_result = ["select", "a", "d", "r", "r", "r", "r", "a"]
+
+        self.assertEqual(len(expected_result), len(button_sequence))
+        self.assertEqual(expected_result, button_sequence)
+
 if __name__ == "__main__":
     unittest.main()