1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
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()
|
