From 5660c899a388f5c78e01057f695b71148951ba34 Mon Sep 17 00:00:00 2001 From: Michael Date: Fri, 19 Apr 2024 15:22:22 -0700 Subject: [PATCH] add accuracy test --- neurodsp/tests/spectral/test_power.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/neurodsp/tests/spectral/test_power.py b/neurodsp/tests/spectral/test_power.py index af982f6f..bd3e3947 100644 --- a/neurodsp/tests/spectral/test_power.py +++ b/neurodsp/tests/spectral/test_power.py @@ -105,10 +105,13 @@ def test_compute_spectrum_medfilt(tsig, tsig_sine): _, psd_medfilt = compute_spectrum(tsig_sine, FS, method='medfilt', filt_len=0.1) assert np.allclose(psd, psd_medfilt, atol=EPS) -def test_compute_spectrum_multitaper(tsig): - - freqs, spectrum = compute_spectrum_multitaper(tsig, FS) +def test_compute_spectrum_multitaper(tsig_sine): + # Shape test + freqs, spectrum = compute_spectrum_multitaper(tsig_sine, FS) assert freqs.shape == spectrum.shape - + # Accuracy test: peak at sine frequency + idx_freq_sine = np.argmin(np.abs(freqs - FREQ_SINE)) + idx_peak = np.argmax(spectrum) + assert idx_freq_sine == idx_peak