return vaex test

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

Author: anmyachev

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

@@ -268,6 +268,19 @@ def __dataframe__(self):
         mock_from_dataframe.assert_called_once_with(input_dataframe)
 
 
+def test_build_df_from_vaex():
+    import vaex
+
+    # 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)
+    args = dict(data_frame=iris_vaex, x="petal_width", y="sepal_length")
+    out = build_dataframe(args, go.Scatter)
+    assert_frame_equal(
+        iris_pandas.reset_index()[out["data_frame"].columns], out["data_frame"]
+    )
+
+
 def test_timezones():
     df = pd.DataFrame({"date": ["2015-04-04 19:31:30+1:00"], "value": [3]})
     df["date"] = pd.to_datetime(df["date"])

packages/python/plotly/test_requirements/requirements_36_optional.txt

@@ -19,3 +19,4 @@ matplotlib==2.2.3
 scikit-image==0.14.4
 psutil==5.7.0
 kaleido
+vaex

packages/python/plotly/test_requirements/requirements_37_optional.txt

@@ -20,3 +20,4 @@ scikit-image==0.14.4
 psutil==5.7.0
 kaleido
 orjson==3.8.12
+vaex

packages/python/plotly/test_requirements/requirements_38_optional.txt

@@ -19,3 +19,4 @@ matplotlib==2.2.3
 scikit-image==0.18.1
 psutil==5.7.0
 kaleido
+vaex

packages/python/plotly/test_requirements/requirements_39_optional.txt

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