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))