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