make polars optional for test

Author: blink1073

bindings/python/justfile

@@ -14,7 +14,7 @@ import-check:
     uv run python -c "from pymongoarrow.lib import libbson_version"
 
 benchmark *args:
-    uv sync --dev --extra test
+    uv sync --dev --extra test --extra test-polars
     uv run asv run -e --python=$(uv run python -c "import sys;print(sys.executable)") {{args}}
 
 install:
@@ -23,7 +23,7 @@ install:
     uv run pre-commit install
 
 test *args:
-    uv sync --extra test
+    uv sync --extra test --extra test-polars || uv sync --extra test
     uv run pytest {{args}}
 
 lint:

bindings/python/pyproject.toml

@@ -51,7 +51,8 @@ Source = "https://github.com/mongodb-labs/mongo-arrow/tree/main/bindings/python"
 Tracker = "https://jira.mongodb.org/projects/INTPYTHON/issues"
 
 [project.optional-dependencies]
-test = ["pytz", "pytest", "polars"]
+test = ["pytz", "pytest"]
+test-polars = ["polars"]
 
 [tool.setuptools]
 zip-safe = false

bindings/python/test/test_polars.py

@@ -22,9 +22,8 @@
 from test.utils import AllowListEventListener
 
 import bson
-import polars as pl
 import pyarrow as pa
-from polars.testing import assert_frame_equal
+import pytest
 from pyarrow import int32, int64
 from pymongo import DESCENDING, WriteConcern
 from pymongo.collection import Collection
@@ -40,6 +39,15 @@
     ObjectIdType,
 )
 
+try:
+    import polars as pl
+    from polars.testing import assert_frame_equal
+except ImportError:
+    pl = None
+
+if pl is None:
+    pytest.skip("Requires polars.", allow_module_level=True)
+
 
 class PolarsTestBase(unittest.TestCase):
     @classmethod