def test_init(self):
rg = RandomGenerator(self.brng())
state = rg.state
rg.standard_normal(1)
rg.standard_normal(1)
rg.state = state
new_state = rg.state
assert_(comp_state(state, new_state))
def test_reset_state_uintegers(self):
rg = RandomGenerator(self.brng(*self.seed))
rg.random_uintegers(bits=32)
state = rg.state
n1 = rg.random_uintegers(bits=32, size=10)
rg2 = RandomGenerator(self.brng())
rg2.state = state
n2 = rg2.random_uintegers(bits=32, size=10)
assert_((n1 == n2).all())
def test_reset_state_uint32(self):
rg = RandomGenerator(self.brng(*self.seed))
rg.randint(0, 2**24, 120, dtype=np.uint32)
state = rg.state
n1 = rg.randint(0, 2**24, 10, dtype=np.uint32)
rg2 = RandomGenerator(self.brng())
rg2.state = state
n2 = rg2.randint(0, 2**24, 10, dtype=np.uint32)
assert_array_equal(n1, n2)
def test_reset_state_gauss(self):
rg = RandomGenerator(self.brng(*self.seed))
rg.standard_normal()
state = rg.state
n1 = rg.standard_normal(size=10)
rg2 = RandomGenerator(self.brng())
rg2.state = state
n2 = rg2.standard_normal(size=10)
assert_array_equal(n1, n2)
def test_normal_zig_floats(self):
rg = RandomGenerator(self.brng())
warmup(rg)
state = rg.state
r1 = rg.standard_normal(11, dtype=np.float32)
rg2 = RandomGenerator(self.brng())
warmup(rg2)
rg2.state = state
r2 = rg2.standard_normal(11, dtype=np.float32)
assert_array_equal(r1, r2)
assert_equal(r1.dtype, np.float32)
assert_(comp_state(rg.state, rg2.state))
def test_uniform_float(self):
rg = RandomGenerator(self.brng(12345))
warmup(rg)
state = rg.state
r1 = rg.random_sample(11, dtype=np.float32)
rg2 = RandomGenerator(self.brng())
warmup(rg2)
rg2.state = state
r2 = rg2.random_sample(11, dtype=np.float32)
assert_array_equal(r1, r2)
assert_equal(r1.dtype, np.float32)
assert_(comp_state(rg.state, rg2.state))
