Robustify SQLA version determination logic

Author: Dev-iL

pyproject.toml

@@ -50,6 +50,7 @@ development = [
   "pytest",
   "setuptools",
   "pytest-cov",
+  "pytest-mock",
   "pytest-timeout",
   "pytest-rerunfailures",
   "pytest-xdist",

src/snowflake/sqlalchemy/compat.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import functools
+import re
 from typing import Callable
 
 from sqlalchemy import __version__ as SA_VERSION
@@ -11,7 +12,8 @@
 string_types = (str,)
 returns_unicode = util.symbol("RETURNS_UNICODE")
 
-IS_VERSION_20 = tuple(int(v) for v in SA_VERSION.split(".")) >= (2, 0, 0)
+_match = re.match(r"(\d+)\.(\d+)\.(\d+)", SA_VERSION)
+IS_VERSION_20 = tuple(int(x) for x in _match.groups()) >= (2, 0, 0)
 
 
 def args_reducer(positions_to_drop: tuple):

tests/test_compat.py

@@ -0,0 +1,24 @@
+import importlib
+
+import pytest
+
+import snowflake.sqlalchemy.compat as compat_module
+
+
+@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, mocker):
+    mocker.patch("sqlalchemy.__version__", version)
+    importlib.reload(compat_module)
+    assert compat_module.IS_VERSION_20 == expected