Add polars to test packages and handle if polars not installed

holoviz · Sep 7, 2023 · 8f60375 · 8f60375
1 parent 638b509
commit 8f60375
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/hvplot/tests/plotting/testcore.py b/hvplot/tests/plotting/testcore.py
@@ -7,8 +7,13 @@
 try:
     import polars as pl
     import hvplot.polars  # noqa
+    pl_import = True
 except ImportError:
-    pl = None
+    class pl:
+        DataFrame = None
+        LazyFrame = None
+        Series = None
+    pl_import = False
 
 
 FRAME_IGNORE_TYPES = {"bivariate", "heatmap", "hexbin", "labels", "vectorfield"}
@@ -44,7 +49,7 @@ def test_series_pandas():
 
 
 
-@pytest.mark.skipif(pl is None, reason="polars not installed")
+@pytest.mark.skipif(pl_import, reason="polars not installed")
 @pytest.mark.parametrize("cast", (pl.DataFrame, pl.LazyFrame))
 @y_combinations
 def test_diffent_input_types_polars(y, cast):
@@ -56,7 +61,7 @@ def test_diffent_input_types_polars(y, cast):
         df.hvplot(y=y, kind=t)
 
 
-@pytest.mark.skipif(pl is None, reason="polars not installed")
+@pytest.mark.skipif(pl_import, reason="polars not installed")
 def test_series_polars():
     ser = pl.Series(values=np.random.rand(10), name="A")
     assert isinstance(ser, pl.Series)

diff --git a/setup.py b/setup.py
@@ -60,6 +60,7 @@ def get_setup_version(reponame):
     'pooch',
     'scipy',
     'ipywidgets',
+    'polars',
 ]
 
 # Dependencies required to run the notebooks