python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-02-04 13:40:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

Additional tests, tests in functions

Browse Source
This commit is contained in:
wiredfool 2014-06-19 14:55:43 -07:00
parent b95eb3d3d1
commit b8834fa3a6
8 changed files with 109 additions and 80 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
Tests/images/corner.lut Normal file
View File

Binary file not shown.

BIN
Tests/images/dilation4.lut Normal file
View File

Binary file not shown.

BIN
Tests/images/dilation8.lut Normal file
View File

Binary file not shown.

BIN
Tests/images/edge.lut Normal file
View File

Binary file not shown.

BIN
Tests/images/erosion4.lut Normal file
View File

Binary file not shown.

BIN
Tests/images/erosion8.lut Normal file
View File

Binary file not shown.

BIN
Tests/images/morph_a.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 83 B

183
Tests/test_imagemorph.py
View File

@ -4,6 +4,8 @@ from tester import *
from PIL import Image from PIL import Image
from PIL import ImageMorph from PIL import ImageMorph
import binascii
def img_to_string(im): def img_to_string(im):
"""Turn a (small) binary image into a string representation""" """Turn a (small) binary image into a string representation"""
chars = '.1' chars = '.1'
@ -48,91 +50,118 @@ A = string_to_img(
""" """
) )
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()
print (binascii.hexlify(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(binascii.hexlify(lut), binascii.hexlify(bytearray(f.read())))
# Test the named patterns # 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...
.......
.......
.......
""")
# erosion8 def test_dialation8():
mop = ImageMorph.MorphOp(op_name='erosion8') # dialation8
count,Aout = mop.apply(A) mop = ImageMorph.MorphOp(op_name='dilation8')
assert_equal(count,8) count,Aout = mop.apply(A)
assert_img_equal_img_string(Aout, assert_equal(count,16)
""" assert_img_equal_img_string(Aout,
....... """
....... .......
....... .11111.
...1... .11111.
....... .11111.
....... .11111.
....... .11111.
""") .......
""")
# erosion8 def test_erosion4():
mop = ImageMorph.MorphOp(op_name='dilation8') # erosion4
count,Aout = mop.apply(A) mop = ImageMorph.MorphOp(op_name='dilation4')
assert_equal(count,16) count,Aout = mop.apply(A)
assert_img_equal_img_string(Aout, assert_equal(count,12)
""" assert_img_equal_img_string(Aout,
....... """
.11111. .......
.11111. ..111..
.11111. .11111.
.11111. .11111.
.11111. .11111.
....... ..111..
""") .......
""")
# erosion4 def test_edge():
mop = ImageMorph.MorphOp(op_name='dilation4') # edge
count,Aout = mop.apply(A) mop = ImageMorph.MorphOp(op_name='edge')
assert_equal(count,12) count,Aout = mop.apply(A)
assert_img_equal_img_string(Aout, assert_equal(count,1)
""" assert_img_equal_img_string(Aout,
....... """
..111.. .......
.11111. .......
.11111. ..111..
.11111. ..1.1..
..111.. ..111..
....... .......
""") .......
""")
# 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..
.......
.......
""")
# Create a corner detector pattern def test_corner():
mop = ImageMorph.MorphOp(patterns = ['1:(... ... ...)->0', # Create a corner detector pattern
mop = ImageMorph.MorphOp(patterns = ['1:(... ... ...)->0',
'4:(00. 01. ...)->1']) '4:(00. 01. ...)->1'])
count,Aout = mop.apply(A) count,Aout = mop.apply(A)
assert_equal(count,5) assert_equal(count,5)
assert_img_equal_img_string(Aout, assert_img_equal_img_string(Aout,
""" """
....... .......
....... .......
..1.1.. ..1.1..
....... .......
..1.1.. ..1.1..
....... .......
....... .......
""") """)
# Test the coordinate counting with the same operator
coords = mop.match(A) # Test the coordinate counting with the same operator
assert_equal(len(coords), 4) coords = mop.match(A)
assert_equal(tuple(coords), assert_equal(len(coords), 4)
assert_equal(tuple(coords),
((2,2),(4,2),(2,4),(4,4))) ((2,2),(4,2),(2,4),(4,4)))
coords = mop.get_on_pixels(Aout) coords = mop.get_on_pixels(Aout)
assert_equal(len(coords), 4) assert_equal(len(coords), 4)
assert_equal(tuple(coords), assert_equal(tuple(coords),
((2,2),(4,2),(2,4),(4,4))) ((2,2),(4,2),(2,4),(4,4)))