def test_uint16(): im16, eroded16, dilated16, opened16, closed16 = map( img_as_uint, [im, eroded, dilated, opened, closed]) cp.testing.assert_allclose(grey.erosion(im16), eroded16) cp.testing.assert_allclose(grey.dilation(im16), dilated16) cp.testing.assert_allclose(grey.opening(im16), opened16) cp.testing.assert_allclose(grey.closing(im16), closed16)
def test_3d_fallback_default_selem(): # 3x3x3 cube inside a 7x7x7 image: image = cp.zeros((7, 7, 7), cp.bool_) image[2:-2, 2:-2, 2:-2] = 1 opened = grey.opening(image) # expect a "hyper-cross" centered in the 5x5x5: image_expected = cp.zeros((7, 7, 7), dtype=bool) image_expected[2:5, 2:5, 2:5] = ndi.generate_binary_structure(3, 1) cp.testing.assert_array_equal(opened, image_expected)
def test_2d_ndimage_equivalence(): image = cp.zeros((9, 9), cp.uint8) image[2:-2, 2:-2] = 128 image[3:-3, 3:-3] = 196 image[4, 4] = 255 opened = grey.opening(image) closed = grey.closing(image) selem = ndi.generate_binary_structure(2, 1) ndimage_opened = ndi.grey_opening(image, footprint=selem) ndimage_closed = ndi.grey_closing(image, footprint=selem) cp.testing.assert_array_equal(opened, ndimage_opened) cp.testing.assert_array_equal(closed, ndimage_closed)
def test_open_white_pixel(self): for s in self.selems: assert cp.all(grey.opening(self.white_pixel, s) == 0)
def test_open_black_pixel(self): for s in self.selems: grey_open = grey.opening(self.black_pixel, s) assert cp.all(grey_open == self.black_pixel)
def test_float(): cp.testing.assert_allclose(grey.erosion(im), eroded) cp.testing.assert_allclose(grey.dilation(im), dilated) cp.testing.assert_allclose(grey.opening(im), opened) cp.testing.assert_allclose(grey.closing(im), closed)
def test_binary_opening(): strel = selem.square(3) binary_res = binary.binary_opening(bw_img, strel) grey_res = img_as_bool(grey.opening(bw_img, strel)) testing.assert_array_equal(binary_res, grey_res)