Merge pull request #6159 from radarhere/jpeg_zero

Tests/test_file_jpeg.py

@@ -68,6 +68,13 @@ def test_sanity(self):
             assert im.format == "JPEG"
             assert im.get_format_mimetype() == "image/jpeg"
 
+    @pytest.mark.parametrize("size", ((1, 0), (0, 1), (0, 0)))
+    def test_zero(self, size, tmp_path):
+        f = str(tmp_path / "temp.jpg")
+        im = Image.new("RGB", size)
+        with pytest.raises(ValueError):
+            im.save(f)
+
     def test_app(self):
         # Test APP/COM reader (@PIL135)
         with Image.open(TEST_FILE) as im:

Tests/test_image.py

@@ -656,7 +656,7 @@ def test_no_new_file_on_error(self, tmp_path):
         temp_file = str(tmp_path / "temp.jpg")
 
         im = Image.new("RGB", (0, 0))
-        with pytest.raises(SystemError):
+        with pytest.raises(ValueError):
             im.save(temp_file)
 
         assert not os.path.exists(temp_file)

src/PIL/JpegImagePlugin.py

@@ -626,6 +626,8 @@ def get_sampling(im):
 
 
 def _save(im, fp, filename):
+    if im.width == 0 or im.height == 0:
+        raise ValueError("cannot write empty image as JPEG")
 
     try:
         rawmode = RAWMODE[im.mode]