Function test_incorrect_param

dali/test/python/operator_2/test_random_beta.py:207–215

(a, b, dtype, msg)

    (1, np.full((2,), 1), types.DALIDataType.FLOAT, "cannot be converted"),
)
def test_incorrect_param(a, b, dtype, msg):

    @pipeline_def(batch_size=2, device_id=0, num_threads=4, seed=111)
    def pipeline():
        return fn.random.beta(alpha=a, beta=b, dtype=dtype)

    with assert_raises(RuntimeError, glob=msg):
        p = pipeline()
        p.run()