Skip to content

Commit 3c049ff

Browse files
committed
small fix to periodic tests
1 parent 4f53829 commit 3c049ff

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

specparam/objs/event.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -325,8 +325,9 @@ def get_group(self, event_inds, window_inds, output_type='event'):
325325
326326
Parameters
327327
----------
328-
event_inds, window_inds : array_like of int or array_like of bool
328+
event_inds, window_inds : array_like of int or array_like of bool or None
329329
Indices to extract from the object, for event and time windows.
330+
If None, selects all available indices.
330331
output_type : {'time', 'group'}, optional
331332
Type of model object to extract:
332333
'event' : SpectralTimeEventObject
@@ -384,6 +385,7 @@ def get_group(self, event_inds, window_inds, output_type='event'):
384385

385386
return output
386387

388+
387389
def print_results(self, concise=False):
388390
"""Print out SpectralTimeEventModel results.
389391

specparam/tests/analysis/test_periodic.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def test_get_band_peak_group(tfg):
1515

1616
assert np.all(get_band_peak_group(tfg, (8, 12)))
1717

18-
def test_get_band_peak_group():
18+
def test_get_band_peak_group_arr():
1919

2020
data = np.array([[10, 1, 1.8, 0], [13, 1, 2, 2], [14, 2, 4, 2]])
2121

@@ -27,7 +27,7 @@ def test_get_band_peak_group():
2727
assert out2.shape == (3, 3)
2828
assert np.array_equal(out2[2, :], [14, 2, 4])
2929

30-
def test_get_band_peak():
30+
def test_get_band_peak_arr():
3131

3232
data = np.array([[10, 1, 1.8], [14, 2, 4]])
3333

0 commit comments

Comments
 (0)