|
| 1 | +import numpy as np |
| 2 | +import plotly.graph_objects as go |
| 3 | +import pytest |
| 4 | +from plotly.subplots import make_subplots |
| 5 | + |
| 6 | +from plotly_resampler import FigureResampler, FigureWidgetResampler |
| 7 | +from plotly_resampler.aggregation import MinMaxLTTB |
| 8 | + |
| 9 | + |
| 10 | +@pytest.mark.parametrize("fig_type", [FigureResampler, FigureWidgetResampler]) |
| 11 | +def test_multiple_axes_figure(fig_type): |
| 12 | + # Generate some data |
| 13 | + x = np.arange(200_000) |
| 14 | + sin = 3 + np.sin(x / 200) + np.random.randn(len(x)) / 30 |
| 15 | + |
| 16 | + fig = fig_type( |
| 17 | + default_n_shown_samples=2000, default_downsampler=MinMaxLTTB(parallel=True) |
| 18 | + ) |
| 19 | + |
| 20 | + # all traces will be plotted against the same x-axis |
| 21 | + # note: the first added trace its yaxis will be used as reference |
| 22 | + fig.add_trace(go.Scatter(name="orig", yaxis="y1", line_width=1), hf_x=x, hf_y=sin) |
| 23 | + fig.add_trace( |
| 24 | + go.Scatter(name="negative", yaxis="y2", line_width=1), hf_x=x, hf_y=-sin |
| 25 | + ) |
| 26 | + fig.add_trace( |
| 27 | + go.Scatter(name="sqrt(orig)", yaxis="y3", line_width=1), |
| 28 | + hf_x=x, |
| 29 | + hf_y=np.sqrt(sin * 10), |
| 30 | + ) |
| 31 | + fig.add_trace( |
| 32 | + go.Scatter(name="orig**2", yaxis="y4", line_width=1), |
| 33 | + hf_x=x, |
| 34 | + hf_y=(sin - 3) ** 2, |
| 35 | + ) |
| 36 | + |
| 37 | + # in order for autoshift to work, you need to set x-anchor to free |
| 38 | + fig.update_layout( |
| 39 | + # NOTE: you can use the domain key to set the x-axis range (if you want to display) |
| 40 | + # the legend on the right instead of the top as done here |
| 41 | + xaxis=dict(domain=[0, 1]), |
| 42 | + # Add a title to the y-axis |
| 43 | + yaxis=dict(title="orig"), |
| 44 | + # by setting anchor=free, overlaying, and autoshift, the axis will be placed |
| 45 | + # automatically, without overlapping any other axes |
| 46 | + yaxis2=dict( |
| 47 | + title="negative", |
| 48 | + anchor="free", |
| 49 | + overlaying="y1", |
| 50 | + side="left", |
| 51 | + autoshift=True, |
| 52 | + ), |
| 53 | + yaxis3=dict( |
| 54 | + title="sqrt(orig)", |
| 55 | + anchor="free", |
| 56 | + overlaying="y1", |
| 57 | + side="right", |
| 58 | + autoshift=True, |
| 59 | + ), |
| 60 | + yaxis4=dict( |
| 61 | + title="orig ** 2", |
| 62 | + anchor="free", |
| 63 | + overlaying="y1", |
| 64 | + side="right", |
| 65 | + autoshift=True, |
| 66 | + ), |
| 67 | + ) |
| 68 | + |
| 69 | + # Update layout properties |
| 70 | + fig.update_layout( |
| 71 | + title_text="multiple y-axes example", |
| 72 | + height=600, |
| 73 | + legend=dict( |
| 74 | + orientation="h", |
| 75 | + yanchor="bottom", |
| 76 | + y=1.02, |
| 77 | + xanchor="right", |
| 78 | + x=1, |
| 79 | + ), |
| 80 | + template="plotly_white", |
| 81 | + ) |
| 82 | + |
| 83 | + # Test: check whether a single update triggers all traces to be updated |
| 84 | + out = fig.construct_update_data({"xaxis.range[0]": 0, "xaxis.range[1]": 50_000}) |
| 85 | + assert len(out) == 5 |
| 86 | + # fig.show_dash |
| 87 | + |
| 88 | + |
| 89 | +@pytest.mark.parametrize("fig_type", [FigureResampler, FigureWidgetResampler]) |
| 90 | +def test_multiple_axes_subplot_rows(fig_type): |
| 91 | + # Generate some data |
| 92 | + x = np.arange(200_000) |
| 93 | + sin = 3 + np.sin(x / 200) + np.random.randn(len(x)) / 30 |
| 94 | + |
| 95 | + # create a figure with 2 rows and 1 column |
| 96 | + # NOTE: instead of the above methods, we don't add the "yaxis" argument to the |
| 97 | + # scatter object |
| 98 | + fig = fig_type(make_subplots(rows=2, cols=1, shared_xaxes=True)) |
| 99 | + fig.add_trace(go.Scatter(name="orig"), hf_x=x, hf_y=sin, row=2, col=1) |
| 100 | + fig.add_trace(go.Scatter(name="-orig"), hf_x=x, hf_y=-sin, row=2, col=1) |
| 101 | + fig.add_trace(go.Scatter(name="sqrt"), hf_x=x, hf_y=np.sqrt(sin * 10), row=2, col=1) |
| 102 | + fig.add_trace(go.Scatter(name="orig**2"), hf_x=x, hf_y=(sin - 3) ** 2, row=2, col=1) |
| 103 | + |
| 104 | + # NOTE: because of the row and col specification, the yaxis is automatically set to y2 |
| 105 | + for i, data in enumerate(fig.data[1:], 3): |
| 106 | + data.update(yaxis=f"y{i}") |
| 107 | + |
| 108 | + # add the original signal to the first row subplot |
| 109 | + fig.add_trace(go.Scatter(name="<b>orig</b>"), row=1, col=1, hf_x=x, hf_y=sin) |
| 110 | + |
| 111 | + # in order for autoshift to work, you need to set x-anchor to free |
| 112 | + fig.update_layout( |
| 113 | + xaxis2=dict(domain=[0, 1], anchor="y2"), |
| 114 | + yaxis2=dict(title="orig"), |
| 115 | + yaxis3=dict( |
| 116 | + title="-orig", |
| 117 | + anchor="free", |
| 118 | + overlaying="y2", |
| 119 | + side="left", |
| 120 | + autoshift=True, |
| 121 | + ), |
| 122 | + yaxis4=dict( |
| 123 | + title="sqrt(orig)", |
| 124 | + anchor="free", |
| 125 | + overlaying="y2", |
| 126 | + side="right", |
| 127 | + autoshift=True, |
| 128 | + ), |
| 129 | + yaxis5=dict( |
| 130 | + title="orig ** 2", |
| 131 | + anchor="free", |
| 132 | + overlaying="y2", |
| 133 | + side="right", |
| 134 | + autoshift=True, |
| 135 | + ), |
| 136 | + ) |
| 137 | + |
| 138 | + # Update layout properties |
| 139 | + fig.update_layout( |
| 140 | + title_text="multiple y-axes example", |
| 141 | + height=800, |
| 142 | + legend=dict( |
| 143 | + orientation="h", |
| 144 | + yanchor="bottom", |
| 145 | + y=1.02, |
| 146 | + xanchor="right", |
| 147 | + x=1, |
| 148 | + ), |
| 149 | + template="plotly_white", |
| 150 | + ) |
| 151 | + |
| 152 | + # Test: check whether a single update triggers all traces to be updated |
| 153 | + out = fig.construct_update_data( |
| 154 | + { |
| 155 | + "xaxis.range[0]": 0, |
| 156 | + "xaxis.range[1]": 50_000, |
| 157 | + "xaxis2.range[0]": 0, |
| 158 | + "xaxis2.range[1]": 50_000, |
| 159 | + } |
| 160 | + ) |
| 161 | + assert len(out) == 6 |
| 162 | + |
| 163 | + |
| 164 | +@pytest.mark.parametrize("fig_type", [FigureResampler, FigureWidgetResampler]) |
| 165 | +def test_multiple_axes_subplot_cols(fig_type): |
| 166 | + x = np.arange(200_000) |
| 167 | + sin = 3 + np.sin(x / 200) + np.random.randn(len(x)) / 30 |
| 168 | + |
| 169 | + # Create a figure with 1 row and 2 columns |
| 170 | + fig = fig_type(make_subplots(rows=1, cols=2)) |
| 171 | + fig.add_trace(go.Scatter(name="orig"), hf_x=x, hf_y=sin, row=1, col=2) |
| 172 | + fig.add_trace(go.Scatter(name="-orig"), hf_x=x, hf_y=-sin, row=1, col=2) |
| 173 | + fig.add_trace(go.Scatter(name="sqrt"), hf_x=x, hf_y=np.sqrt(sin * 10), row=1, col=2) |
| 174 | + fig.add_trace(go.Scatter(name="orig**2"), hf_x=x, hf_y=(sin - 3) ** 2, row=1, col=2) |
| 175 | + |
| 176 | + # NOTE: because of the row & col specification, the yaxis is automatically set to y2 |
| 177 | + for i, data in enumerate(fig.data[1:], 3): |
| 178 | + data.update(yaxis=f"y{i}") |
| 179 | + |
| 180 | + fig.add_trace(go.Scatter(name="<b>orig</b>"), row=1, col=1, hf_x=x, hf_y=sin) |
| 181 | + |
| 182 | + # In order for autoshift to work, you need to set x-anchor to free |
| 183 | + fig.update_layout( |
| 184 | + xaxis=dict(domain=[0, 0.4]), |
| 185 | + xaxis2=dict(domain=[0.56, 1]), |
| 186 | + yaxis2=dict(title="orig"), |
| 187 | + yaxis3=dict( |
| 188 | + title="-orig", |
| 189 | + anchor="free", |
| 190 | + overlaying="y2", |
| 191 | + side="left", |
| 192 | + autoshift=True, |
| 193 | + ), |
| 194 | + yaxis4=dict( |
| 195 | + title="sqrt(orig)", |
| 196 | + anchor="free", |
| 197 | + overlaying="y2", |
| 198 | + side="right", |
| 199 | + autoshift=True, |
| 200 | + ), |
| 201 | + yaxis5=dict( |
| 202 | + title="orig ** 2", |
| 203 | + anchor="free", |
| 204 | + overlaying="y2", |
| 205 | + side="right", |
| 206 | + autoshift=True, |
| 207 | + ), |
| 208 | + ) |
| 209 | + |
| 210 | + # Update layout properties |
| 211 | + fig.update_layout( |
| 212 | + title_text="multiple y-axes example", |
| 213 | + height=300, |
| 214 | + legend=dict( |
| 215 | + orientation="h", |
| 216 | + yanchor="bottom", |
| 217 | + y=1.02, |
| 218 | + xanchor="right", |
| 219 | + x=1, |
| 220 | + ), |
| 221 | + template="plotly_white", |
| 222 | + ) |
| 223 | + |
| 224 | + out = fig.construct_update_data( |
| 225 | + { |
| 226 | + "xaxis.range[0]": 0, |
| 227 | + "xaxis.range[1]": 50_000, |
| 228 | + } |
| 229 | + ) |
| 230 | + assert len(out) == 2 |
| 231 | + |
| 232 | + out = fig.construct_update_data( |
| 233 | + { |
| 234 | + "xaxis2.range[0]": 0, |
| 235 | + "xaxis2.range[1]": 50_000, |
| 236 | + } |
| 237 | + ) |
| 238 | + assert len(out) == 5 |
0 commit comments