by version

tests/assignment_test.py

@@ -75,7 +75,7 @@ class ImportAssignmentTests(DisplayAnnotatedTestCase):
     def test_global_import(self):
         self.assertAnnotationWorks(
             """
-            import os
+            import {>=3.10!g}os
             {g}def f():
                 {g}os
             """
@@ -84,20 +84,21 @@ def test_local_import(self):
         self.assertAnnotationWorks(
             """
             {g}def f():
-                import os
+                import {>=3.10!~f@1:0}os
                 {~f@1:0}os
             """
         )
     def test_sub_import(self):
         self.assertAnnotationWorks(
             """
             {g}def f():
-                from os import system
+                from os import {>=3.10!~f@1:0}system
                 {~f@1:0}system
                 {g}os
             """
         )
     def test_star_import(self):
+        # note: no annotation in any version
         self.assertAnnotationWorks(
             """
             from os import *
@@ -107,8 +108,8 @@ def test_aliases(self):
         self.assertAnnotationWorks(
             """
             {g}def f():
-                from os import system as a
-                import sys as b
+                from os import {>=3.10!~f@1:0}system as a
+                import {>=3.10!~f@1:0}sys as b
                 {g}os
                 {g}system
                 {~f@1:0}a

tests/utils.py

@@ -101,6 +101,14 @@ def _check_nodes(self, mapping, *scopes):
         self.assertCountEqual([node for _, node in overall_scope], list(mapping))
 
     def assertAnnotationWorks(self, annotated_code, code=None, *, class_binds_near=False):
+        # directives of the form {>version!scope} are removed unless the version is satisfied
+        regex = r"\{>=(\d+\.\d+)!([^\}]+)\}"
+        def replacer(match):
+            version, scope = match.groups()
+            if sys.version_info >= tuple(map(int, version.split("."))):
+                return "{" + scope + "}"
+            return ""
+        annotated_code = re.sub(regex, replacer, annotated_code)
         if code is None:
             code = trim(re.sub(r"\{[^\}]+\}", "", annotated_code))