Function _testimpl_numpy_reader_roi_empty_axes

dali/test/python/reader/test_numpy.py:609–646

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

def _testimpl_numpy_reader_roi_empty_axes(
    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=[],
            roi_end=[],
            roi_axes=[],
            shard_id=0,
            num_shards=1,
            cache_header_information=False,
            seed=1234,
        )
        return data0, data1

    p = pipe()
    try:
        data0, data1 = p.run()
    finally:
        del p
    for i in range(batch_size):
        arr = to_array(data0[i])
        roi_arr = to_array(data1[i])
        assert_array_equal(arr, roi_arr)


def _testimpl_numpy_reader_roi_empty_range(