def test_vector_derivs(self):
surrogate = KrigingSurrogate()
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 = KrigingSurrogate()
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)
