Merge pull request #1 from radarhere/patch-2

farizrahman4u · web-flow · commit c42e6bab1c60 · 2021-07-31T01:05:42.000+04:00
Added test and flush file pointer
diff --git a/Tests/test_file_sgi.py b/Tests/test_file_sgi.py
@@ -73,6 +73,13 @@ def roundtrip(img):
         img.save(out, format="sgi")
         assert_image_equal_tofile(img, out)
 
+        out = str(tmp_path / "fp.sgi")
+        with open(out, "wb") as fp:
+            img.save(fp)
+            assert_image_equal_tofile(img, out)
+
+            assert not fp.closed
+
     for mode in ("L", "RGB", "RGBA"):
         roundtrip(hopper(mode))
 
diff --git a/src/PIL/SgiImagePlugin.py b/src/PIL/SgiImagePlugin.py
@@ -193,6 +193,9 @@ def _save(im, fp, filename):
     for channel in im.split():
         fp.write(channel.tobytes("raw", rawmode, 0, orientation))
 
+    if hasattr(fp, "flush"):
+        fp.flush()
+
 
 class SGI16Decoder(ImageFile.PyDecoder):
     _pulls_fd = True
