def test_rangeindex_get_slice_bound(bounds, indices, side, kind): start, stop = bounds pd_index = pd.RangeIndex(start, stop) cudf_index = RangeIndex(start, stop) for idx in indices: expect = pd_index.get_slice_bound(idx, side, kind) got = cudf_index.get_slice_bound(idx, side, kind) assert expect == got
def test_rangeindex_get_slice_bound_step(bounds, label, side, kind): start, stop, step = bounds pd_index = pd.RangeIndex(start, stop, step) cudf_index = RangeIndex(start, stop, step) expect = pd_index.get_slice_bound(label, side, kind) got = cudf_index.get_slice_bound(label, side, kind) assert expect == got