Skip to content

Commit 1a2ceff

Browse files
committed
TST: Change random to RandomState
1 parent 5c97c50 commit 1a2ceff

File tree

1 file changed

+15
-16
lines changed

1 file changed

+15
-16
lines changed

numpy/_core/tests/test_multiarray.py

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -6837,7 +6837,7 @@ class TestDot:
68376837
N = 7
68386838

68396839
def _create_data(self):
6840-
rng = np.random.default_rng(128)
6840+
rng = np.random.RandomState(128)
68416841
A = rng.random((4, 2))
68426842
b1 = rng.random((2, 1))
68436843
b2 = rng.random(2)
@@ -6848,52 +6848,51 @@ def _create_data(self):
68486848
def test_dotmatmat(self):
68496849
A, _, _, _, _ = self._create_data()
68506850
res = np.dot(A.transpose(), A)
6851-
tgt = np.array([[1.47632191, 1.30902162],
6852-
[1.30902162, 1.44047178]])
6851+
tgt = np.array([[1.45046013, 0.86323640],
6852+
[0.86323640, 0.84934569]])
68536853
assert_almost_equal(res, tgt, decimal=self.N)
68546854

68556855
def test_dotmatvec(self):
68566856
A, b1, _, _, _ = self._create_data()
68576857
res = np.dot(A, b1)
6858-
tgt = np.array([[0.53147439], [0.3881244],
6859-
[0.28278056], [0.81799083]])
6858+
tgt = np.array([[0.32114320], [0.04889721],
6859+
[0.15696029], [0.33612621]])
68606860
assert_almost_equal(res, tgt, decimal=self.N)
68616861

68626862
def test_dotmatvec2(self):
68636863
A, _, b2, _, _ = self._create_data()
68646864
res = np.dot(A, b2)
6865-
tgt = np.array([0.73687498, 0.51701463, 0.27242132, 1.09264356])
6865+
tgt = np.array([0.29677940, 0.04518649, 0.14468333, 0.31039293])
68666866
assert_almost_equal(res, tgt, decimal=self.N)
68676867

68686868
def test_dotvecmat(self):
68696869
A, _, _, _, b4 = self._create_data()
68706870
res = np.dot(b4, A)
6871-
tgt = np.array([0.30087302, 0.76540662])
6871+
tgt = np.array([1.23495091, 1.12222648])
68726872
assert_almost_equal(res, tgt, decimal=self.N)
68736873

68746874
def test_dotvecmat2(self):
68756875
A, _, _, b3, _ = self._create_data()
68766876
res = np.dot(b3, A.transpose())
6877-
tgt = np.array([[0.87004937, 0.63750863, 0.47499902, 1.3432763]])
6877+
tgt = np.array([[0.58793804, 0.08957460, 0.30605758, 0.62716383]])
68786878
assert_almost_equal(res, tgt, decimal=self.N)
68796879

68806880
def test_dotvecmat3(self):
68816881
A, _, _, _, b4 = self._create_data()
68826882
res = np.dot(A.transpose(), b4)
6883-
tgt = np.array([0.30087302, 0.76540662])
6883+
tgt = np.array([1.23495091, 1.12222648])
68846884
assert_almost_equal(res, tgt, decimal=self.N)
68856885

68866886
def test_dotvecvecouter(self):
68876887
_, b1, _, b3, _ = self._create_data()
68886888
res = np.dot(b1, b3)
6889-
tgt = np.array([[0.41852888, 0.35668856],
6890-
[0.33830126, 0.28831508]])
6889+
tgt = np.array([[0.20128610, 0.08400440], [0.07190947, 0.03001058]])
68916890
assert_almost_equal(res, tgt, decimal=self.N)
68926891

68936892
def test_dotvecvecinner(self):
68946893
_, b1, _, b3, _ = self._create_data()
68956894
res = np.dot(b3, b1)
6896-
tgt = np.array([[0.70684396]])
6895+
tgt = np.array([[0.23129668]])
68976896
assert_almost_equal(res, tgt, decimal=self.N)
68986897

68996898
def test_dotcolumnvect1(self):
@@ -6911,19 +6910,19 @@ def test_dotcolumnvect2(self):
69116910
assert_almost_equal(res, tgt, decimal=self.N)
69126911

69136912
def test_dotvecscalar(self):
6914-
rng = np.random.default_rng(100)
6913+
rng = np.random.RandomState(100)
69156914
b1 = rng.random((1, 1))
69166915
b2 = rng.random((1, 4))
69176916
res = np.dot(b1, b2)
6918-
tgt = np.array([[0.49811165, 0.2411955, 0.03586377, 0.81298353]])
6917+
tgt = np.array([[0.15126730, 0.23068496, 0.45905553, 0.00256425]])
69196918
assert_almost_equal(res, tgt, decimal=self.N)
69206919

69216920
def test_dotvecscalar2(self):
6922-
rng = np.random.default_rng(100)
6921+
rng = np.random.RandomState(100)
69236922
b1 = rng.random((4, 1))
69246923
b2 = rng.random((1, 1))
69256924
res = np.dot(b1, b2)
6926-
tgt = np.array([[0.81298353], [0.58083745], [0.28125296], [0.04181999]])
6925+
tgt = np.array([[0.00256425], [0.00131359], [0.00200324], [0.00398638]])
69276926
assert_almost_equal(res, tgt, decimal=self.N)
69286927

69296928
def test_all(self):

0 commit comments

Comments
 (0)