Fixed linear_gradient I mode

python-pillow · Feb 18, 2021 · 553ce81 · 553ce81
1 parent 1857bf5
commit 553ce81
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/Tests/test_image.py b/Tests/test_image.py
@@ -521,7 +521,7 @@ def test_linear_gradient(self):
 
         # Arrange
         target_file = "Tests/images/linear_gradient.png"
-        for mode in ["L", "P"]:
+        for mode in ["L", "P", "I"]:
 
             # Act
             im = Image.linear_gradient(mode)

diff --git a/src/libImaging/Fill.c b/src/libImaging/Fill.c
@@ -76,8 +76,17 @@ ImagingFillLinearGradient(const char *mode) {
         return NULL;
     }
 
-    for (y = 0; y < 256; y++) {
-        memset(im->image8[y], (unsigned char)y, 256);
+    if (im->image8) {
+        for (y = 0; y < 256; y++) {
+            memset(im->image8[y], (unsigned char)y, 256);
+        }
+    } else {
+        int x;
+        for (y = 0; y < 256; y++) {
+            for (x = 0; x < 256; x++) {
+                im->image32[y][x] = y;
+            }
+        }
     }
 
     return im;