def METHOD_NAME(self):
    surrogate = KrigingSurrogate(nugget=0., eval_rmse=True)
    x = np.array([[0., 0., 0.], [1., 1., 1.]])
    y = np.array([[0.], [3.]])
    surrogate.train(x, y)
    for x0, y0 in zip(x, y):
        mu, sigma = surrogate.predict(x0)
        assert_near_equal(mu, [y0], 1e-9)
        assert_near_equal(sigma, [[0]], 1e-6)