Skip to content

Commit 6abf757

Browse files
committed
🌬️ formatting + extend tests
1 parent fd8c2f6 commit 6abf757

File tree

2 files changed

+24
-14
lines changed

2 files changed

+24
-14
lines changed

tests/test_registering.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -127,9 +127,12 @@ def test_registering_plotly_express_and_kwargs(registering_cleanup):
127127
assert len(fig.data) == 1
128128
assert len(fig.data[0].y) == 500
129129

130-
register_plotly_resampler(default_n_shown_samples=50)
130+
register_plotly_resampler(
131+
default_n_shown_samples=50, show_dash_kwargs=dict(mode="inline", port=8051)
132+
)
131133
fig = px.scatter(y=np.arange(500))
132134
assert isinstance(fig, FigureResampler)
135+
assert fig._show_dash_kwargs == dict(mode="inline", port=8051)
133136
assert len(fig.data) == 1
134137
assert len(fig.data[0].y) == 50
135138
assert len(fig.hf_data) == 1
@@ -138,6 +141,7 @@ def test_registering_plotly_express_and_kwargs(registering_cleanup):
138141
register_plotly_resampler()
139142
fig = px.scatter(y=np.arange(5000))
140143
assert isinstance(fig, FigureResampler)
144+
assert fig._show_dash_kwargs == dict()
141145
assert len(fig.data) == 1
142146
assert len(fig.data[0].y) == 1000
143147
assert len(fig.hf_data) == 1
@@ -201,4 +205,3 @@ def test_compasibility_when_registered(registering_cleanup):
201205
assert len(f.data[0].y) == 1000
202206
assert len(f.hf_data) == 1
203207
assert len(f.hf_data[0]["y"]) == 1005
204-

tests/test_serialization.py

Lines changed: 19 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,10 @@ def test_pickle_figure_resampler_registered(registering_cleanup, pickle_figure):
7676
nb_traces = 4
7777
nb_samples = 5_043
7878

79-
register_plotly_resampler(mode="figure", default_n_shown_samples=50, show_dash_kwargs=dict(port=8051))
80-
79+
register_plotly_resampler(
80+
mode="figure", default_n_shown_samples=50, show_dash_kwargs=dict(port=8051)
81+
)
82+
8183
fig = go.Figure()
8284
for i in range(nb_traces):
8385
fig.add_trace(go.Scattergl(name=f"trace--{i}"), hf_y=np.arange(nb_samples))
@@ -163,7 +165,7 @@ def test_pickle_figurewidget_resampler_registered(registering_cleanup, pickle_fi
163165
nb_samples = 3_643
164166

165167
register_plotly_resampler(mode="widget", default_n_shown_samples=50)
166-
168+
167169
fig = go.Figure()
168170
for i in range(nb_traces):
169171
fig.add_trace(go.Scattergl(name=f"trace--{i}"), hf_y=np.arange(nb_samples))
@@ -281,7 +283,7 @@ def test_copy_and_deepcopy_figure_resampler():
281283
hf_trace = fig_copy.hf_data[i]
282284
assert len(hf_trace["y"]) == nb_samples
283285
assert np.all(hf_trace["y"] == np.arange(nb_samples))
284-
286+
285287

286288
def test_copy_and_deepcopy_figurewidget_resampler():
287289
nb_traces = 3
@@ -318,15 +320,18 @@ def test_copy_and_deepcopy_figurewidget_resampler():
318320
hf_trace = fig_copy.hf_data[i]
319321
assert len(hf_trace["y"]) == nb_samples
320322
assert np.all(hf_trace["y"] == np.arange(nb_samples))
321-
322-
## Test basic (deep)copy with PR registered
323+
324+
325+
## Test basic (deep)copy with PR registered
323326

324327
def test_copy_figure_resampler_registered():
325328
nb_traces = 3
326329
nb_samples = 4_069
327330

328-
register_plotly_resampler(mode="figure", default_n_shown_samples=50, show_dash_kwargs=dict(port=8051))
329-
331+
register_plotly_resampler(
332+
mode="figure", default_n_shown_samples=50, show_dash_kwargs=dict(port=8051)
333+
)
334+
330335
fig = go.Figure()
331336
for i in range(nb_traces):
332337
fig.add_trace(go.Scattergl(name=f"trace--{i}"), hf_y=np.arange(nb_samples))
@@ -391,8 +396,10 @@ def test_deepcopy_figure_resampler_registered():
391396
nb_traces = 4
392397
nb_samples = 3_169
393398

394-
register_plotly_resampler(mode="figure", default_n_shown_samples=50, show_dash_kwargs=dict(port=8051))
395-
399+
register_plotly_resampler(
400+
mode="figure", default_n_shown_samples=50, show_dash_kwargs=dict(port=8051)
401+
)
402+
396403
fig = go.Figure()
397404
for i in range(nb_traces):
398405
fig.add_trace(go.Scattergl(name=f"trace--{i}"), hf_y=np.arange(nb_samples))
@@ -458,7 +465,7 @@ def test_copy_figurewidget_resampler_registered():
458465
nb_samples = 3_012
459466

460467
register_plotly_resampler(mode="widget", default_n_shown_samples=50)
461-
468+
462469
fig = go.Figure()
463470
for i in range(nb_traces):
464471
fig.add_trace(go.Scattergl(name=f"trace--{i}"), hf_y=np.arange(nb_samples))
@@ -520,7 +527,7 @@ def test_deepcopy_figurewidget_resampler_registered():
520527
nb_samples = 3_012
521528

522529
register_plotly_resampler(mode="widget", default_n_shown_samples=50)
523-
530+
524531
fig = go.Figure()
525532
for i in range(nb_traces):
526533
fig.add_trace(go.Scattergl(name=f"trace--{i}"), hf_y=np.arange(nb_samples))

0 commit comments

Comments
 (0)