🍞 extend tests

Author: jvdd

tests/test_serialization.py

@@ -5,6 +5,7 @@
 import pickle
 import copy
 
+from plotly.subplots import make_subplots
 from plotly_resampler import FigureResampler, FigureWidgetResampler
 from plotly_resampler.registering import (
     register_plotly_resampler,
@@ -45,6 +46,42 @@ def test_pickle_figure_resampler(pickle_figure):
         assert len(hf_trace["y"]) == nb_samples
         assert np.all(hf_trace["y"] == np.arange(nb_samples))
 
+    # Test for figure with subplots (check non-pickled private properties)
+    fig = FigureResampler(
+        make_subplots(rows=2, cols=1, shared_xaxes=True),
+        default_n_shown_samples=50, show_dash_kwargs=dict(port=8051),
+    )
+    for i in range(nb_traces):
+        fig.add_trace(
+            go.Scattergl(name=f"trace--{i}"), hf_y=np.arange(nb_samples),
+            row=(i % 2) + 1, col=1,
+        )
+    assert fig._global_n_shown_samples == 50
+    assert fig._show_dash_kwargs["port"] == 8051
+    assert fig._figure_class == go.Figure
+    assert fig._xaxis_list == ['xaxis', 'xaxis2']
+    assert fig._yaxis_list == ['yaxis', 'yaxis2']
+
+    pickle.dump(fig, open(pickle_figure, "wb"))
+    fig_pickle = pickle.load(open(pickle_figure, "rb"))
+
+    assert isinstance(fig_pickle, FigureResampler)
+    assert fig_pickle._global_n_shown_samples == 50
+    assert fig_pickle._show_dash_kwargs["port"] == 8051
+    assert fig_pickle._figure_class == go.Figure
+    assert fig_pickle._xaxis_list == ['xaxis', 'xaxis2']
+    assert fig_pickle._yaxis_list == ['yaxis', 'yaxis2']
+    assert len(fig_pickle.data) == nb_traces
+    assert len(fig_pickle.hf_data) == nb_traces
+    for i in range(nb_traces):
+        trace = fig_pickle.data[i]
+        assert isinstance(trace, go.Scattergl)
+        assert len(trace.y) == 50
+        assert f"trace--{i}" in trace.name
+        hf_trace = fig_pickle.hf_data[i]
+        assert len(hf_trace["y"]) == nb_samples
+        assert np.all(hf_trace["y"] == np.arange(nb_samples))
+
 
 def test_pickle_figurewidget_resampler(pickle_figure):
     nb_traces = 2
@@ -69,6 +106,40 @@ def test_pickle_figurewidget_resampler(pickle_figure):
         assert len(hf_trace["y"]) == nb_samples
         assert np.all(hf_trace["y"] == np.arange(nb_samples))
 
+    # Test for figure with subplots (check non-pickled private properties)
+    fig = FigureWidgetResampler(
+        make_subplots(rows=2, cols=1, shared_xaxes=True),
+        default_n_shown_samples=50,
+    )
+    for i in range(nb_traces):
+        fig.add_trace(
+            go.Scattergl(name=f"trace--{i}"), hf_y=np.arange(nb_samples),
+            row=(i % 2) + 1, col=1,
+        )
+    assert fig._global_n_shown_samples == 50
+    assert fig._figure_class == go.FigureWidget
+    assert fig._xaxis_list == ['xaxis', 'xaxis2']
+    assert fig._yaxis_list == ['yaxis', 'yaxis2']
+
+    pickle.dump(fig, open(pickle_figure, "wb"))
+    fig_pickle = pickle.load(open(pickle_figure, "rb"))
+
+    assert isinstance(fig_pickle, FigureWidgetResampler)
+    assert fig_pickle._global_n_shown_samples == 50
+    assert fig_pickle._figure_class == go.FigureWidget
+    assert fig_pickle._xaxis_list == ['xaxis', 'xaxis2']
+    assert fig_pickle._yaxis_list == ['yaxis', 'yaxis2']
+    assert len(fig_pickle.data) == nb_traces
+    assert len(fig_pickle.hf_data) == nb_traces
+    for i in range(nb_traces):
+        trace = fig_pickle.data[i]
+        assert isinstance(trace, go.Scattergl)
+        assert len(trace.y) == 50
+        assert f"trace--{i}" in trace.name
+        hf_trace = fig_pickle.hf_data[i]
+        assert len(hf_trace["y"]) == nb_samples
+        assert np.all(hf_trace["y"] == np.arange(nb_samples))
+
 
 ## Test pickling when registered