def test_cosh(): fun = lambda x : 3.0 * np.cosh(x) d_fun = grad(fun) check_grads(fun, npr.randn()) check_grads(d_fun, npr.randn())