Enhance the test for the fix.

This commit is contained in:
Moriyoshi Koizumi 2014-11-06 19:10:36 +09:00
parent 9570864163
commit d00c3c02ad

View File

@ -28,13 +28,22 @@ class TestFileGif(PillowTestCase):
def test_optimize(self): def test_optimize(self):
from io import BytesIO from io import BytesIO
def test(optimize): def test_grayscale(optimize):
im = Image.new("L", (1, 1), 0) im = Image.new("L", (1, 1), 0)
file = BytesIO() file = BytesIO()
im.save(file, "GIF", optimize=optimize) im.save(file, "GIF", optimize=optimize)
return len(file.getvalue()) return len(file.getvalue())
self.assertEqual(test(0), 800)
self.assertEqual(test(1), 38) def test_bilevel(optimize):
im = Image.new("1", (1, 1), 0)
file = BytesIO()
im.save(file, "GIF", optimize=optimize)
return len(file.getvalue())
self.assertEqual(test_grayscale(0), 800)
self.assertEqual(test_grayscale(1), 38)
self.assertEqual(test_bilevel(0), 800)
self.assertEqual(test_bilevel(1), 800)
def test_optimize_full_l(self): def test_optimize_full_l(self):
from io import BytesIO from io import BytesIO