Function check_arithm_mod

dali/test/python/operator_1/test_arithmetic_ops.py:907–939

(kinds, types, shape)

def check_arithm_mod(kinds, types, shape):
    # DALI uses C-style truncated remainder (same sign as the dividend), matching np.fmod.
    # bool is excluded: bool % bool = 0 but disallow_zeros makes it tricky with bool generators.
    left_type, right_type = types
    target_type = bin_promote(left_type, right_type)
    # Avoid zeros in the right operand.
    iterator = iter(
        ExternalInputIterator(
            batch_size,
            shape,
            types,
            kinds,
            disallow_zeros=(False, True),
            limited_range=[(-1000, 1000), (-100, 100)],
        )
    )
    pipe = ExprOpPipeline(
        kinds,
        types,
        iterator,
        (lambda x, y: x % y),
        batch_size=batch_size,
        num_threads=2,
        device_id=0,
    )
    pipe_out = pipe.run()
    for sample in range(batch_size):
        l_np, r_np, out = extract_data(pipe_out, sample, kinds, target_type)
        assert_equals(out.dtype, target_type)

        assert np.array_equal(
            out, np.fmod(l_np.astype(np.float64), r_np.astype(np.float64)).astype(target_type)
        )


@cartesian_params(selected_bin_input_kinds, iter_pow(selected_integer_types, 2))