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))