Corrected test

REDxEYE · Oct 10, 2022 · 982d7c4 · 982d7c4
1 parent f9a3178
commit 982d7c4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Tests/test_image_access.py b/Tests/test_image_access.py
@@ -345,13 +345,14 @@ def test_reference_counting(self):
 
     @pytest.mark.parametrize("mode", ("P", "PA"))
     def test_p_putpixel_rgb_rgba(self, mode):
-        for color in [(255, 0, 0), (255, 0, 0, 127)]:
+        for color in ((255, 0, 0), (255, 0, 0, 127 if mode == "PA" else 255)):
             im = Image.new(mode, (1, 1))
             access = PyAccess.new(im, False)
             access.putpixel((0, 0), color)
 
-            alpha = color[3] if len(color) == 4 and mode == "PA" else 255
-            assert im.convert("RGBA").getpixel((0, 0)) == (255, 0, 0, alpha)
+            if len(color) == 3:
+                color += (255,)
+            assert im.convert("RGBA").getpixel((0, 0)) == color
 
 
 class TestImagePutPixelError(AccessTest):