def test_map(self):
g = Gaussian(mean=Gaussian(np.zeros(2), np.ones(2)), var=np.ones(2))
g.map(np.array([[1., 0.], [0., 1.], [-1., 0.], [0., -1.]]))
self.assertTrue((g.mean == 0).all())
g1 = Gaussian(mean=Gaussian(np.zeros(2), np.ones(2)), var=np.ones(2))
g1.bayes(np.array([[1., 0.], [0., 1.], [-1., 0.], [0., -1.]]))
self.assertTrue((g.mean == g1.mean.mean).all())
def test_bayes(self):
g = Gaussian(mean=Gaussian(np.zeros(2), np.ones(2)), var=np.ones(2))
g.bayes(np.array([[1., 0.], [0., 1.], [-1., 0.], [0., -1.]]))
self.assertTrue((g.mean.var == np.ones(2) * 0.2).all())
