make sure passing a blank comment removes existing comment

bigcat88 · Dec 3, 2022 · 1ed1a3a · 1ed1a3a
1 parent e71f7c1
commit 1ed1a3a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Tests/test_file_jpeg.py b/Tests/test_file_jpeg.py
@@ -98,6 +98,13 @@ def test_comment_write(self):
             with Image.open(out) as reloaded:
                 assert im.info["comment"] == reloaded.info["comment"]
 
+            # Ensure that a blank comment causes any existing comment to be removed
+            for comment in ("", b"", None):
+                out = BytesIO()
+                im.save(out, format="JPEG", comment=comment)
+                with Image.open(out) as reloaded:
+                    assert "comment" not in reloaded.info
+
             # Test that a comment argument overrides the default comment
             for comment in ("Test comment text", b"Text comment text"):
                 out = BytesIO()