Skip to content

Commit

Permalink
Added test
Browse files Browse the repository at this point in the history
  • Loading branch information
radarhere committed Sep 4, 2017
1 parent 567351f commit 55d0a81
Showing 1 changed file with 20 additions and 4 deletions.
24 changes: 20 additions & 4 deletions Tests/test_image.py
Original file line number Diff line number Diff line change
Expand Up @@ -459,6 +459,22 @@ def test_radial_gradient(self):
target = Image.open(target_file).convert(mode)
self.assert_image_equal(im, target)

def test_register_extensions(self):
test_format = "a"
exts = ["b", "c"]
for ext in exts:
Image.register_extension(test_format, ext)
ext_individual = Image.EXTENSION.copy()
for ext in exts:
del Image.EXTENSION[ext]

Image.register_extensions(test_format, exts)
ext_multiple = Image.EXTENSION.copy()
for ext in exts:
del Image.EXTENSION[ext]

self.assertEqual(ext_individual, ext_multiple)

def test_remap_palette(self):
# Test illegal image mode
im = hopper()
Expand All @@ -467,15 +483,15 @@ def test_remap_palette(self):

def test__new(self):
from PIL import ImagePalette

im = hopper('RGB')
im_p = hopper('P')

blank_p = Image.new('P', (10,10))
blank_pa = Image.new('PA', (10,10))
blank_p.palette = None
blank_pa.palette = None

def _make_new(base_image, im, palette_result=None):
new_im = base_image._new(im)
self.assertEqual(new_im.mode, im.mode)
Expand All @@ -485,12 +501,12 @@ def _make_new(base_image, im, palette_result=None):
self.assertEqual(new_im.palette.tobytes(), palette_result.tobytes())
else:
self.assertEqual(new_im.palette, None)

_make_new(im, im_p, im_p.palette)
_make_new(im_p, im, None)
_make_new(im, blank_p, ImagePalette.ImagePalette())
_make_new(im, blank_pa, ImagePalette.ImagePalette())


class MockEncoder(object):
pass
Expand Down

0 comments on commit 55d0a81

Please sign in to comment.