mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 18:06:18 +03:00
e544fd5fb7
Use the builtin shutil.which() instead of reimplementing. For the single use that used the output of the command, use subprocess.run().
335 lines
8.7 KiB
Python
335 lines
8.7 KiB
Python
"""
|
|
Helper functions.
|
|
"""
|
|
|
|
import logging
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from io import BytesIO
|
|
|
|
import pytest
|
|
from PIL import Image, ImageMath
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
HAS_UPLOADER = False
|
|
|
|
if os.environ.get("SHOW_ERRORS", None):
|
|
# local img.show for errors.
|
|
HAS_UPLOADER = True
|
|
|
|
class test_image_results:
|
|
@staticmethod
|
|
def upload(a, b):
|
|
a.show()
|
|
b.show()
|
|
|
|
|
|
elif "GITHUB_ACTIONS" in os.environ:
|
|
HAS_UPLOADER = True
|
|
|
|
class test_image_results:
|
|
@staticmethod
|
|
def upload(a, b):
|
|
dir_errors = os.path.join(os.path.dirname(__file__), "errors")
|
|
os.makedirs(dir_errors, exist_ok=True)
|
|
tmpdir = tempfile.mkdtemp(dir=dir_errors)
|
|
a.save(os.path.join(tmpdir, "a.png"))
|
|
b.save(os.path.join(tmpdir, "b.png"))
|
|
return tmpdir
|
|
|
|
|
|
else:
|
|
try:
|
|
import test_image_results
|
|
|
|
HAS_UPLOADER = True
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
def convert_to_comparable(a, b):
|
|
new_a, new_b = a, b
|
|
if a.mode == "P":
|
|
new_a = Image.new("L", a.size)
|
|
new_b = Image.new("L", b.size)
|
|
new_a.putdata(a.getdata())
|
|
new_b.putdata(b.getdata())
|
|
elif a.mode == "I;16":
|
|
new_a = a.convert("I")
|
|
new_b = b.convert("I")
|
|
return new_a, new_b
|
|
|
|
|
|
def assert_deep_equal(a, b, msg=None):
|
|
try:
|
|
assert len(a) == len(b), msg or "got length {}, expected {}".format(
|
|
len(a), len(b)
|
|
)
|
|
except Exception:
|
|
assert a == b, msg
|
|
|
|
|
|
def assert_image(im, mode, size, msg=None):
|
|
if mode is not None:
|
|
assert im.mode == mode, msg or "got mode {!r}, expected {!r}".format(
|
|
im.mode, mode
|
|
)
|
|
|
|
if size is not None:
|
|
assert im.size == size, msg or "got size {!r}, expected {!r}".format(
|
|
im.size, size
|
|
)
|
|
|
|
|
|
def assert_image_equal(a, b, msg=None):
|
|
assert a.mode == b.mode, msg or "got mode {!r}, expected {!r}".format(
|
|
a.mode, b.mode
|
|
)
|
|
assert a.size == b.size, msg or "got size {!r}, expected {!r}".format(
|
|
a.size, b.size
|
|
)
|
|
if a.tobytes() != b.tobytes():
|
|
if HAS_UPLOADER:
|
|
try:
|
|
url = test_image_results.upload(a, b)
|
|
logger.error("Url for test images: %s" % url)
|
|
except Exception:
|
|
pass
|
|
|
|
assert False, msg or "got different content"
|
|
|
|
|
|
def assert_image_equal_tofile(a, filename, msg=None, mode=None):
|
|
with Image.open(filename) as img:
|
|
if mode:
|
|
img = img.convert(mode)
|
|
assert_image_equal(a, img, msg)
|
|
|
|
|
|
def assert_image_similar(a, b, epsilon, msg=None):
|
|
assert a.mode == b.mode, msg or "got mode {!r}, expected {!r}".format(
|
|
a.mode, b.mode
|
|
)
|
|
assert a.size == b.size, msg or "got size {!r}, expected {!r}".format(
|
|
a.size, b.size
|
|
)
|
|
|
|
a, b = convert_to_comparable(a, b)
|
|
|
|
diff = 0
|
|
for ach, bch in zip(a.split(), b.split()):
|
|
chdiff = ImageMath.eval("abs(a - b)", a=ach, b=bch).convert("L")
|
|
diff += sum(i * num for i, num in enumerate(chdiff.histogram()))
|
|
|
|
ave_diff = diff / (a.size[0] * a.size[1])
|
|
try:
|
|
assert epsilon >= ave_diff, (
|
|
msg or ""
|
|
) + " average pixel value difference %.4f > epsilon %.4f" % (ave_diff, epsilon)
|
|
except Exception as e:
|
|
if HAS_UPLOADER:
|
|
try:
|
|
url = test_image_results.upload(a, b)
|
|
logger.error("Url for test images: %s" % url)
|
|
except Exception:
|
|
pass
|
|
raise e
|
|
|
|
|
|
def assert_image_similar_tofile(a, filename, epsilon, msg=None, mode=None):
|
|
with Image.open(filename) as img:
|
|
if mode:
|
|
img = img.convert(mode)
|
|
assert_image_similar(a, img, epsilon, msg)
|
|
|
|
|
|
def assert_all_same(items, msg=None):
|
|
assert items.count(items[0]) == len(items), msg
|
|
|
|
|
|
def assert_not_all_same(items, msg=None):
|
|
assert items.count(items[0]) != len(items), msg
|
|
|
|
|
|
def assert_tuple_approx_equal(actuals, targets, threshold, msg):
|
|
"""Tests if actuals has values within threshold from targets"""
|
|
value = True
|
|
for i, target in enumerate(targets):
|
|
value *= target - threshold <= actuals[i] <= target + threshold
|
|
|
|
assert value, msg + ": " + repr(actuals) + " != " + repr(targets)
|
|
|
|
|
|
def skip_known_bad_test(msg=None):
|
|
# Skip if PILLOW_RUN_KNOWN_BAD is not true in the environment.
|
|
if not os.environ.get("PILLOW_RUN_KNOWN_BAD", False):
|
|
pytest.skip(msg or "Known bad test")
|
|
|
|
|
|
class PillowTestCase(unittest.TestCase):
|
|
def delete_tempfile(self, path):
|
|
try:
|
|
os.remove(path)
|
|
except OSError:
|
|
pass # report?
|
|
|
|
def tempfile(self, template):
|
|
assert template[:5] in ("temp.", "temp_")
|
|
fd, path = tempfile.mkstemp(template[4:], template[:4])
|
|
os.close(fd)
|
|
|
|
self.addCleanup(self.delete_tempfile, path)
|
|
return path
|
|
|
|
def open_withImagemagick(self, f):
|
|
if not imagemagick_available():
|
|
raise OSError()
|
|
|
|
outfile = self.tempfile("temp.png")
|
|
rc = subprocess.call(
|
|
[IMCONVERT, f, outfile], stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT
|
|
)
|
|
if rc:
|
|
raise OSError
|
|
return Image.open(outfile)
|
|
|
|
|
|
@unittest.skipIf(sys.platform.startswith("win32"), "requires Unix or macOS")
|
|
class PillowLeakTestCase(PillowTestCase):
|
|
# requires unix/macOS
|
|
iterations = 100 # count
|
|
mem_limit = 512 # k
|
|
|
|
def _get_mem_usage(self):
|
|
"""
|
|
Gets the RUSAGE memory usage, returns in K. Encapsulates the difference
|
|
between macOS and Linux rss reporting
|
|
|
|
:returns: memory usage in kilobytes
|
|
"""
|
|
|
|
from resource import getrusage, RUSAGE_SELF
|
|
|
|
mem = getrusage(RUSAGE_SELF).ru_maxrss
|
|
if sys.platform == "darwin":
|
|
# man 2 getrusage:
|
|
# ru_maxrss
|
|
# This is the maximum resident set size utilized (in bytes).
|
|
return mem / 1024 # Kb
|
|
else:
|
|
# linux
|
|
# man 2 getrusage
|
|
# ru_maxrss (since Linux 2.6.32)
|
|
# This is the maximum resident set size used (in kilobytes).
|
|
return mem # Kb
|
|
|
|
def _test_leak(self, core):
|
|
start_mem = self._get_mem_usage()
|
|
for cycle in range(self.iterations):
|
|
core()
|
|
mem = self._get_mem_usage() - start_mem
|
|
msg = "memory usage limit exceeded in iteration %d" % cycle
|
|
self.assertLess(mem, self.mem_limit, msg)
|
|
|
|
|
|
# helpers
|
|
|
|
|
|
def fromstring(data):
|
|
return Image.open(BytesIO(data))
|
|
|
|
|
|
def tostring(im, string_format, **options):
|
|
out = BytesIO()
|
|
im.save(out, string_format, **options)
|
|
return out.getvalue()
|
|
|
|
|
|
def hopper(mode=None, cache={}):
|
|
if mode is None:
|
|
# Always return fresh not-yet-loaded version of image.
|
|
# Operations on not-yet-loaded images is separate class of errors
|
|
# what we should catch.
|
|
return Image.open("Tests/images/hopper.ppm")
|
|
# Use caching to reduce reading from disk but so an original copy is
|
|
# returned each time and the cached image isn't modified by tests
|
|
# (for fast, isolated, repeatable tests).
|
|
im = cache.get(mode)
|
|
if im is None:
|
|
if mode == "F":
|
|
im = hopper("L").convert(mode)
|
|
elif mode[:4] == "I;16":
|
|
im = hopper("I").convert(mode)
|
|
else:
|
|
im = hopper().convert(mode)
|
|
cache[mode] = im
|
|
return im.copy()
|
|
|
|
|
|
def djpeg_available():
|
|
return bool(shutil.which("djpeg"))
|
|
|
|
|
|
def cjpeg_available():
|
|
return bool(shutil.which("cjpeg"))
|
|
|
|
|
|
def netpbm_available():
|
|
return bool(shutil.which("ppmquant") and shutil.which("ppmtogif"))
|
|
|
|
|
|
def imagemagick_available():
|
|
return bool(IMCONVERT and shutil.which(IMCONVERT))
|
|
|
|
|
|
def on_appveyor():
|
|
return "APPVEYOR" in os.environ
|
|
|
|
|
|
def on_github_actions():
|
|
return "GITHUB_ACTIONS" in os.environ
|
|
|
|
|
|
def on_ci():
|
|
# Travis and AppVeyor have "CI"
|
|
# Azure Pipelines has "TF_BUILD"
|
|
# GitHub Actions has "GITHUB_ACTIONS"
|
|
return (
|
|
"CI" in os.environ or "TF_BUILD" in os.environ or "GITHUB_ACTIONS" in os.environ
|
|
)
|
|
|
|
|
|
def is_big_endian():
|
|
return sys.byteorder == "big"
|
|
|
|
|
|
def is_win32():
|
|
return sys.platform.startswith("win32")
|
|
|
|
|
|
def is_pypy():
|
|
return hasattr(sys, "pypy_translation_info")
|
|
|
|
|
|
if sys.platform == "win32":
|
|
IMCONVERT = os.environ.get("MAGICK_HOME", "")
|
|
if IMCONVERT:
|
|
IMCONVERT = os.path.join(IMCONVERT, "convert.exe")
|
|
else:
|
|
IMCONVERT = "convert"
|
|
|
|
|
|
class cached_property:
|
|
def __init__(self, func):
|
|
self.func = func
|
|
|
|
def __get__(self, instance, cls=None):
|
|
result = instance.__dict__[self.func.__name__] = self.func(instance)
|
|
return result
|