tests/basics/fun_code_lnotab: Test removal of co_lnotab from v2.

AJMansfield · AJMansfield · commit 7b38fa4fa3a1 · 2025-07-18T10:59:21.000-04:00
Signed-off-by: Anson Mansfield &lt;amansfield@mantaro.com&gt;
diff --git a/tests/basics/fun_code_full.py b/tests/basics/fun_code_full.py
@@ -6,12 +6,6 @@
     print("SKIP")
     raise SystemExit
 
-try:
-    import warnings
-    warnings.simplefilter("ignore") # ignore deprecation warning about co_lnotab
-except ImportError:
-    pass
-
 def f(x, y):
     a = x + y
     b = x - y
@@ -25,7 +19,6 @@ def f(x, y):
 print(type(code.co_firstlineno)) # both ints (but mpy points to first line inside, cpy points to declaration)
 print(code.co_name)
 print(iter(code.co_names) is not None) # both iterable (but mpy returns dict with names as keys, cpy only the names; and not necessarily the same set)
-print(type(code.co_lnotab)) # both bytes
 
 co_lines = code.co_lines()
 
diff --git a/tests/basics/fun_code_lnotab.py b/tests/basics/fun_code_lnotab.py
@@ -0,0 +1,34 @@
+# Test deprecation of co_lnotab
+
+try:
+    (lambda: 0).__code__.co_code
+except AttributeError:
+    print("SKIP")
+    raise SystemExit
+
+
+import unittest
+import sys
+
+
+mpy_is_v2 = getattr(sys.implementation, '_v2', False)
+
+
+def f():
+    pass
+
+
+class Test(unittest.TestCase):
+
+    @unittest.skipIf(mpy_is_v2, "Removed in MicroPython v2 and later.")
+    def test_co_lnotab_exists(self):
+        self.assertIsInstance(f.__code__.co_lnotab, bytes)
+
+    @unittest.skipUnless(mpy_is_v2, "Not removed before MicroPython v2.")
+    def test_co_lnotab_removed(self):
+        with self.assertRaises(AttributeError):
+            f.__code__.co_lnotab
+
+
+if __name__ == "__main__":
+    unittest.main()
