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.0], [1.0], [2.0], [3.0]]) y = x.copy() surrogate.train(x, y) jac = surrogate.jacobian(array([[0.0]])) assert_rel_error(self, jac[0][0], 1.0, 1e-3)