def test_prob(self):
np.random.seed(992)
mean, sd = np.random.rand(2)
g = Gaussian(mean, sd)
norm = 1 / np.sqrt(2 * np.pi * sd**2)
self.assertTrue(np.allclose(g.prob(mean), norm))
self.assertTrue(np.allclose(g.prob(mean + sd), np.exp(-1 / 2) * norm))
def test_prob(self):
mean, sd = np.random.rand(2)
g = Gaussian(mean, sd)
bg = BoundedGaussian(mean, sd, -1, 2)
self.assertEqual(bg.prob(1), g.prob(1))
self.assertEqual(bg.prob(-1), g.prob(-1))
self.assertEqual(bg.prob(2), g.prob(2))
self.assertEqual(bg.prob(-2), 0)
self.assertEqual(bg.prob(3), 0)