fix

tests/compiler_test.py

@@ -36,7 +36,8 @@
 except ImportError:
     from sqlalchemy.ext.declarative import declarative_base
 
-from sqlalchemy_cratedb import SA_1_4, SA_2_0, SA_VERSION, ObjectType
+from sqlalchemy_cratedb import ObjectType
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_2_0, SA_VERSION
 
 from .util import ParametrizedTestCase
 

tests/datetime_test.py

@@ -29,8 +29,8 @@
 import sqlalchemy as sa
 from sqlalchemy.orm import Session, sessionmaker
 
-from sqlalchemy_cratedb import SA_1_4, SA_VERSION
 from sqlalchemy_cratedb.dialect import DateTime
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_VERSION
 
 try:
     from sqlalchemy.orm import declarative_base

tests/dialect_test.py

@@ -28,7 +28,8 @@
 from sqlalchemy import inspect
 from sqlalchemy.orm import Session
 
-from sqlalchemy_cratedb import SA_1_4, SA_2_0, SA_VERSION, ObjectType
+from sqlalchemy_cratedb import ObjectType
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_2_0, SA_VERSION
 
 try:
     from sqlalchemy.orm import declarative_base

tests/dict_test.py

@@ -35,7 +35,8 @@
 
 from crate.client.cursor import Cursor
 
-from sqlalchemy_cratedb import SA_1_4, SA_VERSION, ObjectArray, ObjectType
+from sqlalchemy_cratedb import ObjectArray, ObjectType
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_VERSION
 
 fake_cursor = MagicMock(name="fake_cursor")
 FakeCursor = MagicMock(name="FakeCursor", spec=Cursor)

tests/insert_from_select_test.py

@@ -25,7 +25,7 @@
 import sqlalchemy as sa
 from sqlalchemy.orm import Session
 
-from sqlalchemy_cratedb import SA_1_4, SA_VERSION
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_VERSION
 
 try:
     from sqlalchemy.orm import declarative_base

tests/query_caching.py

@@ -28,7 +28,8 @@
 from sqlalchemy.orm import Session
 from sqlalchemy.sql.operators import eq
 
-from sqlalchemy_cratedb import SA_1_4, SA_VERSION, ObjectArray, ObjectType
+from sqlalchemy_cratedb import ObjectArray, ObjectType
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_VERSION
 
 try:
     from sqlalchemy.orm import declarative_base

tests/test_support_polyfill.py

@@ -6,7 +6,7 @@
 from sqlalchemy.exc import IntegrityError
 from sqlalchemy.orm import sessionmaker
 
-from sqlalchemy_cratedb import SA_1_4, SA_VERSION
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_VERSION
 
 try:
     from sqlalchemy.orm import declarative_base

tests/update_test.py

@@ -25,7 +25,8 @@
 import sqlalchemy as sa
 from sqlalchemy.orm import Session
 
-from sqlalchemy_cratedb import SA_1_4, SA_VERSION, ObjectType
+from sqlalchemy_cratedb import ObjectType
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_VERSION
 
 try:
     from sqlalchemy.orm import declarative_base

tests/vector_test.py

@@ -38,7 +38,8 @@
 
 from crate.client.cursor import Cursor
 
-from sqlalchemy_cratedb import SA_1_4, SA_VERSION, FloatVector, knn_match
+from sqlalchemy_cratedb import FloatVector, knn_match
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_VERSION
 from sqlalchemy_cratedb.type.vector import from_db, to_db
 
 fake_cursor = MagicMock(name="fake_cursor")

tests/warnings_test.py

@@ -3,7 +3,7 @@
 import warnings
 from unittest import TestCase, skipIf
 
-from sqlalchemy_cratedb import SA_1_4, SA_VERSION
+from sqlalchemy_cratedb.sa_version import SA_1_4, SA_VERSION
 from tests.util import ExtraAssertions