mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-14 05:36:48 +03:00
115 lines
3.6 KiB
Python
115 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import warnings
|
|
|
|
from PIL import Image
|
|
|
|
from .helper import assert_image_similar
|
|
|
|
base = os.path.join("Tests", "images", "bmp")
|
|
|
|
|
|
def get_files(d: str, ext: str = ".bmp") -> list[str]:
|
|
return [
|
|
os.path.join(base, d, f) for f in os.listdir(os.path.join(base, d)) if ext in f
|
|
]
|
|
|
|
|
|
def test_bad() -> None:
|
|
"""These shouldn't crash/dos, but they shouldn't return anything
|
|
either"""
|
|
for f in get_files("b"):
|
|
# Assert that there is no unclosed file warning
|
|
with warnings.catch_warnings():
|
|
try:
|
|
with Image.open(f) as im:
|
|
im.load()
|
|
except Exception: # as msg:
|
|
pass
|
|
|
|
|
|
def test_questionable() -> None:
|
|
"""These shouldn't crash/dos, but it's not well defined that these
|
|
are in spec"""
|
|
supported = [
|
|
"pal8os2v2.bmp",
|
|
"rgb24prof.bmp",
|
|
"pal1p1.bmp",
|
|
"pal4rletrns.bmp",
|
|
"pal8offs.bmp",
|
|
"rgb24lprof.bmp",
|
|
"rgb32fakealpha.bmp",
|
|
"rgb24largepal.bmp",
|
|
"pal8os2sp.bmp",
|
|
"pal8rletrns.bmp",
|
|
"rgb32bf-xbgr.bmp",
|
|
"rgba32.bmp",
|
|
"rgb32h52.bmp",
|
|
"rgba32h56.bmp",
|
|
]
|
|
for f in get_files("q"):
|
|
try:
|
|
with Image.open(f) as im:
|
|
im.load()
|
|
if os.path.basename(f) not in supported:
|
|
print(f"Please add {f} to the partially supported bmp specs.")
|
|
except Exception: # as msg:
|
|
if os.path.basename(f) in supported:
|
|
raise
|
|
|
|
|
|
def test_good() -> None:
|
|
"""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: str) -> str:
|
|
name = os.path.split(f)[1]
|
|
if name in file_map:
|
|
return os.path.join(base, "html", file_map[name])
|
|
name = os.path.splitext(name)[0]
|
|
return os.path.join(base, "html", f"{name}.png")
|
|
|
|
for f in get_files("g"):
|
|
try:
|
|
with Image.open(f) as im:
|
|
im.load()
|
|
with Image.open(get_compare(f)) as compare:
|
|
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:
|
|
# 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"),
|
|
)
|
|
assert f in unsupported, f"Unsupported Image {f}: {msg}"
|