def test_wrong_parameters_2():
image = numpy.zeros((25, 25, 3), dtype=numpy.uint8)
try:
clean_noise(image)
except Exception, e:
assert e.message == 'binary_image must be 2D array'
assert type(e) == ValueError
def test_wrong_parameters_4():
image = numpy.zeros((25, 25), dtype=numpy.uint8)
mask = numpy.zeros((25, 25, 3), dtype=numpy.uint8)
try:
clean_noise(image, mask=mask)
except Exception, e:
assert e.message == 'mask must be 2D array'
assert type(e) == ValueError
def test_simply_working_1():
image = numpy.zeros((25, 25), dtype=numpy.uint8)
mask = numpy.zeros((25, 25), dtype=numpy.uint8)
image_cleaning = clean_noise(image, mask=mask)
assert isinstance(image_cleaning, numpy.ndarray)
assert image_cleaning.ndim == 2
def test_wrong_parameters_1():
try:
clean_noise(None)
except Exception, e:
assert e.message == 'binary_image must be a numpy.ndarray'
assert type(e) == TypeError
