From 41c871b9b178ec4d92a4f6a62cbf237eceb58e01 Mon Sep 17 00:00:00 2001 From: Tom Donoghue Date: Wed, 26 Jul 2023 01:29:15 -0400 Subject: [PATCH 1/2] add get_freq_ind helper function --- specparam/tests/utils/test_data.py | 7 +++++++ specparam/utils/data.py | 19 +++++++++++++++++++ 2 files changed, 26 insertions(+) diff --git a/specparam/tests/utils/test_data.py b/specparam/tests/utils/test_data.py index 2d54c1748..c271a7fa0 100644 --- a/specparam/tests/utils/test_data.py +++ b/specparam/tests/utils/test_data.py @@ -9,6 +9,13 @@ ################################################################################################### ################################################################################################### +def test_get_freq_ind(): + + freqs = np.array([1, 2, 3, 4, 5]) + + assert get_freq_ind(freqs, 2.2) == 1 + assert get_freq_ind(freqs, 3.7) == 3 + def test_trim_spectrum(): f_in = np.array([0., 1., 2., 3., 4., 5.]) diff --git a/specparam/utils/data.py b/specparam/utils/data.py index e30ad2060..a39ff1a01 100644 --- a/specparam/utils/data.py +++ b/specparam/utils/data.py @@ -10,6 +10,25 @@ ################################################################################################### ################################################################################################### +def get_freq_ind(freqs, freq): + """Get the index of the closest frequency value to a specified input frequency. + + Parameters + ---------- + freqs : 1d array + Frequency values. + freq : float + Frequency value to select closest index to. + + Returns + ------- + int + Index of closest value in `freqs` to `freq`. + """ + + return np.argmin(np.abs(freqs - freq)) + + def trim_spectrum(freqs, power_spectra, f_range): """Extract a frequency range from power spectra. From 1a93e5eda1884c1bf7f719cd7e8f1dc10df6b4c1 Mon Sep 17 00:00:00 2001 From: Tom Donoghue Date: Tue, 11 Mar 2025 22:32:39 -0400 Subject: [PATCH 2/2] fix spacing --- specparam/utils/reports.py | 1 + 1 file changed, 1 insertion(+) diff --git a/specparam/utils/reports.py b/specparam/utils/reports.py index b7ce85448..89e6ea0ef 100644 --- a/specparam/utils/reports.py +++ b/specparam/utils/reports.py @@ -29,6 +29,7 @@ def methods_report_info(model_obj=None, concise=False): print(gen_settings_str(model_obj, concise=concise)) print(gen_freq_range_str(model_obj, concise=concise)) + def methods_report_text(model_obj=None): """Prints out a text template of methods reporting information.