def checkAlgorithmSameObjects(image): # check if same objects after and before pruning skel = Skeleton(image) skel.setPrunedSkeletonOutput() label_img, countObjectsn = ndimage.measurements.label(skel.outputStack, structure=np.ones((3, 3, 3), dtype=np.uint8)) label_img, countObjects = ndimage.measurements.label(image, structure=np.ones((3, 3, 3), dtype=np.uint8)) assert (countObjectsn == countObjects) or np.sum(label_img == 2) == 1