try to import ComplexWarning from np.exceptions

simpeg · Oct 13, 2024 · 3f7288d · 3f7288d
1 parent 8bdf3b1
commit 3f7288d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -3,6 +3,11 @@
 from geoana.utils import check_xyz_dim, mkvc, ndgrid
 import pytest
 
+try:
+    from numpy.exceptions import ComplexWarning
+except ImportError:
+    from numpy import ComplexWarning
+
 
 def test_config_info():
     info = geoana.show_config()
@@ -95,7 +100,7 @@ def test_dtype_cast_good():
 
 def test_bad_dtype_cast():
     xyz = np.random.rand(10, 3) + 1j* np.random.rand(10, 3)
-    with pytest.warns(np.ComplexWarning):
+    with pytest.warns(ComplexWarning):
         xyz2 = check_xyz_dim(xyz)
     xyz = [['0', 'O', 'o']]
     with pytest.raises(ValueError):