def setUp(self): self.black_pixel = 255 * cp.ones((4, 4), dtype=cp.uint8) self.black_pixel[1, 1] = 0 self.white_pixel = 255 - self.black_pixel self.selems = [ selem.square(2), selem.rectangle(2, 2), selem.rectangle(2, 1), selem.rectangle(1, 2), ]
def test_square_selem(self): """Test square structuring elements""" for k in range(0, 5): actual_mask = selem.square(k) expected_mask = cp.ones((k, k), dtype='uint8') assert_array_equal(expected_mask, actual_mask)
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)
def test_non_square_image(): strel = selem.square(3) binary_res = binary.binary_erosion(bw_img[:100, :200], strel) grey_res = img_as_bool(grey.erosion(bw_img[:100, :200], strel)) testing.assert_array_equal(binary_res, grey_res)