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