Merge "implement cython for cache_anon_map, prefix_anon_map" into main

1 files changed, 15 insertions, 0 deletions

diff --git a/lib/sqlalchemy/testing/plugin/plugin_base.py b/lib/sqlalchemy/testing/plugin/plugin_base.py
index 7bc88a14b..2a6691fc8 100644
--- a/lib/sqlalchemy/testing/plugin/plugin_base.py
+++ b/lib/sqlalchemy/testing/plugin/plugin_base.py
@@ -16,6 +16,7 @@ is pytest.
 import abc
 import configparser
 import logging
+import os
 import re
 import sys
 
@@ -370,6 +371,20 @@ def _monkeypatch_cdecimal(options, file_config):
 
 
 @post
+def __ensure_cext(opt, file_config):
+    if os.environ.get("REQUIRE_SQLALCHEMY_CEXT", "0") == "1":
+        from sqlalchemy.util import has_compiled_ext
+
+        try:
+            has_compiled_ext(raise_=True)
+        except ImportError as err:
+            raise AssertionError(
+                "REQUIRE_SQLALCHEMY_CEXT is set but can't import the "
+                "cython extensions"
+            ) from err
+
+
+@post
 def _init_symbols(options, file_config):
     from sqlalchemy.testing import config