Function check_unary_op

dali/test/python/operator_1/test_arithmetic_ops.py:494–506

(kind, type, op, shape, _)

def check_unary_op(kind, type, op, shape, _):
    # Regular arithmetic ops that can be validated as straight numpy
    iterator = iter(ExternalInputIterator(batch_size, shape, type, kind))
    pipe = ExprOpPipeline(
        kind, type, iterator, op, batch_size=batch_size, num_threads=2, device_id=0
    )
    pipe_out = pipe.run()
    for sample in range(batch_size):
        in_np, out = extract_un_data(pipe_out, sample, kind, type)
        if "f" in np.dtype(type).kind:
            np.testing.assert_allclose(out, op(in_np), rtol=1e-07 if type != np.float16 else 0.005)
        else:
            assert np.array_equal(out, op(in_np))


def test_unary_arithmetic_ops():