class TestCDF(unittest.TestCase):
def setUp(self):
self.pmf = PMF()
self.pmf.uniform_dist("abcde")
self.cdf = CDF(self.pmf)
def test_percentile(self):
self.assertEqual("a", self.cdf.percentile(0.0))
self.assertEqual("a", self.cdf.percentile(0.1))
self.assertEqual("a", self.cdf.percentile(0.2))
self.assertEqual("b", self.cdf.percentile(0.3))
self.assertEqual("b", self.cdf.percentile(0.4))
self.assertEqual("c", self.cdf.percentile(0.5))
self.assertEqual("c", self.cdf.percentile(0.6))
self.assertEqual("d", self.cdf.percentile(0.7))
self.assertEqual("d", self.cdf.percentile(0.8))
self.assertEqual("e", self.cdf.percentile(0.9))
self.assertEqual("e", self.cdf.percentile(1.0))
def test_percentiles(self):
self.assertEqual(("b", "d"), self.cdf.percentiles(0.3, 0.8))
class TestCDF(unittest.TestCase):
def setUp(self):
self.pmf = PMF()
self.pmf.uniform_dist('abcde')
self.cdf = CDF(self.pmf)
def test_percentile(self):
self.assertEqual('a', self.cdf.percentile(0.0))
self.assertEqual('a', self.cdf.percentile(0.1))
self.assertEqual('a', self.cdf.percentile(0.2))
self.assertEqual('b', self.cdf.percentile(0.3))
self.assertEqual('b', self.cdf.percentile(0.4))
self.assertEqual('c', self.cdf.percentile(0.5))
self.assertEqual('c', self.cdf.percentile(0.6))
self.assertEqual('d', self.cdf.percentile(0.7))
self.assertEqual('d', self.cdf.percentile(0.8))
self.assertEqual('e', self.cdf.percentile(0.9))
self.assertEqual('e', self.cdf.percentile(1.0))
def test_percentiles(self):
self.assertEqual(('b', 'd'), self.cdf.percentiles(0.3, 0.8))
