示例#1
0
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
示例#2
0
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