def test_sum_three_pmfs(self):
pmfs = [PMF.fromkeys((0, 1), 0.5) for n in range(3)]
sum_pmf = sum_independent_pmfs(pmfs)
self.assertTrue(0.124 < sum_pmf[0] < 0.126)
self.assertTrue(0.374 < sum_pmf[1] < 0.376)
self.assertTrue(0.374 < sum_pmf[2] < 0.376)
self.assertTrue(0.124 < sum_pmf[3] < 0.126)
def test_sum_three_pmfs(self):
pmfs = [PMF.fromkeys((0, 1), 0.5) for n in range(3)]
sum_pmf = sum_independent_pmfs(pmfs)
self.assertTrue(0.124 < sum_pmf[0] < 0.126)
self.assertTrue(0.374 < sum_pmf[1] < 0.376)
self.assertTrue(0.374 < sum_pmf[2] < 0.376)
self.assertTrue(0.124 < sum_pmf[3] < 0.126)
def test_zerosum(self):
self.pmf = PMF.fromkeys("abcde", 0)
self.pmf.normalize()
total = sum(self.pmf.itervalues())
# This is how we verify total is 'nan': only 'nan' is not equal to itself.
self.assertNotEqual(total, total)
def test_floats(self):
self.pmf = PMF.fromkeys("abcde", 1.0)
self.exercise_pmf()
def setUp(self):
# Stabilize the random number generator, so test results using random are
# consistent. (This has side effects, since state of python RNG is static.)
random.seed(0)
self.pmf = PMF.fromkeys("abcde", 1)
def test_expectation(self):
pmf = PMF.fromkeys((1, 2, 3), 1.0)
pmf.normalize()
self.assertTrue(1.999 < pmf.expectation() < 2.001)
def test_sum_two_pmfs(self):
pmfs = [PMF.fromkeys((0, 1), 0.5) for n in range(2)]
sum_pmf = sum_independent_pmfs(pmfs)
self.assertTrue(0.249 < sum_pmf[0] < 0.251)
self.assertTrue(0.499 < sum_pmf[1] < 0.501)
self.assertTrue(0.249 < sum_pmf[2] < 0.251)
def test_zerosum(self):
self.pmf = PMF.fromkeys('abcde', 0)
self.pmf.normalize()
total = sum(self.pmf.itervalues())
# This is how we verify total is 'nan': only 'nan' is not equal to itself.
self.assertNotEqual(total, total)
def test_floats(self):
self.pmf = PMF.fromkeys('abcde', 1.)
self.exercise_pmf()
def setUp(self):
# Stabilize the random number generator, so test results using random are
# consistent. (This has side effects, since state of python RNG is static.)
random.seed(0)
self.pmf = PMF.fromkeys('abcde', 1)
def test_expectation(self):
pmf = PMF.fromkeys((1, 2, 3), 1.)
pmf.normalize()
self.assertTrue(1.999 < pmf.expectation() < 2.001)
def test_sum_two_pmfs(self):
pmfs = [PMF.fromkeys((0, 1), 0.5) for n in range(2)]
sum_pmf = sum_independent_pmfs(pmfs)
self.assertTrue(0.249 < sum_pmf[0] < 0.251)
self.assertTrue(0.499 < sum_pmf[1] < 0.501)
self.assertTrue(0.249 < sum_pmf[2] < 0.251)
