def test_get_random_state_with_cupy(self):
rvs0 = self._get_rvs(_generator.RandomState(6))
os.environ['CUPY_SEED'] = '6'
rvs1 = self._get_rvs_reset()
self._check_same(rvs0, rvs1)
def setUp(self):
self.rs = _generator.RandomState(seed=testing.generate_seed())
def test_set_random_state_call_multiple_times(self):
_generator.set_random_state(_generator.RandomState())
rs = _generator.RandomState()
_generator.set_random_state(rs)
assert _generator.get_random_state() is rs
def test_set_random_state(self):
rs = _generator.RandomState()
_generator.set_random_state(rs)
assert _generator.get_random_state() is rs
