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