def test_close_wrong_parameters_2():
image = numpy.zeros((25, 25, 3), dtype=numpy.uint8)
try:
phm_img.close(image)
except Exception, e:
assert e.message == 'image must be 2D array'
assert type(e) == ValueError
def test_close_wrong_parameters_1():
try:
phm_img.close(None)
except Exception as e:
assert str(e) == 'image must be a numpy.ndarray'
assert type(e) == TypeError
else:
assert False
def test_close_wrong_parameters_3():
image = numpy.zeros((25, 25), dtype=numpy.uint8)
mask = 42
try:
phm_img.close(image, mask=mask)
except Exception, e:
assert e.message == 'mask must be a numpy.ndarray'
assert type(e) == TypeError
def test_close_wrong_parameters_4():
image = numpy.zeros((25, 25), dtype=numpy.uint8)
mask = numpy.zeros((25, 25, 3), dtype=numpy.uint8)
try:
phm_img.close(image, mask=mask)
except Exception as e:
assert str(e) == 'mask must be 2D array'
assert type(e) == ValueError
else:
assert False
def test_close_2():
image = numpy.zeros((25, 25), dtype=numpy.uint8)
image_cleaning = phm_img.close(image)
assert isinstance(image_cleaning, numpy.ndarray)
assert image_cleaning.ndim == 2
def test_close_wrong_parameters_1():
try:
phm_img.close(None)
except Exception, e:
assert e.message == 'image must be a numpy.ndarray'
assert type(e) == TypeError
