updated tests for numpy 1.19.1

Author: oleksandr-pavlyk

mkl_random/tests/test_random.py

@@ -27,8 +27,9 @@
 from __future__ import division, absolute_import, print_function
 
 import numpy as np
+from unittest import TestCase
 from numpy.testing import (
-        TestCase, run_module_suite, assert_, assert_raises, assert_equal,
+        run_module_suite, assert_, assert_raises, assert_equal,
         assert_warns, suppress_warnings)
 import mkl_random as rnd
 from numpy.compat import asbytes
@@ -909,7 +910,7 @@ def test_zipf(self):
         np.testing.assert_array_equal(actual, desired)
 
 
-class TestThread_Intel(object):
+class TestThread_Intel(TestCase):
     # make sure each state produces the same sequence even in threads
     def setUp(self):
         self.seeds = range(4)

mkl_random/tests/test_regression.py

@@ -95,9 +95,10 @@ def test_shuffle_mixed_dimension(self):
                   [1, (2, 2), (3, 3), None],
                   [(1, 1), 2, 3, None]]:
             rnd.seed(12345, brng='MT2203')
-            shuffled = list(t)
+            shuffled = np.array(list(t), dtype=object)
             rnd.shuffle(shuffled)
-            assert_array_equal(shuffled, [t[0], t[2], t[1], t[3]])
+            expected = np.array([t[0], t[2], t[1], t[3]], dtype=object)
+            assert_array_equal(shuffled, expected)
 
     def test_call_within_randomstate(self):
         # Check that custom RandomState does not call into global state
@@ -158,7 +159,7 @@ def test_shuffle_of_array_of_objects(self):
         # a segfault on garbage collection.
         # See gh-7719
         rnd.seed(1234)
-        a = np.array([np.arange(1), np.arange(4)])
+        a = np.array([np.arange(4), np.arange(4)])
 
         for _ in range(1000):
             rnd.shuffle(a)