Merge pull request #10 from homm/cache

Return fresh image is no mode specified and cached version otherwise
This commit is contained in:
Hugo 2014-11-27 22:04:42 +02:00
commit 22298f7689

View File

@ -182,22 +182,24 @@ def tostring(im, format, **options):
return out.getvalue() return out.getvalue()
def hopper(mode="RGB", cache={}): def hopper(mode=None, cache={}):
from PIL import Image from PIL import Image
im = None 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 # 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 # returned each time and the cached image isn't modified by tests
# (for fast, isolated, repeatable tests). # (for fast, isolated, repeatable tests).
im = cache.get(mode) im = cache.get(mode)
if im is None: if im is None:
if mode == "RGB": if mode == "F":
im = Image.open("Tests/images/hopper.ppm")
elif mode == "F":
im = hopper("L").convert(mode) im = hopper("L").convert(mode)
elif mode[:4] == "I;16": elif mode[:4] == "I;16":
im = hopper("I").convert(mode) im = hopper("I").convert(mode)
else: else:
im = hopper("RGB").convert(mode) im = hopper().convert(mode)
cache[mode] = im cache[mode] = im
return im.copy() return im.copy()