Merge pull request #5551 from radarhere/imageops

python-pillow · Jun 20, 2021 · e5b2d61 · e5b2d61
2 parents 93871c0 + 6639b2a
commit e5b2d61
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Tests/test_imageops.py b/Tests/test_imageops.py
@@ -156,6 +156,12 @@ def test_scale():
     assert newimg.size == (25, 25)
 
 
+def test_expand_palette():
+    im = Image.open("Tests/images/hopper.gif")
+    im_expanded = ImageOps.expand(im)
+    assert_image_equal(im_expanded.convert("RGB"), im.convert("RGB"))
+
+
 def test_colorize_2color():
     # Test the colorizing function with 2-color functionality
 

diff --git a/src/PIL/ImageOps.py b/src/PIL/ImageOps.py
@@ -393,6 +393,8 @@ def expand(image, border=0, fill=0):
     width = left + image.size[0] + right
     height = top + image.size[1] + bottom
     out = Image.new(image.mode, (width, height), _color(fill, image.mode))
+    if image.mode == "P" and image.palette:
+        out.putpalette(image.palette)
     out.paste(image, (left, top))
     return out