def test_set_lut(): # Arrange lb = ImageMorph.LutBuilder(op_name="corner") lut = lb.build_lut() mop = ImageMorph.MorphOp() # Act mop.set_lut(lut) # Assert assert mop.lut == lut
def test_set_lut(self): # Arrange lb = ImageMorph.LutBuilder(op_name='corner') lut = lb.build_lut() mop = ImageMorph.MorphOp() # Act mop.set_lut(lut) # Assert self.assertEqual(mop.lut, lut)
def processImage(path): im = Image.open(path) print("Path:", path) print("Format:", im.format) print("Dimensions:", im.size) print("Mode:", im.mode) # Gerando níveis de dimensões diferentes dimensions = [] # Utilizei potencias de dois por arbitrariedade mesmo for j in [2**i for i in range(5)]: dimensions.append((im.width / j, im.height / j)) print(dimensions[-1]) im_all = [] for (width, height) in dimensions: aux = im.resize((int(width), int(height))) im_all.append(aux.resize((im.width, im.height))) map_char = [] for im in im_all: for it in im_all: if im != it: map_char.append(ImageChops.difference(im, it)) result = map_char[0] for img in map_char[1:]: result = ImageChops.add(result, img, scale=2.0) # get an image # make a blank image for the text, initialized to transparent text color txt = Image.new('RGBA', im.size, (255, 255, 255, 0)) # get a font fnt = ImageFont.truetype('./Roboto-Black.ttf', 30) # get a drawing context d = ImageDraw.Draw(txt) grayscale = result.convert('L') im = Image.open(path).convert('RGBA') out = im for i in range(1, 35): _, maxima = grayscale.getextrema() mask_image = grayscale.point(lambda i: 255 if i == maxima else 0) mask = ImageMorph.MorphOp(lut=ImageMorph.LutBuilder( patterns=["1:(000 010 000)->1"]).build_lut()) match = mask.match(mask_image) d.text(match[0], str(i), font=fnt, fill=(255, 255, 255, 200)) out = Image.alpha_composite(im, txt) supress = Image.new('L', (20, 20), "black") (w, h) = match[0] grayscale.paste(supress, (w - 10, h - 10)) out.show()
def test_non_binary_images(self): im = hopper('RGB') mop = ImageMorph.MorphOp(op_name="erosion8") self.assertRaises(Exception, mop.apply, im) self.assertRaises(Exception, mop.match, im) self.assertRaises(Exception, mop.get_on_pixels, im)
def test_load_invalid_mrl(self): # Arrange invalid_mrl = 'Tests/images/hopper.png' mop = ImageMorph.MorphOp() # Act / Assert self.assertRaises(Exception, mop.load_lut, invalid_mrl)
def test_corner(): # Create a corner detector pattern mop = ImageMorph.MorphOp( patterns=["1:(... ... ...)->0", "4:(00. 01. ...)->1"]) count, Aout = mop.apply(A) assert count == 5 assert_img_equal_img_string( Aout, """ ....... ....... ..1.1.. ....... ..1.1.. ....... ....... """, ) # Test the coordinate counting with the same operator coords = mop.match(A) assert len(coords) == 4 assert tuple(coords) == ((2, 2), (4, 2), (2, 4), (4, 4)) coords = mop.get_on_pixels(Aout) assert len(coords) == 4 assert tuple(coords) == ((2, 2), (4, 2), (2, 4), (4, 4))
def test_lut(self): for op in ('corner', 'dilation4', 'dilation8', 'erosion4', 'erosion8', 'edge'): lb = ImageMorph.LutBuilder(op_name=op) lut = lb.build_lut() with open('Tests/images/%s.lut' % op, 'rb') as f: self.assertEqual(lut, bytearray(f.read()))
def test_corner(self): # Create a corner detector pattern mop = ImageMorph.MorphOp( patterns=['1:(... ... ...)->0', '4:(00. 01. ...)->1']) count, Aout = mop.apply(self.A) self.assertEqual(count, 5) self.assert_img_equal_img_string( Aout, """ ....... ....... ..1.1.. ....... ..1.1.. ....... ....... """) # Test the coordinate counting with the same operator coords = mop.match(self.A) self.assertEqual(len(coords), 4) self.assertEqual(tuple(coords), ((2, 2), (4, 2), (2, 4), (4, 4))) coords = mop.get_on_pixels(Aout) self.assertEqual(len(coords), 4) self.assertEqual(tuple(coords), ((2, 2), (4, 2), (2, 4), (4, 4)))
def create_lut(self): for op in ('corner', 'dilation4', 'dilation8', 'erosion4', 'erosion8', 'edge'): lb = ImageMorph.LutBuilder(op_name=op) lut = lb.build_lut() with open('Tests/images/%s.lut' % op, 'wb') as f: f.write(lut)
def create_lut(): for op in ("corner", "dilation4", "dilation8", "erosion4", "erosion8", "edge"): lb = ImageMorph.LutBuilder(op_name=op) lut = lb.build_lut() with open("Tests/images/%s.lut" % op, "wb") as f: f.write(lut)
def test_pattern_syntax_error(self): # Arrange lb = ImageMorph.LutBuilder(op_name='corner') new_patterns = ['a pattern with a syntax error'] lb.add_patterns(new_patterns) # Act / Assert self.assertRaises(Exception, lb.build_lut)
def test_lut(self): for op in ("corner", "dilation4", "dilation8", "erosion4", "erosion8", "edge"): lb = ImageMorph.LutBuilder(op_name=op) self.assertIsNone(lb.get_lut()) lut = lb.build_lut() with open("Tests/images/%s.lut" % op, "rb") as f: self.assertEqual(lut, bytearray(f.read()))
def test_load_invalid_mrl(): # Arrange invalid_mrl = "Tests/images/hopper.png" mop = ImageMorph.MorphOp() # Act / Assert with pytest.raises(Exception) as e: mop.load_lut(invalid_mrl) assert str(e.value) == "Wrong size operator file!"
def test_load_invalid_mrl(self): # Arrange invalid_mrl = 'Tests/images/hopper.png' mop = ImageMorph.MorphOp() # Act / Assert with self.assertRaises(Exception) as e: mop.load_lut(invalid_mrl) self.assertEqual(str(e.exception), 'Wrong size operator file!')
def test_lut(): for op in ("corner", "dilation4", "dilation8", "erosion4", "erosion8", "edge"): lb = ImageMorph.LutBuilder(op_name=op) assert lb.get_lut() is None lut = lb.build_lut() with open("Tests/images/%s.lut" % op, "rb") as f: assert lut == bytearray(f.read())
def count_regions(key): image=draw_disk(key) on_pixels=ImageMorph.MorphOp().get_on_pixels(image.convert('L')) image=image.convert('RGB') region_count=0 for x,y in on_pixels: if image.getpixel((x,y))!=(0,0,0): ImageDraw.floodfill(image,(x,y),(0,0,0)) region_count+=1 return region_count
def test_no_operator_loaded(self): mop = ImageMorph.MorphOp() with self.assertRaises(Exception) as e: mop.apply(None) self.assertEqual(str(e.exception), 'No operator loaded') with self.assertRaises(Exception) as e: mop.match(None) self.assertEqual(str(e.exception), 'No operator loaded') with self.assertRaises(Exception) as e: mop.save_lut(None) self.assertEqual(str(e.exception), 'No operator loaded')
def test_no_operator_loaded(): mop = ImageMorph.MorphOp() with pytest.raises(Exception) as e: mop.apply(None) assert str(e.value) == "No operator loaded" with pytest.raises(Exception) as e: mop.match(None) assert str(e.value) == "No operator loaded" with pytest.raises(Exception) as e: mop.save_lut(None) assert str(e.value) == "No operator loaded"
def test_pattern_syntax_error(): # Arrange lb = ImageMorph.LutBuilder(op_name="corner") new_patterns = ["a pattern with a syntax error"] lb.add_patterns(new_patterns) # Act / Assert with pytest.raises(Exception) as e: lb.build_lut() assert str( e.value) == 'Syntax error in pattern "a pattern with a syntax error"'
def test_roundtrip_mrl(self): # Arrange tempfile = self.tempfile('temp.mrl') mop = ImageMorph.MorphOp(op_name='corner') initial_lut = mop.lut # Act mop.save_lut(tempfile) mop.load_lut(tempfile) # Act / Assert self.assertEqual(mop.lut, initial_lut)
def test_pattern_syntax_error(self): # Arrange lb = ImageMorph.LutBuilder(op_name='corner') new_patterns = ['a pattern with a syntax error'] lb.add_patterns(new_patterns) # Act / Assert with self.assertRaises(Exception) as e: lb.build_lut() self.assertEqual( str(e.exception), 'Syntax error in pattern "a pattern with a syntax error"')
def test_roundtrip_mrl(tmp_path): # Arrange tempfile = str(tmp_path / "temp.mrl") mop = ImageMorph.MorphOp(op_name="corner") initial_lut = mop.lut # Act mop.save_lut(tempfile) mop.load_lut(tempfile) # Act / Assert assert mop.lut == initial_lut
def test_incorrect_mode(): im = hopper("RGB") mop = ImageMorph.MorphOp(op_name="erosion8") with pytest.raises(ValueError) as e: mop.apply(im) assert str(e.value) == "Image mode must be L" with pytest.raises(ValueError) as e: mop.match(im) assert str(e.value) == "Image mode must be L" with pytest.raises(ValueError) as e: mop.get_on_pixels(im) assert str(e.value) == "Image mode must be L"
def test_non_binary_images(): im = hopper("RGB") mop = ImageMorph.MorphOp(op_name="erosion8") with pytest.raises(Exception) as e: mop.apply(im) assert str(e.value) == "Image must be binary, meaning it must use mode L" with pytest.raises(Exception) as e: mop.match(im) assert str(e.value) == "Image must be binary, meaning it must use mode L" with pytest.raises(Exception) as e: mop.get_on_pixels(im) assert str(e.value) == "Image must be binary, meaning it must use mode L"
def test_erosion4(self): # erosion4 mop = ImageMorph.MorphOp(op_name='dilation4') count, Aout = mop.apply(self.A) self.assertEqual(count, 12) self.assert_img_equal_img_string( Aout, """ ....... ..111.. .11111. .11111. .11111. ..111.. ....... """)
def test_edge(self): # edge mop = ImageMorph.MorphOp(op_name='edge') count, Aout = mop.apply(self.A) self.assertEqual(count, 1) self.assert_img_equal_img_string( Aout, """ ....... ....... ..111.. ..1.1.. ..111.. ....... ....... """)
def test_add_patterns(self): # Arrange lb = ImageMorph.LutBuilder(op_name='corner') self.assertEqual(lb.patterns, ['1:(... ... ...)->0', '4:(00. 01. ...)->1']) new_patterns = ['M:(00. 01. ...)->1', 'N:(00. 01. ...)->1'] # Act lb.add_patterns(new_patterns) # Assert self.assertEqual(lb.patterns, [ '1:(... ... ...)->0', '4:(00. 01. ...)->1', 'M:(00. 01. ...)->1', 'N:(00. 01. ...)->1' ])
def test_dialation8(self): # dialation8 mop = ImageMorph.MorphOp(op_name='dilation8') count, Aout = mop.apply(self.A) self.assertEqual(count, 16) self.assert_img_equal_img_string( Aout, """ ....... .11111. .11111. .11111. .11111. .11111. ....... """)
def test_erosion8(self): # erosion8 mop = ImageMorph.MorphOp(op_name='erosion8') count, Aout = mop.apply(self.A) self.assertEqual(count, 8) self.assert_img_equal_img_string( Aout, """ ....... ....... ....... ...1... ....... ....... ....... """)
def test_wrong_mode(): lut = ImageMorph.LutBuilder(op_name="corner").build_lut() imrgb = Image.new("RGB", (10, 10)) iml = Image.new("L", (10, 10)) with pytest.raises(RuntimeError): _imagingmorph.apply(bytes(lut), imrgb.im.id, iml.im.id) with pytest.raises(RuntimeError): _imagingmorph.apply(bytes(lut), iml.im.id, imrgb.im.id) with pytest.raises(RuntimeError): _imagingmorph.match(bytes(lut), imrgb.im.id) # Should not raise _imagingmorph.match(bytes(lut), iml.im.id)