Robustify SQLA version determination logic

Author: Dev-iL

src/snowflake/sqlalchemy/compat.py

@@ -11,7 +11,7 @@
 string_types = (str,)
 returns_unicode = util.symbol("RETURNS_UNICODE")
 
-IS_VERSION_20 = tuple(int(v) for v in SA_VERSION.split(".")) >= (2, 0, 0)
+IS_VERSION_20 = tuple(int(v) for v in SA_VERSION.split(".")[:2]) >= (2, 0)
 
 
 def args_reducer(positions_to_drop: tuple):

tests/test_compat.py

@@ -0,0 +1,30 @@
+import importlib
+
+import pytest
+
+import snowflake.sqlalchemy.compat as compat_module
+
+
+@pytest.fixture(scope="session", autouse=True)
+def init_test_schema():
+    """Neutralize the fixture that requires a live db connection."""
+    yield
+
+
+@pytest.mark.parametrize(
+    ("version", "expected"),
+    [
+        ("2.0.0", True),
+        ("3.1.0", True),
+        ("1.3.0", False),
+        ("2.0.5.post1", True),
+        ("2.0.0rc2", True),
+        ("2.0.0b1", True),
+        ("0.5.0beta3", False),
+        ("0.4.2a", False),
+    ],
+)
+def test_is_version_20(version, expected, monkeypatch):
+    monkeypatch.setattr("sqlalchemy.__version__", version)
+    importlib.reload(compat_module)
+    assert compat_module.IS_VERSION_20 == expected