fix(sv_split): update multicut test

Author: akhileshh

pychunkedgraph/app/segmentation/common.py

@@ -472,8 +472,8 @@ def handle_split(table_id):
         )
         overlap_mask = np.isin(sources_remapped, sinks_remapped)
         for sv_to_split in np.unique(sources_remapped[overlap_mask]):
-            _mask0 = sources_remapped[sources_remapped == sv_to_split]
-            _mask1 = sinks_remapped[sinks_remapped == sv_to_split]
+            _mask0 = sources_remapped == sv_to_split
+            _mask1 = sinks_remapped == sv_to_split
             split_supervoxel(
                 cg,
                 sv_to_split,

pychunkedgraph/graph/edits_sv.py

@@ -56,10 +56,11 @@ def _get_whole_sv(
         mask = _cx_edges[:, 0] == vertex
         neighbors = _cx_edges[mask][:, 1]
 
-        neighbor_coords = cg.get_chunk_coordinates_multiple(neighbors)
-        min_mask = (neighbor_coords >= min_coord).all(axis=1)
-        max_mask = (neighbor_coords < max_coord).all(axis=1)
-        neighbors = neighbors[min_mask & max_mask]
+        if len(neighbors) > 0:
+            neighbor_coords = cg.get_chunk_coordinates_multiple(neighbors)
+            min_mask = (neighbor_coords >= min_coord).all(axis=1)
+            max_mask = (neighbor_coords < max_coord).all(axis=1)
+            neighbors = neighbors[min_mask & max_mask]
 
         for neighbor in neighbors:
             if neighbor in explored_nodes:

pychunkedgraph/tests/test_uncategorized.py

@@ -1872,7 +1872,7 @@ def test_path_augmented_multicut(self, sv_data):
         cut_edges_aug = run_multicut(edges, sv_sources, sv_sinks, path_augment=True)
         assert cut_edges_aug.shape[0] == 350
 
-        with pytest.raises(exceptions.PreconditionError):
+        with pytest.raises(exceptions.SupervoxelSplitRequiredError):
             run_multicut(edges, sv_sources, sv_sinks, path_augment=False)