main commit

venv/lib/python3.12/site-packages/alembic/testing/requirements.py

@@ -1,6 +1,7 @@
 from sqlalchemy.testing.requirements import Requirements
 
 from alembic import util
+from alembic.util import sqla_compat
 from ..testing import exclusions
 
 
@@ -73,6 +74,13 @@ class SuiteRequirements(Requirements):
     def reflects_fk_options(self):
         return exclusions.closed()
 
+    @property
+    def sqlalchemy_14(self):
+        return exclusions.skip_if(
+            lambda config: not util.sqla_14,
+            "SQLAlchemy 1.4 or greater required",
+        )
+
     @property
     def sqlalchemy_1x(self):
         return exclusions.skip_if(
@@ -87,18 +95,6 @@ class SuiteRequirements(Requirements):
             "SQLAlchemy 2.x test",
         )
 
-    @property
-    def asyncio(self):
-        def go(config):
-            try:
-                import greenlet  # noqa: F401
-            except ImportError:
-                return False
-            else:
-                return True
-
-        return exclusions.only_if(go)
-
     @property
     def comments(self):
         return exclusions.only_if(
@@ -113,6 +109,26 @@ class SuiteRequirements(Requirements):
     def computed_columns(self):
         return exclusions.closed()
 
+    @property
+    def computed_columns_api(self):
+        return exclusions.only_if(
+            exclusions.BooleanPredicate(sqla_compat.has_computed)
+        )
+
+    @property
+    def computed_reflects_normally(self):
+        return exclusions.only_if(
+            exclusions.BooleanPredicate(sqla_compat.has_computed_reflection)
+        )
+
+    @property
+    def computed_reflects_as_server_default(self):
+        return exclusions.closed()
+
+    @property
+    def computed_doesnt_reflect_as_server_default(self):
+        return exclusions.closed()
+
     @property
     def autoincrement_on_composite_pk(self):
         return exclusions.closed()
@@ -174,3 +190,9 @@ class SuiteRequirements(Requirements):
     @property
     def identity_columns_alter(self):
         return exclusions.closed()
+
+    @property
+    def identity_columns_api(self):
+        return exclusions.only_if(
+            exclusions.BooleanPredicate(sqla_compat.has_identity)
+        )