Function _test_out_of_range

dali/test/python/operator_2/test_subscript.py:225–230

(device, idx)

def _test_out_of_range(device, idx):
    data = [np.uint8([1, 2, 3]), np.uint8([1, 2])]
    src = fn.external_source(lambda: data, device=device)
    pipe = index_pipe(src, lambda x: x[idx])
    with assert_raises(RuntimeError, glob="out of range"):
        _ = pipe.run()


def test_out_of_range():