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