diff --git a/tests/test_filter_datasets.py b/tests/test_filter_datasets.py index 25c8623..44584da 100644 --- a/tests/test_filter_datasets.py +++ b/tests/test_filter_datasets.py @@ -453,3 +453,12 @@ def test_filter_scanpaths_by_length(fixation_trains, intervals): inds = [0, 2] expected_scanpaths = scanpaths.filter_fixation_trains(inds) compare_scanpaths(filtered_scanpaths, expected_scanpaths) + + +def test_remove_stimuli_without_fixations(file_stimuli_with_attributes, fixation_trains): + fixations = fixation_trains[:] + filtered_stimuli, filtered_fixations = remove_stimuli_without_fixations(file_stimuli_with_attributes, fixations) + inds = [0, 1] + expected_stimuli, expected_fixations = create_subset(file_stimuli_with_attributes, fixations, inds) + compare_fixations(filtered_fixations, expected_fixations) + assert_stimuli_equal(filtered_stimuli, expected_stimuli)