1 files changed, 53 insertions, 0 deletions
diff --git a/python/tests/test_syncrng.py b/python/tests/test_syncrng.py
new file mode 100644
index 0000000..58b504d
--- /dev/null
+++ b/python/tests/test_syncrng.py
@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+
+import os
+import unittest
+
+from SyncRNG import SyncRNG
+
+
+class SyncRNGTestCase(unittest.TestCase):
+    def test_randi(self):
+        s = SyncRNG(seed=123456)
+        self.assertEqual(s.randi(), 959852049)
+        self.assertEqual(s.randi(), 2314333085)
+        self.assertEqual(s.randi(), 2255782734)
+        self.assertEqual(s.randi(), 2921461239)
+        self.assertEqual(s.randi(), 1024197102)
+
+    def test_rand(self):
+        s = SyncRNG(seed=123456)
+        self.assertAlmostEqual(s.rand(), 959852049 / pow(2, 32))
+        self.assertAlmostEqual(s.rand(), 2314333085 / pow(2, 32))
+        self.assertAlmostEqual(s.rand(), 2255782734 / pow(2, 32))
+        self.assertAlmostEqual(s.rand(), 2921461239 / pow(2, 32))
+        self.assertAlmostEqual(s.rand(), 1024197102 / pow(2, 32))
+
+    def test_randbelow(self):
+        s = SyncRNG(seed=123456)
+        self.assertEqual(s.randbelow(5), 4)
+        self.assertEqual(s.randbelow(5), 0)
+        self.assertEqual(s.randbelow(5), 4)
+        self.assertEqual(s.randbelow(5), 4)
+        self.assertEqual(s.randbelow(5), 2)
+
+    def test_shuffle(self):
+        s = SyncRNG(seed=123456)
+        x = [1, 2, 3, 4, 5]
+        y = s.shuffle(x)
+        self.assertEqual(y, [3, 4, 1, 2, 5])
+
+    def test_first_1000(self):
+        s = SyncRNG(seed=0)
+
+        here = os.path.dirname(__file__)
+        test_file = os.path.join(here, "first_1000_seed_0.txt")
+
+        with open(test_file, "r") as fid:
+            for line in fid:
+                exp = int(line.strip())
+                self.assertTrue(exp == s.randi())
+
+
+if __name__ == "__main__":
+    unittest.main()