Add JSON test for serializing customdata as numpy array

s-celles · Jul 8, 2021 · dba2bee · dba2bee
1 parent 5453155
commit dba2bee
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/packages/python/plotly/plotly/tests/test_io/test_to_from_plotly_json.py b/packages/python/plotly/plotly/tests/test_io/test_to_from_plotly_json.py
@@ -1,6 +1,7 @@
 import pytest
 import plotly.io.json as pio
 import plotly.graph_objects as go
+import plotly.express as px
 import numpy as np
 import pandas as pd
 import json
@@ -202,3 +203,9 @@ def to_str(v):
     expected = build_test_dict_string(array_str)
     assert result == expected
     check_roundtrip(result, engine=engine, pretty=pretty)
+
+
+def test_object_array(engine, pretty):
+    fig = px.scatter(px.data.tips(), x="total_bill", y="tip", custom_data=["sex"])
+    result = fig.to_plotly_json()
+    check_roundtrip(result, engine=engine, pretty=pretty)