[test]: test_periodogram_arr_error now catches errors and warnings

mscaudill · Sep 5, 2023 · 58e2c10 · 58e2c10
1 parent 0e14e04
commit 58e2c10
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/test_spectra.py b/tests/test_spectra.py
@@ -153,11 +153,12 @@ def test_periodogram_arr_error():
     return_onesided=True
     axis = -1
 
-    # test that ValueError is raised by Numpy fft
+    # test that ValueError is raised by Numpy fft and np internal
+    # RuntimeWarnings
     with pytest.raises(ValueError):
-
-        op_f, op_res = periodogram(arr, fs, nfft, window, axis, detrend, 
-                                   scaling) 
+        with pytest.warns(RuntimeWarning):
+            op_f, op_res = periodogram(arr, fs, nfft, window, axis, detrend, 
+                                       scaling) 
 
 
 def test_welch_pros():