simplify some tests

Author: ilanschnell

devel/random/binomial.py

@@ -60,10 +60,7 @@ def test_pmf_simple(self):
     def test_pmf_sum(self):
         for n in 10, 100, 1_000, 10_000:
             b = BinomialDist(n, 0.5)
-            tot = 0
-            for k in range(n + 1):
-                tot += b.pmf(k)
-            self.assertAlmostEqual(tot, 1.0)
+            self.assertAlmostEqual(sum(b.pmf(k) for k in range(n + 1)), 1.0)
 
     def test_pmf(self):
         for n in 10, 50, 100, 250:

devel/random/sample.py

@@ -75,11 +75,8 @@ def test_pdf(self):
         smd = SampleMeanDist(n, k)
         N = 1000
         dx = n / N
-        p = 0.0
-        for i in range(N + 1):
-            x = i * dx
-            p += smd.pdf(x) * dx
-        self.assertAlmostEqual(p, 1.0)
+        self.assertAlmostEqual(
+            sum(smd.pdf(i * dx) * dx for i in range(N + 1)), 1.0)
 
     def test_cdf(self):
         smd = SampleMeanDist(100, 30)