def test_vector_derivs(self): surrogate = ResponseSurface() x = array([[a, b] for a, b in itertools.product(linspace(0, 1, 10), repeat=2)]) y = array([[a + b, a - b] for a, b in x]) surrogate.train(x, y) jac = surrogate.jacobian(array([[0.5, 0.5]])) assert_rel_error(self, jac, array([[1, 1], [1, -1]]), 1e-5)
def test_scalar_derivs(self): surrogate = ResponseSurface() x = array([[0.], [1.], [2.], [3.]]) y = x.copy() surrogate.train(x, y) jac = surrogate.jacobian(array([[0.]])) assert_rel_error(self, jac[0][0], 1., 1e-3)