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