Pillow/Tests/test_bmp_reference.py

106 lines
3.9 KiB
Python
Raw Normal View History

2016-11-26 03:05:56 +03:00
from __future__ import print_function
from .helper import PillowTestCase
from PIL import Image
import os
2014-03-20 12:16:31 +04:00
base = os.path.join('Tests', 'images', 'bmp')
2014-06-10 13:10:47 +04:00
class TestBmpReference(PillowTestCase):
2014-06-10 13:10:47 +04:00
def get_files(self, d, ext='.bmp'):
return [os.path.join(base, d, f) for f
in os.listdir(os.path.join(base, d)) if ext in f]
2014-06-10 13:10:47 +04:00
def test_bad(self):
""" These shouldn't crash/dos, but they shouldn't return anything
either """
for f in self.get_files('b'):
2018-11-09 03:35:08 +03:00
def open(f):
try:
im = Image.open(f)
im.load()
except Exception: # as msg:
pass
# Assert that there is no unclosed file warning
self.assert_warning(None, open, f)
2014-06-10 13:10:47 +04:00
def test_questionable(self):
2016-04-13 17:08:44 +03:00
""" These shouldn't crash/dos, but it's not well defined that these
2014-06-10 13:10:47 +04:00
are in spec """
supported = [
"pal8os2v2.bmp",
"rgb24prof.bmp",
"pal1p1.bmp",
"pal8offs.bmp",
"rgb24lprof.bmp",
"rgb32fakealpha.bmp",
"rgb24largepal.bmp",
"pal8os2sp.bmp",
"rgb32bf-xbgr.bmp",
]
2014-06-10 13:10:47 +04:00
for f in self.get_files('q'):
try:
im = Image.open(f)
im.load()
if os.path.basename(f) not in supported:
2018-09-27 13:35:00 +03:00
print("Please add %s to the partially supported"
" bmp specs." % f)
2014-06-10 13:10:47 +04:00
except Exception: # as msg:
if os.path.basename(f) in supported:
raise
2014-06-10 13:10:47 +04:00
def test_good(self):
""" These should all work. There's a set of target files in the
html directory that we can compare against. """
2014-06-10 13:10:47 +04:00
# 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'
}
2014-06-10 13:10:47 +04:00
def get_compare(f):
2015-04-24 11:24:52 +03:00
name = os.path.split(f)[1]
2014-06-10 13:10:47 +04:00
if name in file_map:
return os.path.join(base, 'html', file_map[name])
2015-04-24 11:24:52 +03:00
name = os.path.splitext(name)[0]
2014-06-10 13:10:47 +04:00
return os.path.join(base, 'html', "%s.png" % name)
for f in self.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')
self.assert_image_similar(im, compare, 5)
except Exception as msg:
# there are three here that are unsupported:
unsupported = (os.path.join(base, 'g', 'rgb32bf.bmp'),
os.path.join(base, 'g', 'pal8rle.bmp'),
os.path.join(base, 'g', 'pal4rle.bmp'))
if f not in unsupported:
2017-06-03 07:02:23 +03:00
self.fail("Unsupported Image %s: %s" % (f, msg))