Pillow/Tests/test_file_gif.py

88 lines
2.1 KiB
Python
Raw Normal View History

from tester import *
from PIL import Image
codecs = dir(Image.core)
if "gif_encoder" not in codecs or "gif_decoder" not in codecs:
skip("gif support not available") # can this happen?
# sample gif stream
file = "Images/lena.gif"
2013-05-23 21:02:19 +04:00
with open(file, "rb") as f:
data = f.read()
def test_sanity():
im = Image.open(file)
im.load()
assert_equal(im.mode, "P")
assert_equal(im.size, (128, 128))
assert_equal(im.format, "GIF")
def test_optimize():
def test(optimize):
im = Image.new("L", (1, 1), 0)
file = BytesIO()
im.save(file, "GIF", optimize=optimize)
return len(file.getvalue())
assert_equal(test(0), 800)
assert_equal(test(1), 38)
2013-11-08 04:39:36 +04:00
def test_roundtrip():
out = tempfile('temp.gif')
im = lena()
im.save(out)
reread = Image.open(out)
assert_image_similar(reread.convert('RGB'), im, 50)
2013-11-08 04:39:57 +04:00
def test_roundtrip2():
#see https://github.com/python-pillow/Pillow/issues/403
2014-03-05 10:02:22 +04:00
out = tempfile('temp.gif')
2013-11-08 04:39:57 +04:00
im = Image.open('Images/lena.gif')
im2 = im.copy()
im2.save(out)
reread = Image.open(out)
assert_image_similar(reread.convert('RGB'), lena(), 50)
2014-03-05 10:29:55 +04:00
2014-03-05 10:02:03 +04:00
def test_palette_handling():
# see https://github.com/python-pillow/Pillow/issues/513
2014-03-05 10:02:03 +04:00
im = Image.open('Images/lena.gif')
im = im.convert('RGB')
2014-03-05 10:02:03 +04:00
im = im.resize((100,100), Image.ANTIALIAS)
im2 = im.convert('P', palette=Image.ADAPTIVE, colors=256)
f = tempfile('temp.gif')
im2.save(f, optimize=True)
reloaded = Image.open(f)
2014-03-05 10:02:03 +04:00
assert_image_similar(im, reloaded.convert('RGB'), 10)
2014-03-05 10:29:55 +04:00
def test_palette_434():
# see https://github.com/python-pillow/Pillow/issues/434
2014-03-05 10:29:55 +04:00
def roundtrip(im, *args, **kwargs):
out = tempfile('temp.gif')
im.save(out, *args, **kwargs)
reloaded = Image.open(out)
return [im, reloaded]
orig = "Tests/images/test.colors.gif"
im = Image.open(orig)
assert_image_equal(*roundtrip(im))
assert_image_equal(*roundtrip(im, optimize=True))
2014-03-05 10:29:55 +04:00
im = im.convert("RGB")
# check automatic P conversion
reloaded = roundtrip(im)[1].convert('RGB')
assert_image_equal(im, reloaded)