Added the (Uniform)DistributionTest modules.

[dead/census-tools.git] / src / Tests / Unit / DistributionTest.py

diff --git a/src/Tests/Unit/DistributionTest.py b/src/Tests/Unit/DistributionTest.py
new file mode 100644 (file)
index 0000000..8ab3d0d
--- /dev/null
+++ b/src/Tests/Unit/DistributionTest.py
@@ -0,0 +1,32 @@
+import unittest
+
+import probability.Distribution
+
+
+class DistributionTest(unittest.TestCase):
+
+    def testSampledValuesWithinComponentBounds(self):
+        """
+        Create a distribution from the sum of two uniforms, and make
+        sure that the values sampled from the sum lie in the interval
+        (min1+min2, max1+max2).
+        """
+        min1 = 0.1515
+        max1 = 22
+        min2 = 14
+        max2 = 100
+
+        ud1 = probability.Distribution.Uniform(min1, max1)
+        ud2 = probability.Distribution.Uniform(min2, max2)
+        distro = ud1 + ud2
+
+        for x in range(0,1000):
+            s = distro.sample()
+            self.assertTrue(s >= (min1+min2))
+            self.assertTrue(s <= (max1+max2))
+
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(DistributionTest))
+    return suite