Function test_numpy_reader_roi_empty_axes

dali/test/python/reader/test_numpy.py:820–862

(i, fortran_order, device, axes_or_range)

@params(*list(_get_roi_empty_axes_params()))
def test_numpy_reader_roi_empty_axes(i, fortran_order, device, axes_or_range):
    # setup file
    shapes = [(10, 10), (12, 10), (10, 12), (20, 15), (10, 11), (12, 11), (13, 11), (19, 10)]
    ndim = 2
    dtype = np.uint8
    batch_size = 8
    file_filter = "*.npy"
    rng = np.random.default_rng(4242 + i)

    with tempfile.TemporaryDirectory(prefix=gds_data_root) as test_data_root:
        index = 0
        for sh in shapes:
            filename = os.path.join(test_data_root, "test_{:02d}.npy".format(index))
            index += 1
            if fortran_order is not None:
                actual_fortran_order = fortran_order
            else:
                actual_fortran_order = rng.choice([False, True])
            create_numpy_file(filename, sh, dtype, actual_fortran_order)

        if axes_or_range == "axes":
            _testimpl_numpy_reader_roi_empty_axes(
                "empty axes",
                test_data_root,
                batch_size,
                ndim,
                dtype,
                device,
                fortran_order,
                file_filter,
            )
        else:
            assert axes_or_range == "range"
            _testimpl_numpy_reader_roi_empty_range(
                "empty range",
                test_data_root,
                batch_size,
                ndim,
                dtype,
                device,
                fortran_order,
                file_filter,
            )


def _testimpl_numpy_reader_roi_error(