[issue-307] add 2 extra (on left and right) dots out of zommed box

Author: alemikhov

plotly_resampler/aggregation/plotly_aggregator_parser.py

@@ -97,6 +97,8 @@ def get_start_end_indices(hf_trace_data, axis_type, start, end) -> Tuple[int, in
         # Search the index-positions
         start_idx = bisect.bisect_left(hf_trace_data["x"], start)
         end_idx = bisect.bisect_right(hf_trace_data["x"], end)
+        start_idx = max(0, start_idx - 1)
+        end_idx = min(end_idx + 1, len(hf_trace_data["x"]))
         return start_idx, end_idx
 
     @staticmethod

tests/test_figure_resampler.py

@@ -284,7 +284,7 @@ def test_log_axis():
         assert len(out) == 2
         assert (x1 - x0) < 10
         assert len(out[1]["x"]) == 1000
-        assert out[-1]["x"][0] >= 100
+        assert out[-1]["x"][0] >= 99
         assert out[-1]["x"][-1] <= 50_000
 
 
@@ -1061,8 +1061,8 @@ def test_time_tz_slicing():
         start_idx, end_idx = PlotlyAggregatorParser.get_start_end_indices(
             hf_data_dict, hf_data_dict["axis_type"], t_start, t_stop
         )
-        assert (s.index[start_idx] - t_start) <= pd.Timedelta(seconds=1)
-        assert (s.index[min(end_idx, n - 1)] - t_stop) <= pd.Timedelta(seconds=1)
+        assert (s.index[start_idx] - t_start) <= pd.Timedelta(seconds=2)
+        assert (s.index[min(end_idx, n - 1)] - t_stop) <= pd.Timedelta(seconds=2)
 
 
 def test_time_tz_slicing_different_timestamp():
@@ -1147,10 +1147,10 @@ def test_different_tz_no_tz_series_slicing():
         )
         assert (
             s.tz_localize(None).index[start_idx].tz_localize(t_start.tz) - t_start
-        ) <= pd.Timedelta(seconds=1)
+        ) <= pd.Timedelta(seconds=2)
         assert (
             s.tz_localize(None).index[end_idx].tz_localize(t_stop.tz) - t_stop
-        ) <= pd.Timedelta(seconds=1)
+        ) <= pd.Timedelta(seconds=2)
 
 
 def test_multiple_tz_no_tz_series_slicing():

tests/test_figurewidget_resampler.py

@@ -813,7 +813,7 @@ def test_hf_data_property_reload_data():
     assert np.all(fwr.hf_data[0]["y"] == new_y)
 
     fwr.reload_data()
-    assert (fwr.data[0]["x"][0] >= 10_000) & (fwr.data[0]["x"][-1] <= 20_000)
+    assert (fwr.data[0]["x"][0] >= 9_999) & (fwr.data[0]["x"][-1] <= 20_001)
     assert np.all(fwr.data[0]["y"] == new_y[fwr.data[0]["x"]])
     assert (fwr.layout["yaxis"].range[0] == -20) & (fwr.layout["yaxis"].range[-1] == 3)
 
@@ -888,8 +888,8 @@ def test_hf_data_property_subplots_reload_data():
     assert np.all(fwr.hf_data[0]["y"] == new_y)
 
     fwr.reload_data()
-    assert (fwr.data[0]["x"][0] >= 10_000) & (fwr.data[0]["x"][-1] <= 20_000)
-    assert (fwr.data[1]["x"][0] >= 40_000) & (fwr.data[1]["x"][-1] <= 60_000)
+    assert (fwr.data[0]["x"][0] >= 9_999) & (fwr.data[0]["x"][-1] <= 20_001)
+    assert (fwr.data[1]["x"][0] >= 39_999) & (fwr.data[1]["x"][-1] <= 60_001)
     assert np.all(fwr.data[0]["y"] == new_y[fwr.data[0]["x"]])
     assert np.all(fwr.data[1]["y"] == new_y[fwr.data[1]["x"]])
     assert (fwr.layout["yaxis"].range[0] == -20) & (fwr.layout["yaxis"].range[-1] == 3)
@@ -919,8 +919,8 @@ def test_hf_data_subplots_non_shared_xaxes():
     assert 0 <= x_0[0] <= (n / 1000)
     assert (n - 1000) <= x_0[-1] <= n - 1
     x_1 = fwr.data[1]["x"]
-    assert 40_000 <= x_1[0] <= 40_000 + (20_000 / 1000)
-    assert (60_000 - 20_000 / 1_000) <= x_1[-1] <= 60_000
+    assert 39_999 <= x_1[0] <= 40_001 + (20_000 / 1000)
+    assert (60_000 - 20_000 / 1_000) <= x_1[-1] <= 60_001
 
 
 def test_hf_data_subplots_non_shared_xaxes_row_col_none():
@@ -944,8 +944,8 @@ def test_hf_data_subplots_non_shared_xaxes_row_col_none():
     assert 0 <= x_0[0] <= (n / 1000)
     assert (n - 1000) <= x_0[-1] <= n - 1
     x_1 = fwr.data[1]["x"]
-    assert 40_000 <= x_1[0] <= 40_000 + (20_000 / 1000)
-    assert (60_000 - 20_000 / 1_000) <= x_1[-1] <= 60_000
+    assert 39_999 <= x_1[0] <= 40_001 + (20_000 / 1000)
+    assert (60_000 - 20_000 / 1_000) <= x_1[-1] <= 60_001
 
 
 def test_updates_two_traces():