diff options
Diffstat (limited to 'numpy/random/tests/test_generator_mt19937_regressions.py')
-rw-r--r-- | numpy/random/tests/test_generator_mt19937_regressions.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/numpy/random/tests/test_generator_mt19937_regressions.py b/numpy/random/tests/test_generator_mt19937_regressions.py index e328c0c71..3a937f997 100644 --- a/numpy/random/tests/test_generator_mt19937_regressions.py +++ b/numpy/random/tests/test_generator_mt19937_regressions.py @@ -56,10 +56,9 @@ class TestRegression(object): [1, (2, 2), (3, 3), None], [(1, 1), 2, 3, None]]: mt19937 = Generator(MT19937(12345)) - shuffled = np.array(t, dtype=object) + shuffled = list(t) mt19937.shuffle(shuffled) - expected = np.array([t[2], t[0], t[3], t[1]], dtype=object) - assert_array_equal(np.array(shuffled, dtype=object), expected) + assert_array_equal(shuffled, [t[2], t[0], t[3], t[1]]) def test_call_within_randomstate(self): # Check that custom BitGenerator does not call into global state @@ -119,7 +118,7 @@ class TestRegression(object): # a segfault on garbage collection. # See gh-7719 mt19937 = Generator(MT19937(1234)) - a = np.array([np.arange(1), np.arange(4)], dtype=object) + a = np.array([np.arange(1), np.arange(4)]) for _ in range(1000): mt19937.shuffle(a) |