Function _tests

dali/test/python/operator_2/test_resize.py:592–631

(dim, backend)

def _tests(dim, backend):
    batch_size = 2 if dim == 3 else 10
    # - Cannot test linear against PIL, because PIL uses triangular filter when downscaling
    # - Cannot test Nearest Neighbor because rounding errors cause gross discrepancies (pixel shift)
    for mode in ["default", "stretch", "not_smaller", "not_larger"]:
        for (
            interp,
            dtype,
            channel_first,
            use_size_arg,
            use_size_input,
            w_input,
            h_input,
            d_input,
            use_roi,
        ) in [
            (0, types.UINT8, True, False, False, False, False, False, False),
            (1, types.FLOAT, False, False, False, False, True, True, True),
            (0, types.FLOAT, True, False, False, True, True, False, True),
            (1, types.FLOAT, False, False, False, True, False, True, False),
            (0, types.UINT8, True, True, False, False, False, False, True),
            (1, types.UINT8, False, True, True, False, False, False, False),
        ]:
            interp = [types.INTERP_TRIANGULAR, types.INTERP_LANCZOS3][interp]
            yield (
                _test_ND,
                backend,
                dim,
                batch_size,
                channel_first,
                mode,
                interp,
                dtype,
                w_input,
                h_input,
                d_input,
                use_size_arg,
                use_size_input,
                use_roi,
            )


def test_2D_gpu():