Exemplo n.º 1
0
    def test_get_uncertain_value(self):
        x = array([[0.05], [.25], [0.61], [0.95]])
        y = array([0.738513784857542, -0.210367746201974, -0.489015457891476, 12.3033138316612])
        krig1 = KrigingSurrogate()
        krig1.train(x, y)

        self.assertEqual(krig1.get_uncertain_value(1).mu, NormalDistribution(1., 0.).mu)
        self.assertEqual(krig1.get_uncertain_value(1).sigma, NormalDistribution(1., 0.).sigma)