Method test_sqrt_abs_sin

tests/python/test_kernel.py:76–82

(self)

class TestMathFunctions:
    def test_sqrt_abs_sin(self) -> None:
        x = kernel.arg("x")
        transform = kernel.elementwise(kernel.sqrt(kernel.abs(x)) + kernel.sin(x))
        bx = rt.asarray(np.array([4.0, 9.0]))
        out = rt.go(kernel.map(transform, x=bx)).result().numpy()
        assert abs(out[0] - (math.sqrt(4.0) + math.sin(4.0))) < 1e-10
        assert abs(out[1] - (math.sqrt(9.0) + math.sin(9.0))) < 1e-10

    def test_log_exp(self) -> None:
        x = kernel.arg("x")