Updated error message

bigcat88 · Oct 10, 2022 · 0b2cef5 · 0b2cef5
1 parent 982d7c4
commit 0b2cef5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 11 deletions.
diff --git a/Tests/test_imagepalette.py b/Tests/test_imagepalette.py
@@ -50,11 +50,12 @@ def test_getcolor():
         palette.getcolor("unknown")
 
 
-def test_getcolor_raises_on_incompatible_color():
-    palette = ImagePalette.ImagePalette(mode="RGB")
-    # Opaque RGBA colors should work
-    palette.getcolor((0, 0, 0, 255))
-    assert palette.getcolor((0, 0, 0)) == palette.getcolor((0, 0, 0, 255))
+def test_getcolor_rgba_color_rgb_palette():
+    palette = ImagePalette.ImagePalette("RGB")
+
+    # Opaque RGBA colors are converted
+    assert palette.getcolor((0, 0, 0, 255)) == palette.getcolor((0, 0, 0))
+
     with pytest.raises(ValueError):
         palette.getcolor((0, 0, 0, 128))
 

diff --git a/src/PIL/ImagePalette.py b/src/PIL/ImagePalette.py
@@ -114,13 +114,13 @@ def getcolor(self, color, image=None):
         if self.rawmode:
             raise ValueError("palette contains raw palette data")
         if isinstance(color, tuple):
-            if self.mode == "RGB" and len(color) == 4:
-                if color[3] == 255:
+            if self.mode == "RGB":
+                if len(color) == 4:
+                    if color[3] != 255:
+                        raise ValueError(
+                            "cannot add non-opaque RGBA color to RGB palette"
+                        )
                     color = color[:3]
-                else:
-                    raise ValueError(
-                        "RGB ImagePalette can't handle non-opaque RGBA colors"
-                    )
             elif self.mode == "RGBA":
                 if len(color) == 3:
                     color += (255,)