test polars

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

packages/python/plotly/plotly/tests/test_optional/test_px/test_px_input.py

@@ -273,12 +273,16 @@ def __dataframe__(self):
     version.parse(pd.__version__) < version.parse("2.0.2"),
     reason="plotly doesn't use a dataframe interchange protocol for pandas < 2.0.2",
 )
-def test_build_df_from_vaex():
-    import vaex
+@pytest.mark.parametrize("test_lib", ["vaex", "polars"])
+def test_build_df_from_vaex_and_polars(test_lib):
+    if test_lib == "vaex":
+        import vaex as lib
+    else:
+        import polars as lib
 
     # take out the 'species' columns since the vaex implementation does not cover strings yet
     iris_pandas = px.data.iris()[["petal_width", "sepal_length"]]
-    iris_vaex = vaex.from_pandas(iris_pandas)
+    iris_vaex = lib.from_pandas(iris_pandas)
     args = dict(data_frame=iris_vaex, x="petal_width", y="sepal_length")
     out = build_dataframe(args, go.Scatter)
     assert_frame_equal(

packages/python/plotly/test_requirements/requirements_39_pandas_2_optional.txt

@@ -20,3 +20,4 @@ scikit-image==0.18.1
 psutil==5.7.0
 kaleido
 vaex
+polars