Ignore DeprecationWarning in test for cupyx.scipy.linalg.kron

chainer-ci · Jan 24, 2025 · 2385e1f · 2385e1f
1 parent f72cee4
commit 2385e1f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/cupyx_tests/scipy_tests/linalg_tests/test_special_matrices.py b/tests/cupyx_tests/scipy_tests/linalg_tests/test_special_matrices.py
@@ -1,4 +1,5 @@
 import unittest
+import warnings
 
 from cupy import testing
 import cupyx.scipy.linalg  # NOQA
@@ -63,7 +64,13 @@ class TestSpecialMatrices(TestSpecialMatricesBase):
                                  accept_error=ValueError)
     def test_special_matrix(self, xp, scp):
         function = getattr(scp.linalg, self.function)
-        return function(*[self._get_arg(xp, arg) for arg in self.args])
+
+        if self.function == "kron":
+            with warnings.catch_warnings():
+                warnings.filterwarnings('ignore', category=DeprecationWarning)
+                return function(*[self._get_arg(xp, arg) for arg in self.args])
+        else:
+            return function(*[self._get_arg(xp, arg) for arg in self.args])
 
 
 @testing.parameterize(*(