Merge pull request #442 from EIT-ALIVE/fix/441-watershed-excluded-peaks

Fix the excluded peaks in WatershedLungspace

Author: psomhorst

eitprocessing/roi/watershed.py

@@ -168,7 +168,7 @@ def apply(  # noqa: PLR0915
 
             included_marker_indices = np.isin(peaks_loc_int, markers_inside_tiv_mask)
             included_peaks = np.argwhere(included_marker_indices)
-            excluded_peaks = np.argwhere(~included_marker_indices)
+            excluded_peaks = np.argwhere(peaks_loc_bool & ~included_marker_indices)
 
             included_watershed_regions = np.where(included_region, watershed_regions, np.nan)
 

tests/test_watershed.py

@@ -159,6 +159,8 @@ def test_watershed_captures(draeger1: Sequence):
     ]:
         assert key in captures, f"captures should have a '{key}' entry"
 
+    assert len(captures["local peaks"]) == len(captures["included peaks"]) + len(captures["excluded peaks"])
+
 
 def test_watershed_no_amplitude():
     eit_data = EITData(