]> asedeno.scripts.mit.edu Git - bluechips.git/commitdiff
Make the distribution test actually do something
authorEvan Broder <broder@mit.edu>
Sun, 20 Jul 2008 14:53:17 +0000 (14:53 +0000)
committerEvan Broder <broder@mit.edu>
Sun, 20 Jul 2008 14:53:17 +0000 (14:53 +0000)
bluechips/tests/splits/test_random.py

index 566c14f33c0741ae037b27b005fe042061f067d0..f553f7ac7e4315cc131a649336b6cf8f3a2f5886 100644 (file)
@@ -24,7 +24,9 @@ class TestSplitRandom(TestCase):
             self.assertEqual(sum(s.share for s in e.splits), e.amount)
     
     def test_splitDistribution(self):
+        user_count = meta.Session.query(model.User).count()
         for e in meta.Session.query(model.Expenditure):
-            even_total = (e.amount / meta.Session.query(model.User).count())
-            self.assert_(std_dev(list(s.share for s in e.splits)) <= even_total, \
+            even_total = (e.amount / user_count) * user_count
+            difference = abs(even_total - e.amount)
+            self.assert_(std_dev(list(int(s.share) for s in e.splits)) <= difference, \
                 "Expenditure doesn't appear to be evenly distributed")