def test_array(self):
s = mt19937.RandomState(range(10))
assert_equal(s.randint(1000), 468)
s = np.random.RandomState(range(10))
assert_equal(s.randint(1000), 468)
s = mt19937.RandomState(np.arange(10))
assert_equal(s.randint(1000), 468)
s = mt19937.RandomState([0])
assert_equal(s.randint(1000), 973)
s = mt19937.RandomState([4294967295])
assert_equal(s.randint(1000), 265)
def test_scalar(self):
s = mt19937.RandomState(0)
assert_equal(s.randint(1000), 684)
s1 = np.random.RandomState(0)
assert_equal(s1.randint(1000), 684)
assert_equal(s1.randint(1000), s.randint(1000))
s = mt19937.RandomState(4294967295)
assert_equal(s.randint(1000), 419)
s1 = np.random.RandomState(4294967295)
assert_equal(s1.randint(1000), 419)
assert_equal(s1.randint(1000), s.randint(1000))
self.rs.seed(4294967295)
self.nprs.seed(4294967295)
self._is_state_common()
def test_call_within_randomstate(self):
# Check that custom RandomState does not call into global state
m = mt19937.RandomState()
res = np.array([0, 8, 7, 2, 1, 9, 4, 7, 0, 3])
for i in range(3):
mt19937.seed(i)
m.seed(4321)
# If m.state is not honored, the result will change
assert_array_equal(m.choice(10, size=10, p=np.ones(10)/10.), res)
