def test_erode_edges_2d(self):
        for img in _generate_test_masks():
            img = label(img)
            img = np.squeeze(img)

            erode_0 = transform_utils.erode_edges(img, erosion_width=0)
            erode_1 = transform_utils.erode_edges(img, erosion_width=1)
            erode_2 = transform_utils.erode_edges(img, erosion_width=2)

            self.assertEqual(img.shape, erode_0.shape)
            self.assertEqual(erode_0.shape, erode_1.shape)
            self.assertEqual(erode_1.shape, erode_2.shape)
            self.assertAllEqual(erode_0, img)
            self.assertGreater(np.sum(erode_0), np.sum(erode_1))
            self.assertGreater(np.sum(erode_1), np.sum(erode_2))

            # test too few dims
            with self.assertRaises(ValueError):
                erode_1 = transform_utils.erode_edges(img[0], erosion_width=1)
Beispiel #2
0
    def test_erode_edges_3d(self):
        mask_stack = np.array(_generate_test_masks())
        unique_mask_stack = np.zeros(mask_stack.shape)

        for i, mask in enumerate(_generate_test_masks()):
            unique_mask_stack[i] = label(mask)

        unique_mask_stack = np.squeeze(unique_mask_stack)

        erode_0 = erode_edges(unique_mask_stack, erosion_width=0)
        erode_1 = erode_edges(unique_mask_stack, erosion_width=1)
        erode_2 = erode_edges(unique_mask_stack, erosion_width=2)

        self.assertEqual(unique_mask_stack.shape, erode_0.shape)
        self.assertEqual(erode_0.shape, erode_1.shape)
        self.assertEqual(erode_1.shape, erode_2.shape)
        self.assertAllEqual(erode_0, unique_mask_stack)
        self.assertGreater(np.sum(erode_0), np.sum(erode_1))
        self.assertGreater(np.sum(erode_1), np.sum(erode_2))

        # test too many dims
        with self.assertRaises(ValueError):
            erode_1 = erode_edges(np.expand_dims(unique_mask_stack, axis=-1), erosion_width=1)