MDP-568 Fix index out of bounds error

Can't slice beyond the end of the dts/index array

Author: jamesblackburn

arctic/store/_pandas_ndarray_store.py

@@ -171,8 +171,8 @@ def _index_range(self, version, symbol, date_range=None, **kwargs):
                 start, end = _start_end(date_range, dts)
                 if start > dts[-1]:
                     return -1, -1
-                idxstart = min(np.searchsorted(dts, start), len(dts))
-                idxend = min(np.searchsorted(dts, end), len(dts))
+                idxstart = min(np.searchsorted(dts, start), len(dts) - 1)
+                idxend = min(np.searchsorted(dts, end), len(dts) - 1)
                 return index['index'][idxstart], index['index'][idxend] + 1
         return super(PandasStore, self)._index_range(version, symbol, **kwargs)
 

tests/integration/store/test_pandas_store.py

@@ -663,6 +663,9 @@ def test_daterange_large_DataFrame(library):
     # last row
     result = library.read('MYARR', date_range=DateRange(df.index[-1], df.index[-1])).data
     assert_frame_equal(df[df.index[-1]:df.index[-1]], result, check_names=False)
+    # beyond last row
+    result = library.read('MYARR', date_range=DateRange(df.index[-1], df.index[-1] + dtd(days=1))).data
+    assert_frame_equal(df[df.index[-1]:df.index[-1]], result, check_names=False)
     # somewhere in time
     result = library.read('MYARR', date_range=DateRange(dt(2020, 1, 1), dt(2031, 9, 1))).data
     assert_frame_equal(df[dt(2020, 1, 1):dt(2031, 9, 1)], result, check_names=False)