示例#1
0
def test_get_pos_iterable():
    dsi = DatasetIndex(5)
    assert (dsi.get_pos(np.arange(5)) == np.arange(5)).all()
示例#2
0
def test_get_pos_slice():
    dsi = DatasetIndex(5)
    assert dsi.get_pos(slice(0, 4, 2)) == slice(0, 4, 2)
示例#3
0
def test_get_pos_str():
    dsi = DatasetIndex(['a', 'b', 'c', 'd', 'e'])
    assert dsi.get_pos('c') == 2
示例#4
0
def test_get_pos_int():
    dsi = DatasetIndex(5)
    assert dsi.get_pos(4) == 4