def test_sanity_checks():
assert (bm.log(4.1) == np.log(4.1))
assert (bm.exp(-10.1) == np.exp(-10.1))
assert (bm.sqrt(112.3) == np.sqrt(112.3))
assert (bm.sin(4.1) == np.sin(4.1))
assert (bm.cos(2.2) == np.cos(2.2))
assert (bm.tan(2) == np.tan(2.0))
assert (bm.arcsin(0) == np.arcsin(0))
assert (bm.arccos(0.2) == np.arccos(0.2))
assert (bm.arctan(2) == np.arctan(2))
assert (bm.sinh(2) == np.sinh(2))
assert (bm.cosh(2) == np.cosh(2))
assert (bm.tanh(2) == np.tanh(2))
assert (bm.logk(2, np.exp(1)) == np.log(2))
def sqrt_function(x, y):
return bm.sqrt(x) + bm.sqrt(y)
def sqrt_scalar(x):
return 10 + bm.sqrt(2 * x + 1)
def sqrt_fn(x, y):
return x * bm.sqrt(x * y) + x * x * x