From d015a83b8c5f40849638a71af72b109007ba05c1 Mon Sep 17 00:00:00 2001 From: David Nicholson Date: Sat, 11 May 2024 08:29:43 -0400 Subject: [PATCH] Fix name `to_inds_list` -> segment_inds_list_from_class_labels` in test_transforms/test_frame_labels/test_functional.py --- .../test_frame_labels/test_functional.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_transforms/test_frame_labels/test_functional.py b/tests/test_transforms/test_frame_labels/test_functional.py index b5453240f..9e9013eb6 100644 --- a/tests/test_transforms/test_frame_labels/test_functional.py +++ b/tests/test_transforms/test_frame_labels/test_functional.py @@ -8,7 +8,7 @@ - to_segments: transform to get back segment onsets, offsets, and labels from labeled timebins. Inverse of ``from_segments``. - post-processing transforms that can be used to "clean up" a vector of labeled timebins - - to_inds_list: helper function used to find segments in a vector of labeled timebins + - segment_inds_list_from_class_labels: helper function used to find segments in a vector of labeled timebins - remove_short_segments: remove any segment less than a minimum duration - take_majority_vote: take a "majority vote" within each segment bounded by the background label, and apply the most "popular" label within each segment to all timebins in that segment @@ -264,11 +264,11 @@ def test_to_segments_real_data( ), ], ) -def test_to_inds(frame_labels, seg_inds_list_expected): +def test_segment_inds_list_from_class_labels(frame_labels, seg_inds_list_expected): """Test ``to_inds`` works as expected""" UNLABELED = 0 - seg_inds_list = vak.transforms.frame_labels.to_inds_list( + seg_inds_list = vak.transforms.frame_labels.segment_inds_list_from_class_labels( frame_labels=frame_labels, background_label=UNLABELED ) assert np.array_equal(seg_inds_list, seg_inds_list_expected) @@ -297,7 +297,7 @@ def test_to_inds(frame_labels, seg_inds_list_expected): ) def test_remove_short_segments(lbl_tb, unlabeled, timebin_dur, min_segment_dur, lbl_tb_expected): """Test ``remove_short_segments`` works as expected""" - segment_inds_list = vak.transforms.frame_labels.to_inds_list( + segment_inds_list = vak.transforms.frame_labels.segment_inds_list_from_class_labels( lbl_tb, background_label=unlabeled ) lbl_tb_tfm, segment_inds_list_out = vak.transforms.frame_labels.remove_short_segments( @@ -346,7 +346,7 @@ def test_remove_short_segments(lbl_tb, unlabeled, timebin_dur, min_segment_dur, ) def test_majority_vote(lbl_tb_in, unlabeled, lbl_tb_expected): """Test ``majority_vote`` works as expected""" - segment_inds_list = vak.transforms.frame_labels.to_inds_list( + segment_inds_list = vak.transforms.frame_labels.segment_inds_list_from_class_labels( lbl_tb_in, background_label=unlabeled ) lbl_tb_maj_vote = vak.transforms.frame_labels.take_majority_vote(