Update test_filter_datasets.py

matthias-k · Dec 16, 2023 · c3cbb68 · c3cbb68
1 parent a4d8803
commit c3cbb68
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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)