Only specify image size once

Yay295 · Mar 21, 2024 · 9c41bf4 · 9c41bf4
1 parent a0ab9f4
commit 9c41bf4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Tests/test_image.py b/Tests/test_image.py
@@ -1064,8 +1064,9 @@ def test_roundtrip_bytes_method(self, mode: str) -> None:
 
     @pytest.mark.parametrize(("mode", "pixelsize"), image_modes)
     def test_getdata_putdata(self, mode: str, pixelsize: int) -> None:
-        start_bytes = bytes(range(2 * 2 * pixelsize))
-        im = Image.frombytes(mode, (2, 2), start_bytes)
+        im = Image.new(mode, (2, 2))
+        source_bytes = bytes(range(im.width * im.height * pixelsize))
+        im.frombytes(source_bytes)
 
         reloaded = Image.new(mode, im.size)
         reloaded.putdata(im.getdata())