fix the pytest.warns(None) deprecation.

GeminiDRSoftware · Feb 5, 2024 · 8f2f902 · 8f2f902
1 parent 30ec0d9
commit 8f2f902
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/astrodata/tests/test_nddata.py b/astrodata/tests/test_nddata.py
@@ -1,3 +1,5 @@
+import warnings
+
 import numpy as np
 import pytest
 from numpy.testing import assert_array_almost_equal, assert_array_equal
@@ -93,7 +95,7 @@ def test_uncertainty_negative_numbers():
     arr = np.zeros(5)
 
     # No warning if all 0
-    with pytest.warns(None) as w:
+    with warnings.catch_warnings(record=True) as w:
         ADVarianceUncertainty(arr)
     assert len(w) == 0
 

diff --git a/astrodata/tests/test_object_construction.py b/astrodata/tests/test_object_construction.py
@@ -1,3 +1,5 @@
+import warnings
+
 import astrodata
 import astropy.units as u
 import numpy as np
@@ -294,7 +296,7 @@ def test_table_with_units(tmp_path):
     ad[0].TABLE1 = Table([[1]])
     ad[0].TABLE1['col0'].unit = 'mag(cm2 electron / erg)'
 
-    with pytest.warns(None) as w:
+    with warnings.catch_warnings(record=True) as w:
         ad.write(testfile)
 
     assert len(w) == 0