Add tokens test

TI-Toolkit · Dec 19, 2024 · 17aa6d7 · 17aa6d7
1 parent 05208f9
commit 17aa6d7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/tivars.py b/tests/tivars.py
@@ -4,6 +4,7 @@
 from decimal import Decimal
 
 from tivars.models import *
+from tivars.tokenizer import *
 from tivars.types import *
 from tivars import TIHeader, TIVar, TIFlashHeader
 
@@ -170,6 +171,8 @@ def test_load_from_string(self):
         test_program.load_string(string := "setDate(1")
         self.assertEqual(test_program.string(), string)
         self.assertEqual(f"{test_program:02d: }", f"00: {string}")
+        self.assertEqual(test_program.tokens(), [TI_84PCE.tokens.bytes[b'\xef\x00'],
+                                                 TI_84PCE.tokens.bytes[b'1']])
 
         # Version is wrong(?)
         test_program.version = 0x04