mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-10 19:56:47 +03:00
82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
from tester import *
|
|
|
|
from PIL import Image
|
|
import os
|
|
|
|
base = 'Tests/images/bmp/'
|
|
|
|
|
|
def get_files(d, ext='.bmp'):
|
|
return [os.path.join(base,d,f) for f
|
|
in os.listdir(os.path.join(base, d)) if ext in f]
|
|
|
|
def test_bad():
|
|
""" These shouldn't crash/dos, but they shouldn't return anything either """
|
|
for f in get_files('b'):
|
|
try:
|
|
im = Image.open(f)
|
|
im.load()
|
|
except Exception as msg:
|
|
pass
|
|
# print ("Bad Image %s: %s" %(f,msg))
|
|
|
|
def test_questionable():
|
|
""" These shouldn't crash/dos, but its not well defined that these are in spec """
|
|
for f in get_files('q'):
|
|
try:
|
|
im = Image.open(f)
|
|
im.load()
|
|
except Exception as msg:
|
|
pass
|
|
# print ("Bad Image %s: %s" %(f,msg))
|
|
|
|
|
|
def test_good():
|
|
""" These should all work. There's a set of target files in the
|
|
html directory that we can compare against. """
|
|
|
|
# Target files, if they're not just replacing the extension
|
|
file_map = { 'pal1wb.bmp': 'pal1.png',
|
|
'pal4rle.bmp': 'pal4.png',
|
|
'pal8-0.bmp': 'pal8.png',
|
|
'pal8rle.bmp': 'pal8.png',
|
|
'pal8topdown.bmp': 'pal8.png',
|
|
'pal8nonsquare.bmp': 'pal8nonsquare-v.png',
|
|
'pal8os2.bmp': 'pal8.png',
|
|
'pal8os2sp.bmp': 'pal8.png',
|
|
'pal8os2v2.bmp': 'pal8.png',
|
|
'pal8os2v2-16.bmp': 'pal8.png',
|
|
'pal8v4.bmp': 'pal8.png',
|
|
'pal8v5.bmp': 'pal8.png',
|
|
'rgb16-565pal.bmp': 'rgb16-565.png',
|
|
'rgb24pal.bmp': 'rgb24.png',
|
|
'rgb32.bmp': 'rgb24.png',
|
|
'rgb32bf.bmp': 'rgb24.png'
|
|
}
|
|
|
|
def get_compare(f):
|
|
(head, name) = os.path.split(f)
|
|
if name in file_map:
|
|
return os.path.join(base, 'html', file_map[name])
|
|
(name,ext) = os.path.splitext(name)
|
|
return os.path.join(base, 'html', "%s.png"%name)
|
|
|
|
for f in get_files('g'):
|
|
try:
|
|
im = Image.open(f)
|
|
im.load()
|
|
compare = Image.open(get_compare(f))
|
|
compare.load()
|
|
if im.mode == 'P':
|
|
# assert image similar doesn't really work
|
|
# with paletized image, since the palette might
|
|
# be differently ordered for an equivalent image.
|
|
im = im.convert('RGBA')
|
|
compare = im.convert('RGBA')
|
|
assert_image_similar(im, compare,5)
|
|
|
|
|
|
except Exception as msg:
|
|
print ("Unsupported Image %s: %s" %(f,msg))
|
|
pass
|