add equality test, multiple frames and numpy arrays

FBruzzesi · FBruzzesi · commit 9c1d326c739f · 2024-12-06T21:58:53.000+01:00
diff --git a/packages/python/plotly/_plotly_utils/tests/validators/test_fig_deepcopy.py b/packages/python/plotly/_plotly_utils/tests/validators/test_fig_deepcopy.py
@@ -1,11 +1,38 @@
 import copy
+import pytest
 import plotly.express as px
 
+"""
+This test is in the validators folder since copy.deepcopy ends up calling
+BaseFigure(*args) which hits `validate_coerce`.
 
-def test_deepcopy():
-    gapminder = px.data.gapminder()
+When inputs are dataframes and arrays, then the copied figure is called with
+base64 encoded arrays.
+"""
+
+
+@pytest.mark.parametrize("return_type", ["pandas", "polars", "pyarrow"])
+@pytest.mark.filterwarnings(
+    r"ignore:\*scattermapbox\* is deprecated! Use \*scattermap\* instead"
+)
+def test_deepcopy_dataframe(return_type):
+    gapminder = px.data.gapminder(return_type=return_type)
     fig = px.line(gapminder, x="year", y="gdpPercap", color="country")
+    fig_copied = copy.deepcopy(fig)
+
+    assert fig_copied.to_dict() == fig.to_dict()
+
+
+@pytest.mark.filterwarnings(
+    r"ignore:\*scattermapbox\* is deprecated! Use \*scattermap\* instead"
+)
+def test_deepcopy_array():
+    gapminder = px.data.gapminder()
+    x = gapminder["year"].to_numpy()
+    y = gapminder["gdpPercap"].to_numpy()
+    color = gapminder["country"].to_numpy()
 
-    fig_copy = copy.deepcopy(fig)
+    fig = px.line(x=x, y=y, color=color)
+    fig_copied = copy.deepcopy(fig)
 
-    assert fig_copy is not None
+    assert fig_copied.to_dict() == fig.to_dict()
