Merge pull request cupy#7908 from asi1024/lobpcg-test

Fix `maxiter` in `TestLOBPCG`
YanivDorGalron · Oct 10, 2023 · 9f819ae · 9f819ae
2 parents b6b725d + 46d34ff
commit 9f819ae
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/cupyx_tests/scipy_tests/sparse_tests/test_linalg.py b/tests/cupyx_tests/scipy_tests/sparse_tests/test_linalg.py
@@ -1017,7 +1017,7 @@ def test_generate_input_for_elastic_rod(self, xp, sp):
         X = self._generate_random_initial_ortho_eigvec(n, 20, xp)
         eigvals, eigvecs = sp.linalg.lobpcg(A,
                                             X, B=B,
-                                            tol=1e-5, maxiter=30,
+                                            tol=1e-5, maxiter=50,
                                             largest=False)
         return eigvals, _eigen_vec_transform(eigvecs, xp)
 
@@ -1029,7 +1029,7 @@ def test_generate_input_for_mikota_pair(self, xp, sp):
         X = self._generate_random_initial_ortho_eigvec(n, 20, xp)
         eigvals, eigvecs = sp.linalg.lobpcg(A,
                                             X, B=B,
-                                            tol=1e-5, maxiter=30,
+                                            tol=1e-5, maxiter=50,
                                             largest=False)
         return eigvals, _eigen_vec_transform(eigvecs, xp)
 
@@ -1176,7 +1176,7 @@ def _verbosity_helper(self, xp, sp):
         saved_stdout = io.StringIO()
         with contextlib.redirect_stdout(saved_stdout):
             _, _ = sp.linalg.lobpcg(A, X, B=B, tol=1e-5,
-                                    maxiter=30, largest=False,
+                                    maxiter=10, largest=False,
                                     verbosityLevel=9)
         output = saved_stdout.getvalue().strip()
         return output