Added test cases for sampling

Author: Julian Blank

pysampling/algorithms/halton.py

@@ -4,21 +4,23 @@
 from pysampling.util import calc_primes_until
 
 
+def halton_sequence_by_index(i, b):
+    f = 1.0
+    x = 0.0
+    while i > 0:
+        f /= b
+        x += f * (i % b)
+        i = np.floor(i / b)
+    return x
+
+
+def halton_sequence(n, b):
+    return np.array([halton_sequence_by_index(i, b) for i in range(n)])
+
+
 class HaltonSampling(Sampling):
 
     def _sample(self, n_points, n_dim):
         bases = calc_primes_until(500)[:n_dim]
-        X = np.column_stack([self.halton_sequence(n_points, b) for b in bases])
+        X = np.column_stack([halton_sequence(n_points, b) for b in bases])
         return X
-
-    def halton_sequence(self, n, b):
-        return np.array([self.halton_sequence_by_index(i, b) for i in range(n)])
-
-    def halton_sequence_by_index(self, i, b):
-        f = 1.0
-        x = 0.0
-        while i > 0:
-            f /= b
-            x += f * (i % b)
-            i = np.floor(i / b)
-        return x

pysampling/algorithms/sobol.py

@@ -11,9 +11,10 @@ class SobolSampling(Sampling):
     def __init__(self,
                  n_skip=None,
                  n_leap=0,
-                 setup="joekuo"):
+                 setup="joekuo",
+                 **kwargs):
 
-        super().__init__()
+        super().__init__(**kwargs)
 
         if setup == "matlab":
             if n_skip is None:

tests/test_sampling.py

@@ -0,0 +1,21 @@
+import unittest
+
+from pysampling.sample import sample
+
+
+class SamplingTest(unittest.TestCase):
+
+    def test_random(self):
+        sample("random", 50, 2, seed=1)
+
+    def test_lhs(self):
+        sample("lhs", 50, 2, seed=1)
+
+    def test_sobol(self):
+        sample("sobol", 50, 2, seed=1)
+
+    def test_halton(self):
+        sample("halton", 50, 2, seed=1)
+
+if __name__ == '__main__':
+    unittest.main()