Merge branch 'master' into Mause-patch-5

.pre-commit-config.yaml

@@ -42,6 +42,7 @@ repos:
           - "pytest==6.2.4"
           - "hypothesis==6.14.1"
           - "sqlalchemy-stubs==0.4"
+          - "types-setuptools==57.4.17"
   - repo: https://github.com/PyCQA/flake8
     rev: '4.0.1'  # pick a git hash / tag to point to
     hooks:

duckdb_engine/tests/conftest.py

@@ -1,11 +1,47 @@
-from pytest import fixture
+from typing import Any, Callable, TypeVar
+
+import duckdb
+from packaging.specifiers import SpecifierSet
+from pytest import fixture, mark
 from sqlalchemy import create_engine
 from sqlalchemy.dialects import registry
 from sqlalchemy.engine import Engine
+from typing_extensions import Protocol
+
+FuncT = TypeVar("FuncT", bound=Callable[..., Any])
 
 
 @fixture
 def engine() -> Engine:
     registry.register("duckdb", "duckdb_engine", "Dialect")  # type: ignore
 
     return create_engine("duckdb:///:memory:")
+
+
+class HasVersion(Protocol):
+    __version__: str
+
+
+def library_version(
+    library: HasVersion,
+    specifiers: str,
+) -> Callable[[FuncT], FuncT]:
+    """
+    Specify which versions of a library a test should run against
+    """
+
+    def decorator(func: FuncT) -> FuncT:
+        return mark.xfail(
+            not has_version,
+            reason=f"{library} version not desired - desired {specifiers}, found {installed}",
+        )(func)
+
+    installed = library.__version__
+    has_version = SpecifierSet(specifiers).contains(installed, prereleases=True)
+
+    return decorator
+
+
+def duckdb_version(specifiers: str) -> Callable[[FuncT], FuncT]:
+    "Specify which versions of duckdb a test should run against"
+    return library_version(duckdb, specifiers)