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)