From b9665d426a8af2e685ff78c7980037f9bfa2e0c8 Mon Sep 17 00:00:00 2001 From: Gertjan van den Burg Date: Thu, 14 Jan 2021 17:11:38 +0000 Subject: Move everything to a separate python directory I'm rearranging the repo so that everything is on master and can be tested more easily. --- new_python/tests/test_syncrng.py | 53 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 53 insertions(+) create mode 100644 new_python/tests/test_syncrng.py (limited to 'new_python/tests/test_syncrng.py') diff --git a/new_python/tests/test_syncrng.py b/new_python/tests/test_syncrng.py new file mode 100644 index 0000000..58b504d --- /dev/null +++ b/new_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() -- cgit v1.2.3