add test for issue 27036

mattip · mattip · commit b19b685dc311 · 2024-08-08T09:19:28.000+03:00
diff --git a/numpy/linalg/tests/test_regression.py b/numpy/linalg/tests/test_regression.py
@@ -161,3 +161,17 @@ def test_matrix_rank_rtol_argument(self, rtol):
         x = np.zeros((4, 3, 2))
         res = np.linalg.matrix_rank(x, rtol=rtol)
         assert res.shape == (4,)
+
+    def test_openblas_threading(self):
+        # gh-27036
+        # Test whether matrix multiplication involving a large matrix always gives the same answer
+        x = np.arange(500000, dtype=np.float64)
+        src = np.vstack((x, -10*x)).T
+        matrix = np.array([[0, 1], [1, 0]])
+        expected = np.vstack((-10*x, x)).T  # src @ matrix
+        for i in range(200):
+            result = src @ matrix
+            mismatches = (~np.isclose(result, expected)).sum()
+            if mismatches != 0:
+                assert False, ("unexpected result from matmul, "
+                    "probably due to OpenBLAS threading issues")
