accuracy tests to ensure attenuation

Author: ryanhammonds

fooof/tests/utils/test_data.py

@@ -25,17 +25,27 @@ def test_interpolate_spectrum():
     freqs, powers = gen_power_spectrum(\
         [1, 75], [1, 1], [[10, 0.5, 1.0], [60, 2, 0.1]])
 
-    freqs_out, powers_out = interpolate_spectrum(freqs, powers, [58, 62])
+    exclude = [58, 62]
+
+    freqs_out, powers_out = interpolate_spectrum(freqs, powers, exclude)
 
     assert np.array_equal(freqs, freqs_out)
     assert np.all(powers)
     assert powers.shape == powers_out.shape
+    mask = np.logical_and(freqs >= exclude[0], freqs <= exclude[1])
+    assert powers[mask].sum() > powers_out[mask].sum()
 
     # Test with multiple buffer exclusion zones
     freqs, powers = gen_power_spectrum(\
         [1, 150], [1, 100, 1], [[10, 0.5, 1.0], [60, 1, 0.1], [120, 0.5, 0.1]])
 
-    freqs_out, powers_out = interpolate_spectrum(freqs, powers, [[58, 62], [118, 122]])
+    exclude = [[58, 62], [118, 122]]
+
+    freqs_out, powers_out = interpolate_spectrum(freqs, powers, exclude)
     assert np.array_equal(freqs, freqs_out)
     assert np.all(powers)
     assert powers.shape == powers_out.shape
+
+    for f_range in exclude:
+        mask = np.logical_and(freqs >= f_range[0], freqs <= f_range[1])
+        assert powers[mask].sum() > powers_out[mask].sum()