Merge pull request #1 from radarhere/patch-2

Added test and flush file pointer
scaramallion · Jul 30, 2021 · c42e6ba · c42e6ba
2 parents feb3103 + 9125631
commit c42e6ba
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
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