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