def test_lnprob(self):
mean, sd = np.random.rand(2)
g = Gaussian(mean, sd)
bg = BoundedGaussian(mean, sd, -1, 2)
self.assertEqual(bg.lnprob(1), g.lnprob(1))
self.assertEqual(bg.lnprob(-1), g.lnprob(-1))
self.assertEqual(bg.lnprob(2), g.lnprob(2))
self.assertEqual(bg.lnprob(-2), -np.inf)
self.assertEqual(bg.lnprob(3), -np.inf)
def test_gaussian():
g = Gaussian(1, 1)
assert_equal(g.guess, 1)
assert_obj_close(g.lnprob(0), gold_sigma)
def test_lnprob(self):
g = Gaussian(0, 1)
self.assertTrue(np.allclose(g.lnprob(1), GOLD_SIGMA))
g = Gaussian(0, 2)
self.assertTrue(np.allclose(g.lnprob(2), GOLD_SIGMA - np.log(2)))
def test_gaussian():
g = Gaussian(1, 1)
assert_equal(g.guess, 1)
assert_obj_close(g.lnprob(0),gold_sigma)