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_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_gauss_inv(self):
n = 25
rs = RandomGenerator(self.brng(*self.data1['seed']))
gauss = rs.standard_normal(n)
assert_allclose(gauss, gauss_from_uint(self.data1['data'], n, 'dsfmt'))
rs = RandomGenerator(self.brng(*self.data2['seed']))
gauss = rs.standard_normal(25)
assert_allclose(gauss, gauss_from_uint(self.data2['data'], n, 'dsfmt'))
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))
