summaryrefslogtreecommitdiff
path: root/alembic/testing/util.py
diff options
context:
space:
mode:
Diffstat (limited to 'alembic/testing/util.py')
-rw-r--r--alembic/testing/util.py11
1 files changed, 3 insertions, 8 deletions
diff --git a/alembic/testing/util.py b/alembic/testing/util.py
index e65597d..4517a69 100644
--- a/alembic/testing/util.py
+++ b/alembic/testing/util.py
@@ -6,12 +6,13 @@
# the MIT License: http://www.opensource.org/licenses/mit-license.php
from __future__ import annotations
-import re
import types
from typing import Union
from sqlalchemy.util import inspect_getfullargspec
+from ..util import sqla_2
+
def flag_combinations(*combinations):
"""A facade around @testing.combinations() oriented towards boolean
@@ -114,17 +115,11 @@ def _safe_int(value: str) -> Union[int, str]:
def testing_engine(url=None, options=None, future=False):
from sqlalchemy.testing import config
from sqlalchemy.testing.engines import testing_engine
- from sqlalchemy import __version__
-
- _vers = tuple(
- [_safe_int(x) for x in re.findall(r"(\d+|[abc]\d)", __version__)]
- )
- sqla_1x = _vers < (2,)
if not future:
future = getattr(config._current.options, "future_engine", False)
- if sqla_1x:
+ if not sqla_2:
kw = {"future": future} if future else {}
else:
kw = {}