Function _testimpl_numpy_reader_roi_empty_range

dali/test/python/reader/test_numpy.py:649–690

(
    testcase_name, file_root, batch_size, ndim, dtype, device, fortran_order, file_filter="*.npy"
)

def _testimpl_numpy_reader_roi_empty_range(
    testcase_name, file_root, batch_size, ndim, dtype, device, fortran_order, file_filter="*.npy"
):
    # testcase name used for visibility in the output logs
    @pipeline_def(batch_size=batch_size, device_id=0, num_threads=8)
    def pipe():
        data0 = fn.readers.numpy(
            device=device,
            file_root=file_root,
            file_filter=file_filter,
            shard_id=0,
            num_shards=1,
            cache_header_information=False,
            seed=1234,
        )
        data1 = fn.readers.numpy(
            device=device,
            file_root=file_root,
            file_filter=file_filter,
            roi_start=[1],
            roi_end=[1],
            roi_axes=[1],
            shard_id=0,
            num_shards=1,
            cache_header_information=False,
            seed=1234,
        )
        return data0, data1

    p = pipe()
    try:
        data0, data1 = p.run()
        for i in range(batch_size):
            arr = to_array(data0[i])
            roi_arr = to_array(data1[i])
            for d in range(len(arr.shape)):
                if d == 1:
                    assert roi_arr.shape[d] == 0
                else:
                    assert roi_arr.shape[d] == arr.shape[d]
    finally:
        del p


# roi_start, rel_roi_start, roi_end, rel_roi_end, roi_shape,