Function check_operator_erase_vs_python

dali/test/python/operator_1/test_erase.py:191–232

(
    device, batch_size, input_shape, anchor, shape, axis_names, axes, input_layout, fill_value
)

def check_operator_erase_vs_python(
    device, batch_size, input_shape, anchor, shape, axis_names, axes, input_layout, fill_value
):
    eii1 = RandomDataIterator(batch_size, shape=input_shape, dtype=np.float32)
    eii2 = RandomDataIterator(batch_size, shape=input_shape, dtype=np.float32)

    fill_value_arg1 = fill_value
    fill_value_arg2 = fill_value
    if fill_value == "random":
        fill_eii1 = RandomDataIterator(batch_size, shape=input_shape[-1:], dtype=np.float32)
        fill_eii2 = RandomDataIterator(batch_size, shape=input_shape[-1:], dtype=np.float32)
        fill_value_arg1 = iter(fill_eii1)
        fill_value_arg2 = iter(fill_eii2)

    compare_pipelines(
        ErasePipeline(
            device,
            batch_size,
            input_layout,
            iter(eii1),
            anchor=anchor,
            shape=shape,
            axis_names=axis_names,
            axes=axes,
            fill_value=fill_value_arg1,
        ),
        ErasePythonPipeline(
            device,
            batch_size,
            input_layout,
            iter(eii2),
            anchor=anchor,
            shape=shape,
            axis_names=axis_names,
            axes=axes,
            fill_value=fill_value_arg2,
        ),
        batch_size=batch_size,
        N_iterations=3,
        eps=1e-04,
        expected_layout=input_layout,
    )


def test_operator_erase_vs_python():