mirror of
https://github.com/python-pillow/Pillow.git
synced 2025-01-27 17:54:32 +03:00
Converted to unittest framework to deal with test framework error
This commit is contained in:
wiredfool
parent
6dce921dad
commit
cf828f7612
|
|
@ -1,164 +1,169 @@
|
|||
# Test the ImageMorphology functionality
|
||||
from tester import *
|
||||
from helper import *
|
||||
|
||||
from PIL import Image
|
||||
from PIL import ImageMorph
|
||||
|
||||
def img_to_string(im):
|
||||
"""Turn a (small) binary image into a string representation"""
|
||||
chars = '.1'
|
||||
width, height = im.size
|
||||
return '\n'.join(
|
||||
[''.join([chars[im.getpixel((c,r))>0] for c in range(width)])
|
||||
for r in range(height)])
|
||||
|
||||
def string_to_img(image_string):
|
||||
"""Turn a string image representation into a binary image"""
|
||||
rows = [s for s in image_string.replace(' ','').split('\n')
|
||||
if len(s)]
|
||||
height = len(rows)
|
||||
width = len(rows[0])
|
||||
im = Image.new('L',(width,height))
|
||||
for i in range(width):
|
||||
for j in range(height):
|
||||
c = rows[j][i]
|
||||
v = c in 'X1'
|
||||
im.putpixel((i,j),v)
|
||||
class MorphTests(PillowTestCase):
|
||||
|
||||
return im
|
||||
|
||||
def img_string_normalize(im):
|
||||
return img_to_string(string_to_img(im))
|
||||
|
||||
def assert_img_equal(A,B):
|
||||
assert_equal(img_to_string(A), img_to_string(B))
|
||||
|
||||
def assert_img_equal_img_string(A,Bstring):
|
||||
assert_equal(img_to_string(A), img_string_normalize(Bstring))
|
||||
|
||||
A = string_to_img(
|
||||
"""
|
||||
.......
|
||||
.......
|
||||
..111..
|
||||
..111..
|
||||
..111..
|
||||
.......
|
||||
.......
|
||||
"""
|
||||
)
|
||||
|
||||
def test_str_to_img():
|
||||
im = Image.open('Tests/images/morph_a.png')
|
||||
assert_image_equal(A, im)
|
||||
|
||||
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)
|
||||
|
||||
#create_lut()
|
||||
def test_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 , 'rb') as f:
|
||||
assert_equal(lut, bytearray(f.read()))
|
||||
|
||||
def setUp(self):
|
||||
self.A = self.string_to_img(
|
||||
"""
|
||||
.......
|
||||
.......
|
||||
..111..
|
||||
..111..
|
||||
..111..
|
||||
.......
|
||||
.......
|
||||
"""
|
||||
)
|
||||
|
||||
# Test the named patterns
|
||||
def test_erosion8():
|
||||
# erosion8
|
||||
mop = ImageMorph.MorphOp(op_name='erosion8')
|
||||
count,Aout = mop.apply(A)
|
||||
assert_equal(count,8)
|
||||
assert_img_equal_img_string(Aout,
|
||||
"""
|
||||
.......
|
||||
.......
|
||||
.......
|
||||
...1...
|
||||
.......
|
||||
.......
|
||||
.......
|
||||
""")
|
||||
|
||||
def test_dialation8():
|
||||
# dialation8
|
||||
mop = ImageMorph.MorphOp(op_name='dilation8')
|
||||
count,Aout = mop.apply(A)
|
||||
assert_equal(count,16)
|
||||
assert_img_equal_img_string(Aout,
|
||||
"""
|
||||
.......
|
||||
.11111.
|
||||
.11111.
|
||||
.11111.
|
||||
.11111.
|
||||
.11111.
|
||||
.......
|
||||
""")
|
||||
def img_to_string(self, im):
|
||||
"""Turn a (small) binary image into a string representation"""
|
||||
chars = '.1'
|
||||
width, height = im.size
|
||||
return '\n'.join(
|
||||
[''.join([chars[im.getpixel((c,r))>0] for c in range(width)])
|
||||
for r in range(height)])
|
||||
|
||||
def test_erosion4():
|
||||
# erosion4
|
||||
mop = ImageMorph.MorphOp(op_name='dilation4')
|
||||
count,Aout = mop.apply(A)
|
||||
assert_equal(count,12)
|
||||
assert_img_equal_img_string(Aout,
|
||||
"""
|
||||
.......
|
||||
..111..
|
||||
.11111.
|
||||
.11111.
|
||||
.11111.
|
||||
..111..
|
||||
.......
|
||||
""")
|
||||
def string_to_img(self, image_string):
|
||||
"""Turn a string image representation into a binary image"""
|
||||
rows = [s for s in image_string.replace(' ','').split('\n')
|
||||
if len(s)]
|
||||
height = len(rows)
|
||||
width = len(rows[0])
|
||||
im = Image.new('L',(width,height))
|
||||
for i in range(width):
|
||||
for j in range(height):
|
||||
c = rows[j][i]
|
||||
v = c in 'X1'
|
||||
im.putpixel((i,j),v)
|
||||
|
||||
def test_edge():
|
||||
# edge
|
||||
mop = ImageMorph.MorphOp(op_name='edge')
|
||||
count,Aout = mop.apply(A)
|
||||
assert_equal(count,1)
|
||||
assert_img_equal_img_string(Aout,
|
||||
"""
|
||||
.......
|
||||
.......
|
||||
..111..
|
||||
..1.1..
|
||||
..111..
|
||||
.......
|
||||
.......
|
||||
""")
|
||||
return im
|
||||
|
||||
def img_string_normalize(self, im):
|
||||
return self.img_to_string(self.string_to_img(im))
|
||||
|
||||
def assert_img_equal(self, A, B):
|
||||
self.assertEqual(self.img_to_string(A), self.img_to_string(B))
|
||||
|
||||
def assert_img_equal_img_string(self, A, Bstring):
|
||||
self.assertEqual(self.img_to_string(A), self.img_string_normalize(Bstring))
|
||||
|
||||
def test_str_to_img(self):
|
||||
im = Image.open('Tests/images/morph_a.png')
|
||||
self.assert_image_equal(self.A, im)
|
||||
|
||||
def create_lut(self):
|
||||
for op in ('corner', 'dilation4', 'dilation8', 'erosion4', 'erosion8', 'edge'):
|
||||
lb = ImageMorph.LutBuilder(op_name=op)
|
||||
lut = lb.build_lut(self)
|
||||
with open('Tests/images/%s.lut' % op, 'wb') as f:
|
||||
f.write(lut)
|
||||
|
||||
#create_lut()
|
||||
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():
|
||||
# Create a corner detector pattern
|
||||
mop = ImageMorph.MorphOp(patterns = ['1:(... ... ...)->0',
|
||||
'4:(00. 01. ...)->1'])
|
||||
count,Aout = mop.apply(A)
|
||||
assert_equal(count,5)
|
||||
assert_img_equal_img_string(Aout,
|
||||
"""
|
||||
.......
|
||||
.......
|
||||
..1.1..
|
||||
.......
|
||||
..1.1..
|
||||
.......
|
||||
.......
|
||||
""")
|
||||
# Test the named patterns
|
||||
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_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_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..
|
||||
.......
|
||||
.......
|
||||
""")
|
||||
|
||||
|
||||
# Test the coordinate counting with the same operator
|
||||
coords = mop.match(A)
|
||||
assert_equal(len(coords), 4)
|
||||
assert_equal(tuple(coords),
|
||||
((2,2),(4,2),(2,4),(4,4)))
|
||||
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..
|
||||
.......
|
||||
.......
|
||||
""")
|
||||
|
||||
|
||||
coords = mop.get_on_pixels(Aout)
|
||||
assert_equal(len(coords), 4)
|
||||
assert_equal(tuple(coords),
|
||||
((2,2),(4,2),(2,4),(4,4)))
|
||||
# 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)))
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user