def test_warn(): rl = RangedList(value=range(5)) assert [0, 1, 2] == rl.selector_to_ids([True, True, True, False], warn=True) assert [0, 1, 2] == rl.selector_to_ids([True, True, True, False, False, False], warn=True) assert [0, 1, 2, 3, 4] == list(rl.selector_to_ids(None, warn=True))
def test_selector_to_ids(): rl = RangedList(value=range(5)) assert [0, 1, 2, 3, 4] == list(rl.selector_to_ids(None)) assert [0, 1] == list(rl.selector_to_ids(slice(2))) assert [2] == list(rl.selector_to_ids(2)) with pytest.raises(TypeError): rl.selector_to_ids(-7) with pytest.raises(TypeError): rl.selector_to_ids(7)
def test_numpy_selector(): rl = RangedList(value=range(5)) selector = numpy.array([1, 3, 4]) assert [1, 3, 4] == rl.selector_to_ids(selector)