]> gitweb.michael.orlitzky.com - dead/census-tools.git/blob - src/Tests/Unit/DistributionTest.py
Added the (Uniform)DistributionTest modules.
[dead/census-tools.git] / src / Tests / Unit / DistributionTest.py
1 import unittest
2
3 import probability.Distribution
4
5
6 class DistributionTest(unittest.TestCase):
7
8 def testSampledValuesWithinComponentBounds(self):
9 """
10 Create a distribution from the sum of two uniforms, and make
11 sure that the values sampled from the sum lie in the interval
12 (min1+min2, max1+max2).
13 """
14 min1 = 0.1515
15 max1 = 22
16 min2 = 14
17 max2 = 100
18
19 ud1 = probability.Distribution.Uniform(min1, max1)
20 ud2 = probability.Distribution.Uniform(min2, max2)
21 distro = ud1 + ud2
22
23 for x in range(0,1000):
24 s = distro.sample()
25 self.assertTrue(s >= (min1+min2))
26 self.assertTrue(s <= (max1+max2))
27
28
29 def suite():
30 suite = unittest.TestSuite()
31 suite.addTest(unittest.makeSuite(DistributionTest))
32 return suite