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