.

Author: ev-br

array_api_strict/tests/test_array_object.py

@@ -3,7 +3,7 @@
 import operator
 from builtins import all as all_
 
-from numpy.testing import assert_raises #, suppress_warnings
+from numpy.testing import assert_raises
 import numpy as np
 import pytest
 
@@ -271,11 +271,11 @@ def _check_op_array_scalar(dtypes, a, s, func, func_name, BIG_INT=BIG_INT):
         else:
             # Only test for no error
             with warnings.catch_warnings():
-   #         with suppress_warnings() as sup:
                 # ignore warnings from pow(BIG_INT)
-                warnings.filterwarnings("ignore", category=RuntimeWarning,
-   #             sup.filter(RuntimeWarning,
-                           message="invalid value encountered in power")
+                warnings.filterwarnings(
+                    "ignore", category=RuntimeWarning,
+                    message="invalid value encountered in power"
+                )
                 func(s)
             return True
 

array_api_strict/tests/test_elementwise_functions.py

@@ -2,7 +2,7 @@
 
 import numpy as np
 import pytest
-from numpy.testing import suppress_warnings
+##from numpy.testing import suppress_warnings
 
 
 from .. import asarray, _elementwise_functions
@@ -300,10 +300,16 @@ def _array_vals():
                     if allowed:
                         conv_scalar = a._promote_scalar(s)
 
-                        with suppress_warnings() as sup:
+                        with warnings.catch_warnings():
                             # ignore warnings from pow(BIG_INT)
-                            sup.filter(RuntimeWarning,
-                                       "invalid value encountered in power")
+                            warnings.filterwarnings(
+                                "ignore", category=RuntimeWarning,
+                                message="invalid value encountered in power"
+                            )
+##                        with suppress_warnings() as sup:
+                            # ignore warnings from pow(BIG_INT)
+##                            sup.filter(RuntimeWarning,
+##                                       "invalid value encountered in power")
                             assert func(s, a) == func(conv_scalar, a)
                             assert func(a, s) == func(a, conv_scalar)