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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
import _random
from rpython.rlib.rarithmetic import intmask
from rpython.rlib.rrandom import Random, N, r_uint
from rpython.translator.c.test.test_genc import compile
# the numbers were created by using CPython's _randommodule.c
def test_init_from_zero():
rnd = Random(0)
assert rnd.state[:14] == [0, 1, 1812433255, 1900727105, 1208447044,
2481403966, 4042607538, 337614300, 3232553940,
1018809052, 3202401494, 1775180719, 3192392114, 594215549]
def test_init_from_seed():
rnd = Random(1000)
assert rnd.state[:14] == [1000, 4252021385, 1724402292, 571538732,
73690720, 4283493349, 2222270404, 2464917285, 427676011,
1101193792, 2887716015, 3670250828, 1664757559, 1538426459]
def test_numbers():
rnd = Random(1000)
nums = [rnd.genrand32() for i in range(14)]
assert nums == [2807145907, 882709079, 493951047, 2621574848, 4081433851,
44058974, 2070996316, 1549632257, 3747249597, 3650674304,
911961945, 58396205, 174846504, 1478498153]
def test_init_by_array():
rnd = Random()
rnd.init_by_array([r_uint(n) for n in [1, 2, 3, 4]])
assert rnd.state[:14] == [2147483648, 1269538435, 699006892, 381364451,
172015551, 3237099449, 3609464087, 2187366456, 654585064,
2665903765, 3735624613, 1241943673, 2038528247, 3774211972]
# try arrays of various sizes to test for corner cases
for size in [N, N - 1, N + 1, N // 2, 2 * N]:
rnd.init_by_array([r_uint(n) for n in range(N)])
def test_jumpahead():
rnd = Random()
rnd.state = [r_uint(0)] * N
rnd.state[0] = r_uint(1)
cpyrandom = _random.Random()
cpyrandom.setstate(tuple([int(s) for s in rnd.state] + [rnd.index]))
rnd.jumpahead(100)
cpyrandom.jumpahead(100)
assert tuple(rnd.state) + (rnd.index, ) == cpyrandom.getstate()
def test_jumpahead_badstate():
rnd = Random()
s, j = 4043161618, 2674112291824205302
rnd.init_by_array([s])
rnd.jumpahead(j)
for i in range(500):
r = rnd.random()
assert r <= 1.0, (r, i)
def test_translate():
def f(x, y):
x = r_uint(x)
y = r_uint(y)
rnd = Random(x)
rnd.init_by_array([x, y])
rnd.jumpahead(intmask(y))
return float(rnd.genrand32()) + rnd.random()
fc = compile(f, [int, int])
assert fc(1, 2) == f(1, 2)
|
