Silence warnings output in the tests

Author: asmeurer

array_api_strict/tests/test_array_object.py

@@ -410,7 +410,8 @@ def test_array_namespace():
     assert a.__array_namespace__(api_version="2022.12") is array_api_strict
     assert array_api_strict.__array_api_version__ == "2022.12"
 
-    assert a.__array_namespace__(api_version="2023.12") is array_api_strict
+    with pytest.warns(UserWarning):
+        assert a.__array_namespace__(api_version="2023.12") is array_api_strict
     assert array_api_strict.__array_api_version__ == "2023.12"
 
     with pytest.warns(UserWarning):

array_api_strict/tests/test_elementwise_functions.py

@@ -12,6 +12,8 @@
 )
 from .._flags import set_array_api_strict_flags
 
+import pytest
+
 def nargs(func):
     return len(getfullargspec(func).args)
 
@@ -101,7 +103,8 @@ def _array_vals():
             yield asarray(1.0, dtype=d)
 
     # Use the latest version of the standard so all functions are included
-    set_array_api_strict_flags(api_version="2023.12")
+    with pytest.warns(UserWarning):
+        set_array_api_strict_flags(api_version="2023.12")
 
     for x in _array_vals():
         for func_name, types in elementwise_function_input_types.items():

array_api_strict/tests/test_flags.py

@@ -73,9 +73,10 @@ def test_flags():
                   set_array_api_strict_flags(api_version='2020.12'))
     pytest.raises(ValueError, lambda: set_array_api_strict_flags(
                       enabled_extensions=('linalg', 'fft', 'invalid')))
-    pytest.raises(ValueError, lambda: set_array_api_strict_flags(
-        api_version='2021.12',
-        enabled_extensions=('linalg', 'fft')))
+    with pytest.warns(UserWarning):
+        pytest.raises(ValueError, lambda: set_array_api_strict_flags(
+            api_version='2021.12',
+            enabled_extensions=('linalg', 'fft')))
 
     # Test resetting flags
     with pytest.warns(UserWarning):
@@ -96,7 +97,8 @@ def test_api_version():
     assert xp.__array_api_version__ == '2022.12'
 
     # Test setting the version
-    set_array_api_strict_flags(api_version='2021.12')
+    with pytest.warns(UserWarning):
+        set_array_api_strict_flags(api_version='2021.12')
     assert xp.__array_api_version__ == '2021.12'
 
 def test_data_dependent_shapes():