Merge pull request #5275 from radarhere/tofile

This commit is contained in:
Hugo van Kemenade 2021-02-21 17:10:52 +02:00 committed by GitHub
commit 058b8d3d12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
36 changed files with 310 additions and 392 deletions

View File

@ -1,21 +1,18 @@
from PIL import Image from PIL import Image
from .helper import assert_image_equal from .helper import assert_image_equal_tofile
def test_load_blp2_raw(): def test_load_blp2_raw():
with Image.open("Tests/images/blp/blp2_raw.blp") as im: with Image.open("Tests/images/blp/blp2_raw.blp") as im:
with Image.open("Tests/images/blp/blp2_raw.png") as target: assert_image_equal_tofile(im, "Tests/images/blp/blp2_raw.png")
assert_image_equal(im, target)
def test_load_blp2_dxt1(): def test_load_blp2_dxt1():
with Image.open("Tests/images/blp/blp2_dxt1.blp") as im: with Image.open("Tests/images/blp/blp2_dxt1.blp") as im:
with Image.open("Tests/images/blp/blp2_dxt1.png") as target: assert_image_equal_tofile(im, "Tests/images/blp/blp2_dxt1.png")
assert_image_equal(im, target)
def test_load_blp2_dxt1a(): def test_load_blp2_dxt1a():
with Image.open("Tests/images/blp/blp2_dxt1a.blp") as im: with Image.open("Tests/images/blp/blp2_dxt1a.blp") as im:
with Image.open("Tests/images/blp/blp2_dxt1a.png") as target: assert_image_equal_tofile(im, "Tests/images/blp/blp2_dxt1a.png")
assert_image_equal(im, target)

View File

@ -4,7 +4,7 @@ import pytest
from PIL import BmpImagePlugin, Image from PIL import BmpImagePlugin, Image
from .helper import assert_image_equal, hopper from .helper import assert_image_equal, assert_image_equal_tofile, hopper
def test_sanity(tmp_path): def test_sanity(tmp_path):
@ -111,8 +111,7 @@ def test_load_dib():
assert im.format == "DIB" assert im.format == "DIB"
assert im.get_format_mimetype() == "image/bmp" assert im.get_format_mimetype() == "image/bmp"
with Image.open("Tests/images/clipboard_target.png") as target: assert_image_equal_tofile(im, "Tests/images/clipboard_target.png")
assert_image_equal(im, target)
def test_save_dib(tmp_path): def test_save_dib(tmp_path):
@ -136,5 +135,4 @@ def test_rgba_bitfields():
b, g, r = im.split()[1:] b, g, r = im.split()[1:]
im = Image.merge("RGB", (r, g, b)) im = Image.merge("RGB", (r, g, b))
with Image.open("Tests/images/bmp/q/rgb32bf-xbgr.bmp") as target: assert_image_equal_tofile(im, "Tests/images/bmp/q/rgb32bf-xbgr.bmp")
assert_image_equal(im, target)

View File

@ -5,7 +5,7 @@ import pytest
from PIL import DdsImagePlugin, Image from PIL import DdsImagePlugin, Image
from .helper import assert_image_equal from .helper import assert_image_equal, assert_image_equal_tofile
TEST_FILE_DXT1 = "Tests/images/dxt1-rgb-4bbp-noalpha_MipMaps-1.dds" TEST_FILE_DXT1 = "Tests/images/dxt1-rgb-4bbp-noalpha_MipMaps-1.dds"
TEST_FILE_DXT3 = "Tests/images/dxt3-argb-8bbp-explicitalpha_MipMaps-1.dds" TEST_FILE_DXT3 = "Tests/images/dxt3-argb-8bbp-explicitalpha_MipMaps-1.dds"
@ -41,22 +41,20 @@ def test_sanity_dxt5():
assert im.mode == "RGBA" assert im.mode == "RGBA"
assert im.size == (256, 256) assert im.size == (256, 256)
with Image.open(TEST_FILE_DXT5.replace(".dds", ".png")) as target: assert_image_equal_tofile(im, TEST_FILE_DXT5.replace(".dds", ".png"))
assert_image_equal(target, im)
def test_sanity_dxt3(): def test_sanity_dxt3():
"""Check DXT3 images can be opened""" """Check DXT3 images can be opened"""
with Image.open(TEST_FILE_DXT3.replace(".dds", ".png")) as target: with Image.open(TEST_FILE_DXT3) as im:
with Image.open(TEST_FILE_DXT3) as im: im.load()
im.load()
assert im.format == "DDS" assert im.format == "DDS"
assert im.mode == "RGBA" assert im.mode == "RGBA"
assert im.size == (256, 256) assert im.size == (256, 256)
assert_image_equal(target, im) assert_image_equal_tofile(im, TEST_FILE_DXT3.replace(".dds", ".png"))
def test_dx10_bc7(): def test_dx10_bc7():
@ -69,8 +67,7 @@ def test_dx10_bc7():
assert im.mode == "RGBA" assert im.mode == "RGBA"
assert im.size == (256, 256) assert im.size == (256, 256)
with Image.open(TEST_FILE_DX10_BC7.replace(".dds", ".png")) as target: assert_image_equal_tofile(im, TEST_FILE_DX10_BC7.replace(".dds", ".png"))
assert_image_equal(target, im)
def test_dx10_bc7_unorm_srgb(): def test_dx10_bc7_unorm_srgb():
@ -84,10 +81,9 @@ def test_dx10_bc7_unorm_srgb():
assert im.size == (16, 16) assert im.size == (16, 16)
assert im.info["gamma"] == 1 / 2.2 assert im.info["gamma"] == 1 / 2.2
with Image.open( assert_image_equal_tofile(
TEST_FILE_DX10_BC7_UNORM_SRGB.replace(".dds", ".png") im, TEST_FILE_DX10_BC7_UNORM_SRGB.replace(".dds", ".png")
) as target: )
assert_image_equal(target, im)
def test_dx10_r8g8b8a8(): def test_dx10_r8g8b8a8():
@ -100,8 +96,7 @@ def test_dx10_r8g8b8a8():
assert im.mode == "RGBA" assert im.mode == "RGBA"
assert im.size == (256, 256) assert im.size == (256, 256)
with Image.open(TEST_FILE_DX10_R8G8B8A8.replace(".dds", ".png")) as target: assert_image_equal_tofile(im, TEST_FILE_DX10_R8G8B8A8.replace(".dds", ".png"))
assert_image_equal(target, im)
def test_dx10_r8g8b8a8_unorm_srgb(): def test_dx10_r8g8b8a8_unorm_srgb():
@ -115,10 +110,9 @@ def test_dx10_r8g8b8a8_unorm_srgb():
assert im.size == (16, 16) assert im.size == (16, 16)
assert im.info["gamma"] == 1 / 2.2 assert im.info["gamma"] == 1 / 2.2
with Image.open( assert_image_equal_tofile(
TEST_FILE_DX10_R8G8B8A8_UNORM_SRGB.replace(".dds", ".png") im, TEST_FILE_DX10_R8G8B8A8_UNORM_SRGB.replace(".dds", ".png")
) as target: )
assert_image_equal(target, im)
def test_unimplemented_dxgi_format(): def test_unimplemented_dxgi_format():
@ -137,8 +131,9 @@ def test_uncompressed_rgb():
assert im.mode == "RGBA" assert im.mode == "RGBA"
assert im.size == (800, 600) assert im.size == (800, 600)
with Image.open(TEST_FILE_UNCOMPRESSED_RGB.replace(".dds", ".png")) as target: assert_image_equal_tofile(
assert_image_equal(target, im) im, TEST_FILE_UNCOMPRESSED_RGB.replace(".dds", ".png")
)
def test__validate_true(): def test__validate_true():

View File

@ -4,7 +4,12 @@ import pytest
from PIL import EpsImagePlugin, Image, features from PIL import EpsImagePlugin, Image, features
from .helper import assert_image_similar, hopper, skip_unless_feature from .helper import (
assert_image_similar,
assert_image_similar_tofile,
hopper,
skip_unless_feature,
)
HAS_GHOSTSCRIPT = EpsImagePlugin.has_ghostscript() HAS_GHOSTSCRIPT = EpsImagePlugin.has_ghostscript()
@ -72,8 +77,9 @@ def test_cmyk():
assert cmyk_image.mode == "RGB" assert cmyk_image.mode == "RGB"
if features.check("jpg"): if features.check("jpg"):
with Image.open("Tests/images/pil_sample_rgb.jpg") as target: assert_image_similar_tofile(
assert_image_similar(cmyk_image, target, 10) cmyk_image, "Tests/images/pil_sample_rgb.jpg", 10
)
@pytest.mark.skipif(not HAS_GHOSTSCRIPT, reason="Ghostscript not available") @pytest.mark.skipif(not HAS_GHOSTSCRIPT, reason="Ghostscript not available")

View File

@ -2,7 +2,7 @@ import pytest
from PIL import FliImagePlugin, Image from PIL import FliImagePlugin, Image
from .helper import assert_image_equal, is_pypy from .helper import assert_image_equal_tofile, is_pypy
# created as an export of a palette image from Gimp2.6 # created as an export of a palette image from Gimp2.6
# save as...-> hopper.fli, default options. # save as...-> hopper.fli, default options.
@ -122,5 +122,4 @@ def test_seek():
with Image.open(animated_test_file) as im: with Image.open(animated_test_file) as im:
im.seek(50) im.seek(50)
with Image.open("Tests/images/a_fli.png") as expected: assert_image_equal_tofile(im, "Tests/images/a_fli.png")
assert_image_equal(im, expected)

View File

@ -1,12 +1,11 @@
from PIL import Image from PIL import Image
from .helper import assert_image_equal, assert_image_similar from .helper import assert_image_equal_tofile, assert_image_similar
def test_load_raw(): def test_load_raw():
with Image.open("Tests/images/ftex_uncompressed.ftu") as im: with Image.open("Tests/images/ftex_uncompressed.ftu") as im:
with Image.open("Tests/images/ftex_uncompressed.png") as target: assert_image_equal_tofile(im, "Tests/images/ftex_uncompressed.png")
assert_image_equal(im, target)
def test_load_dxt1(): def test_load_dxt1():

View File

@ -2,7 +2,7 @@ import pytest
from PIL import GbrImagePlugin, Image from PIL import GbrImagePlugin, Image
from .helper import assert_image_equal from .helper import assert_image_equal_tofile
def test_invalid_file(): def test_invalid_file():
@ -14,13 +14,11 @@ def test_invalid_file():
def test_gbr_file(): def test_gbr_file():
with Image.open("Tests/images/gbr.gbr") as im: with Image.open("Tests/images/gbr.gbr") as im:
with Image.open("Tests/images/gbr.png") as target: assert_image_equal_tofile(im, "Tests/images/gbr.png")
assert_image_equal(target, im)
def test_multiple_load_operations(): def test_multiple_load_operations():
with Image.open("Tests/images/gbr.gbr") as im: with Image.open("Tests/images/gbr.gbr") as im:
im.load() im.load()
im.load() im.load()
with Image.open("Tests/images/gbr.png") as target: assert_image_equal_tofile(im, "Tests/images/gbr.png")
assert_image_equal(target, im)

View File

@ -6,6 +6,7 @@ from PIL import GifImagePlugin, Image, ImageDraw, ImagePalette, features
from .helper import ( from .helper import (
assert_image_equal, assert_image_equal,
assert_image_equal_tofile,
assert_image_similar, assert_image_similar,
hopper, hopper,
is_pypy, is_pypy,
@ -317,8 +318,7 @@ def test_dispose_none_load_end():
with Image.open("Tests/images/dispose_none_load_end.gif") as img: with Image.open("Tests/images/dispose_none_load_end.gif") as img:
img.seek(1) img.seek(1)
with Image.open("Tests/images/dispose_none_load_end_second.gif") as expected: assert_image_equal_tofile(img, "Tests/images/dispose_none_load_end_second.gif")
assert_image_equal(img, expected)
def test_dispose_background(): def test_dispose_background():
@ -629,8 +629,7 @@ def test_comment_over_255(tmp_path):
def test_zero_comment_subblocks(): def test_zero_comment_subblocks():
with Image.open("Tests/images/hopper_zero_comment_subblocks.gif") as im: with Image.open("Tests/images/hopper_zero_comment_subblocks.gif") as im:
with Image.open(TEST_GIF) as expected: assert_image_equal_tofile(im, TEST_GIF)
assert_image_equal(im, expected)
def test_version(tmp_path): def test_version(tmp_path):

View File

@ -5,7 +5,7 @@ import pytest
from PIL import IcnsImagePlugin, Image, features from PIL import IcnsImagePlugin, Image, features
from .helper import assert_image_equal, assert_image_similar from .helper import assert_image_equal, assert_image_similar_tofile
# sample icon file # sample icon file
TEST_FILE = "Tests/images/pillow.icns" TEST_FILE = "Tests/images/pillow.icns"
@ -49,8 +49,7 @@ def test_save_append_images(tmp_path):
with Image.open(TEST_FILE) as im: with Image.open(TEST_FILE) as im:
im.save(temp_file, append_images=[provided_im]) im.save(temp_file, append_images=[provided_im])
with Image.open(temp_file) as reread: assert_image_similar_tofile(im, temp_file, 1)
assert_image_similar(reread, im, 1)
with Image.open(temp_file) as reread: with Image.open(temp_file) as reread:
reread.size = (16, 16, 2) reread.size = (16, 16, 2)

View File

@ -4,7 +4,7 @@ import pytest
from PIL import IcoImagePlugin, Image, ImageDraw from PIL import IcoImagePlugin, Image, ImageDraw
from .helper import assert_image_equal, hopper from .helper import assert_image_equal, assert_image_equal_tofile, hopper
TEST_ICO_FILE = "Tests/images/hopper.ico" TEST_ICO_FILE = "Tests/images/hopper.ico"
@ -120,5 +120,4 @@ def test_draw_reloaded(tmp_path):
with Image.open(outfile) as im: with Image.open(outfile) as im:
im.save("Tests/images/hopper_draw.ico") im.save("Tests/images/hopper_draw.ico")
with Image.open("Tests/images/hopper_draw.ico") as reloaded: assert_image_equal_tofile(im, "Tests/images/hopper_draw.ico")
assert_image_equal(im, reloaded)

View File

@ -4,7 +4,7 @@ import pytest
from PIL import Image, ImImagePlugin from PIL import Image, ImImagePlugin
from .helper import assert_image_equal, hopper, is_pypy from .helper import assert_image_equal_tofile, hopper, is_pypy
# sample im # sample im
TEST_IM = "Tests/images/hopper.im" TEST_IM = "Tests/images/hopper.im"
@ -86,8 +86,7 @@ def test_roundtrip(tmp_path):
out = str(tmp_path / "temp.im") out = str(tmp_path / "temp.im")
im = hopper(mode) im = hopper(mode)
im.save(out) im.save(out)
with Image.open(out) as reread: assert_image_equal_tofile(im, out)
assert_image_equal(reread, im)
for mode in ["RGB", "P", "PA"]: for mode in ["RGB", "P", "PA"]:
roundtrip(mode) roundtrip(mode)

View File

@ -17,7 +17,9 @@ from PIL import (
from .helper import ( from .helper import (
assert_image, assert_image,
assert_image_equal, assert_image_equal,
assert_image_equal_tofile,
assert_image_similar, assert_image_similar,
assert_image_similar_tofile,
cjpeg_available, cjpeg_available,
djpeg_available, djpeg_available,
hopper, hopper,
@ -577,7 +579,7 @@ class TestFileJpeg:
def test_load_djpeg(self): def test_load_djpeg(self):
with Image.open(TEST_FILE) as img: with Image.open(TEST_FILE) as img:
img.load_djpeg() img.load_djpeg()
assert_image_similar(img, Image.open(TEST_FILE), 5) assert_image_similar_tofile(img, TEST_FILE, 5)
@pytest.mark.skipif(not cjpeg_available(), reason="cjpeg not available") @pytest.mark.skipif(not cjpeg_available(), reason="cjpeg not available")
def test_save_cjpeg(self, tmp_path): def test_save_cjpeg(self, tmp_path):
@ -585,7 +587,7 @@ class TestFileJpeg:
tempfile = str(tmp_path / "temp.jpg") tempfile = str(tmp_path / "temp.jpg")
JpegImagePlugin._save_cjpeg(img, 0, tempfile) JpegImagePlugin._save_cjpeg(img, 0, tempfile)
# Default save quality is 75%, so a tiny bit of difference is alright # Default save quality is 75%, so a tiny bit of difference is alright
assert_image_similar(img, Image.open(tempfile), 17) assert_image_similar_tofile(img, tempfile, 17)
def test_no_duplicate_0x1001_tag(self): def test_no_duplicate_0x1001_tag(self):
# Arrange # Arrange
@ -767,8 +769,7 @@ class TestFileJpeg:
# Test that the image can still load, even with broken Photoshop data # Test that the image can still load, even with broken Photoshop data
# This image had the APP13 length hexedited to be smaller # This image had the APP13 length hexedited to be smaller
with Image.open("Tests/images/photoshop-200dpi-broken.jpg") as im_broken: assert_image_equal_tofile(im, "Tests/images/photoshop-200dpi-broken.jpg")
assert_image_equal(im_broken, im)
# This image does not contain a Photoshop header string # This image does not contain a Photoshop header string
with Image.open("Tests/images/app13.jpg") as im: with Image.open("Tests/images/app13.jpg") as im:

View File

@ -8,6 +8,7 @@ from PIL import Image, ImageFile, Jpeg2KImagePlugin, features
from .helper import ( from .helper import (
assert_image_equal, assert_image_equal,
assert_image_similar, assert_image_similar,
assert_image_similar_tofile,
is_big_endian, is_big_endian,
skip_unless_feature, skip_unless_feature,
) )
@ -62,9 +63,7 @@ def test_invalid_file():
def test_bytesio(): def test_bytesio():
with open("Tests/images/test-card-lossless.jp2", "rb") as f: with open("Tests/images/test-card-lossless.jp2", "rb") as f:
data = BytesIO(f.read()) data = BytesIO(f.read())
with Image.open(data) as im: assert_image_similar_tofile(test_card, data, 1.0e-3)
im.load()
assert_image_similar(im, test_card, 1.0e-3)
# These two test pre-written JPEG 2000 files that were not written with # These two test pre-written JPEG 2000 files that were not written with
@ -80,9 +79,9 @@ def test_lossless(tmp_path):
def test_lossy_tiled(): def test_lossy_tiled():
with Image.open("Tests/images/test-card-lossy-tiled.jp2") as im: assert_image_similar_tofile(
im.load() test_card, "Tests/images/test-card-lossy-tiled.jp2", 2.0
assert_image_similar(im, test_card, 2.0) )
def test_lossless_rt(): def test_lossless_rt():
@ -193,15 +192,13 @@ def test_16bit_monochrome_has_correct_mode():
@pytest.mark.xfail(is_big_endian(), reason="Fails on big-endian") @pytest.mark.xfail(is_big_endian(), reason="Fails on big-endian")
def test_16bit_monochrome_jp2_like_tiff(): def test_16bit_monochrome_jp2_like_tiff():
with Image.open("Tests/images/16bit.cropped.tif") as tiff_16bit: with Image.open("Tests/images/16bit.cropped.tif") as tiff_16bit:
with Image.open("Tests/images/16bit.cropped.jp2") as jp2: assert_image_similar_tofile(tiff_16bit, "Tests/images/16bit.cropped.jp2", 1e-3)
assert_image_similar(jp2, tiff_16bit, 1e-3)
@pytest.mark.xfail(is_big_endian(), reason="Fails on big-endian") @pytest.mark.xfail(is_big_endian(), reason="Fails on big-endian")
def test_16bit_monochrome_j2k_like_tiff(): def test_16bit_monochrome_j2k_like_tiff():
with Image.open("Tests/images/16bit.cropped.tif") as tiff_16bit: with Image.open("Tests/images/16bit.cropped.tif") as tiff_16bit:
with Image.open("Tests/images/16bit.cropped.j2k") as j2k: assert_image_similar_tofile(tiff_16bit, "Tests/images/16bit.cropped.j2k", 1e-3)
assert_image_similar(j2k, tiff_16bit, 1e-3)
def test_16bit_j2k_roundtrips(): def test_16bit_j2k_roundtrips():

View File

@ -99,15 +99,13 @@ class TestFileLibTiff(LibTiffTestCase):
def test_g4_eq_png(self): def test_g4_eq_png(self):
""" Checking that we're actually getting the data that we expect""" """ Checking that we're actually getting the data that we expect"""
with Image.open("Tests/images/hopper_bw_500.png") as png: with Image.open("Tests/images/hopper_bw_500.png") as png:
with Image.open("Tests/images/hopper_g4_500.tif") as g4: assert_image_equal_tofile(png, "Tests/images/hopper_g4_500.tif")
assert_image_equal(g4, png)
# see https://github.com/python-pillow/Pillow/issues/279 # see https://github.com/python-pillow/Pillow/issues/279
def test_g4_fillorder_eq_png(self): def test_g4_fillorder_eq_png(self):
""" Checking that we're actually getting the data that we expect""" """ Checking that we're actually getting the data that we expect"""
with Image.open("Tests/images/g4-fillorder-test.png") as png: with Image.open("Tests/images/g4-fillorder-test.tif") as g4:
with Image.open("Tests/images/g4-fillorder-test.tif") as g4: assert_image_equal_tofile(g4, "Tests/images/g4-fillorder-test.png")
assert_image_equal(g4, png)
def test_g4_write(self, tmp_path): def test_g4_write(self, tmp_path):
"""Checking to see that the saved image is the same as what we wrote""" """Checking to see that the saved image is the same as what we wrote"""
@ -437,10 +435,7 @@ class TestFileLibTiff(LibTiffTestCase):
im = im.filter(ImageFilter.GaussianBlur(4)) im = im.filter(ImageFilter.GaussianBlur(4))
im.save(out, compression="tiff_adobe_deflate") im.save(out, compression="tiff_adobe_deflate")
with Image.open(out) as im2: assert_image_equal_tofile(im, out)
im2.load()
assert_image_equal(im, im2)
def test_compressions(self, tmp_path): def test_compressions(self, tmp_path):
# Test various tiff compressions and assert similar image content but reduced # Test various tiff compressions and assert similar image content but reduced
@ -453,8 +448,7 @@ class TestFileLibTiff(LibTiffTestCase):
for compression in ("packbits", "tiff_lzw"): for compression in ("packbits", "tiff_lzw"):
im.save(out, compression=compression) im.save(out, compression=compression)
size_compressed = os.path.getsize(out) size_compressed = os.path.getsize(out)
with Image.open(out) as im2: assert_image_equal_tofile(im, out)
assert_image_equal(im, im2)
im.save(out, compression="jpeg") im.save(out, compression="jpeg")
size_jpeg = os.path.getsize(out) size_jpeg = os.path.getsize(out)
@ -463,8 +457,7 @@ class TestFileLibTiff(LibTiffTestCase):
im.save(out, compression="jpeg", quality=30) im.save(out, compression="jpeg", quality=30)
size_jpeg_30 = os.path.getsize(out) size_jpeg_30 = os.path.getsize(out)
with Image.open(out) as im3: assert_image_similar_tofile(im2, out, 30)
assert_image_similar(im2, im3, 30)
assert size_raw > size_compressed assert size_raw > size_compressed
assert size_compressed > size_jpeg assert size_compressed > size_jpeg
@ -498,8 +491,7 @@ class TestFileLibTiff(LibTiffTestCase):
out = str(tmp_path / "temp.tif") out = str(tmp_path / "temp.tif")
im.save(out, compression="tiff_adobe_deflate") im.save(out, compression="tiff_adobe_deflate")
with Image.open(out) as im2: assert_image_equal_tofile(im, out)
assert_image_equal(im, im2)
def test_palette_save(self, tmp_path): def test_palette_save(self, tmp_path):
im = hopper("P") im = hopper("P")
@ -649,8 +641,7 @@ class TestFileLibTiff(LibTiffTestCase):
pilim.save(buffer_io, format="tiff", compression=compression) pilim.save(buffer_io, format="tiff", compression=compression)
buffer_io.seek(0) buffer_io.seek(0)
with Image.open(buffer_io) as pilim_load: assert_image_similar_tofile(pilim, buffer_io, 0)
assert_image_similar(pilim, pilim_load, 0)
save_bytesio() save_bytesio()
save_bytesio("raw") save_bytesio("raw")

View File

@ -2,7 +2,7 @@ import pytest
from PIL import Image, McIdasImagePlugin from PIL import Image, McIdasImagePlugin
from .helper import assert_image_equal from .helper import assert_image_equal_tofile
def test_invalid_file(): def test_invalid_file():
@ -27,5 +27,4 @@ def test_valid_file():
assert im.format == "MCIDAS" assert im.format == "MCIDAS"
assert im.mode == "I" assert im.mode == "I"
assert im.size == (1800, 400) assert im.size == (1800, 400)
with Image.open(saved_file) as im2: assert_image_equal_tofile(im, saved_file)
assert_image_equal(im, im2)

View File

@ -4,7 +4,7 @@ import pytest
from PIL import Image, MspImagePlugin from PIL import Image, MspImagePlugin
from .helper import assert_image_equal, hopper from .helper import assert_image_equal, assert_image_equal_tofile, hopper
TEST_FILE = "Tests/images/hopper.msp" TEST_FILE = "Tests/images/hopper.msp"
EXTRA_DIR = "Tests/images/picins" EXTRA_DIR = "Tests/images/picins"
@ -52,8 +52,7 @@ def test_open_windows_v1():
def _assert_file_image_equal(source_path, target_path): def _assert_file_image_equal(source_path, target_path):
with Image.open(source_path) as im: with Image.open(source_path) as im:
with Image.open(target_path) as target: assert_image_equal_tofile(im, target_path)
assert_image_equal(im, target)
@pytest.mark.skipif( @pytest.mark.skipif(

View File

@ -2,7 +2,7 @@ import pytest
from PIL import Image from PIL import Image
from .helper import assert_image_equal, assert_image_similar, hopper from .helper import assert_image_equal_tofile, assert_image_similar, hopper
# sample ppm stream # sample ppm stream
TEST_FILE = "Tests/images/hopper.ppm" TEST_FILE = "Tests/images/hopper.ppm"
@ -24,8 +24,7 @@ def test_16bit_pgm():
assert im.size == (20, 100) assert im.size == (20, 100)
assert im.get_format_mimetype() == "image/x-portable-graymap" assert im.get_format_mimetype() == "image/x-portable-graymap"
with Image.open("Tests/images/16_bit_binary_pgm.png") as tgt: assert_image_equal_tofile(im, "Tests/images/16_bit_binary_pgm.png")
assert_image_equal(im, tgt)
def test_16bit_pgm_write(tmp_path): def test_16bit_pgm_write(tmp_path):
@ -35,8 +34,7 @@ def test_16bit_pgm_write(tmp_path):
f = str(tmp_path / "temp.pgm") f = str(tmp_path / "temp.pgm")
im.save(f, "PPM") im.save(f, "PPM")
with Image.open(f) as reloaded: assert_image_equal_tofile(im, f)
assert_image_equal(im, reloaded)
def test_pnm(tmp_path): def test_pnm(tmp_path):
@ -46,8 +44,7 @@ def test_pnm(tmp_path):
f = str(tmp_path / "temp.pnm") f = str(tmp_path / "temp.pnm")
im.save(f) im.save(f)
with Image.open(f) as reloaded: assert_image_equal_tofile(im, f)
assert_image_equal(im, reloaded)
def test_truncated_file(tmp_path): def test_truncated_file(tmp_path):

View File

@ -2,7 +2,12 @@ import pytest
from PIL import Image, SgiImagePlugin from PIL import Image, SgiImagePlugin
from .helper import assert_image_equal, assert_image_similar, hopper from .helper import (
assert_image_equal,
assert_image_equal_tofile,
assert_image_similar,
hopper,
)
def test_rgb(): def test_rgb():
@ -16,10 +21,7 @@ def test_rgb():
def test_rgb16(): def test_rgb16():
test_file = "Tests/images/hopper16.rgb" assert_image_equal_tofile(hopper(), "Tests/images/hopper16.rgb")
with Image.open(test_file) as im:
assert_image_equal(im, hopper())
def test_l(): def test_l():
@ -38,8 +40,7 @@ def test_rgba():
test_file = "Tests/images/transparent.sgi" test_file = "Tests/images/transparent.sgi"
with Image.open(test_file) as im: with Image.open(test_file) as im:
with Image.open("Tests/images/transparent.png") as target: assert_image_equal_tofile(im, "Tests/images/transparent.png")
assert_image_equal(im, target)
assert im.get_format_mimetype() == "image/sgi" assert im.get_format_mimetype() == "image/sgi"
@ -49,16 +50,14 @@ def test_rle():
test_file = "Tests/images/hopper.sgi" test_file = "Tests/images/hopper.sgi"
with Image.open(test_file) as im: with Image.open(test_file) as im:
with Image.open("Tests/images/hopper.rgb") as target: assert_image_equal_tofile(im, "Tests/images/hopper.rgb")
assert_image_equal(im, target)
def test_rle16(): def test_rle16():
test_file = "Tests/images/tv16.sgi" test_file = "Tests/images/tv16.sgi"
with Image.open(test_file) as im: with Image.open(test_file) as im:
with Image.open("Tests/images/tv.rgb") as target: assert_image_equal_tofile(im, "Tests/images/tv.rgb")
assert_image_equal(im, target)
def test_invalid_file(): def test_invalid_file():
@ -72,8 +71,7 @@ def test_write(tmp_path):
def roundtrip(img): def roundtrip(img):
out = str(tmp_path / "temp.sgi") out = str(tmp_path / "temp.sgi")
img.save(out, format="sgi") img.save(out, format="sgi")
with Image.open(out) as reloaded: assert_image_equal_tofile(img, out)
assert_image_equal(img, reloaded)
for mode in ("L", "RGB", "RGBA"): for mode in ("L", "RGB", "RGBA"):
roundtrip(hopper(mode)) roundtrip(hopper(mode))
@ -89,8 +87,7 @@ def test_write16(tmp_path):
out = str(tmp_path / "temp.sgi") out = str(tmp_path / "temp.sgi")
im.save(out, format="sgi", bpc=2) im.save(out, format="sgi", bpc=2)
with Image.open(out) as reloaded: assert_image_equal_tofile(im, out)
assert_image_equal(im, reloaded)
def test_unsupported_mode(tmp_path): def test_unsupported_mode(tmp_path):

View File

@ -5,7 +5,7 @@ import pytest
from PIL import Image, ImageSequence, SpiderImagePlugin from PIL import Image, ImageSequence, SpiderImagePlugin
from .helper import assert_image_equal, hopper, is_pypy from .helper import assert_image_equal_tofile, hopper, is_pypy
TEST_FILE = "Tests/images/hopper.spider" TEST_FILE = "Tests/images/hopper.spider"
@ -160,5 +160,4 @@ def test_odd_size():
im.save(data, format="SPIDER") im.save(data, format="SPIDER")
data.seek(0) data.seek(0)
with Image.open(data) as im2: assert_image_equal_tofile(im, data)
assert_image_equal(im, im2)

View File

@ -4,7 +4,7 @@ import pytest
from PIL import Image, SunImagePlugin from PIL import Image, SunImagePlugin
from .helper import assert_image_equal, assert_image_similar, hopper from .helper import assert_image_equal_tofile, assert_image_similar, hopper
EXTRA_DIR = "Tests/images/sunraster" EXTRA_DIR = "Tests/images/sunraster"
@ -29,8 +29,7 @@ def test_sanity():
def test_im1(): def test_im1():
with Image.open("Tests/images/sunraster.im1") as im: with Image.open("Tests/images/sunraster.im1") as im:
with Image.open("Tests/images/sunraster.im1.png") as target: assert_image_equal_tofile(im, "Tests/images/sunraster.im1.png")
assert_image_equal(im, target)
@pytest.mark.skipif( @pytest.mark.skipif(
@ -46,7 +45,4 @@ def test_others():
with Image.open(path) as im: with Image.open(path) as im:
im.load() im.load()
assert isinstance(im, SunImagePlugin.SunImageFile) assert isinstance(im, SunImagePlugin.SunImageFile)
target_path = f"{os.path.splitext(path)[0]}.png" assert_image_equal_tofile(im, f"{os.path.splitext(path)[0]}.png")
# im.save(target_file)
with Image.open(target_path) as target:
assert_image_equal(im, target)

View File

@ -483,8 +483,7 @@ class TestFileTiff:
tmpfile = str(tmp_path / "temp.tif") tmpfile = str(tmp_path / "temp.tif")
im.save(tmpfile) im.save(tmpfile)
with Image.open(tmpfile) as reloaded: assert_image_equal_tofile(im, tmpfile)
assert_image_equal(im, reloaded)
def test_strip_raw(self): def test_strip_raw(self):
infile = "Tests/images/tiff_strip_raw.tif" infile = "Tests/images/tiff_strip_raw.tif"

View File

@ -2,7 +2,12 @@ import pytest
from PIL import Image from PIL import Image
from .helper import assert_image_equal, assert_image_similar, hopper from .helper import (
assert_image_equal,
assert_image_similar,
assert_image_similar_tofile,
hopper,
)
_webp = pytest.importorskip("PIL._webp", reason="WebP support not installed") _webp = pytest.importorskip("PIL._webp", reason="WebP support not installed")
@ -29,8 +34,7 @@ def test_read_rgba():
image.tobytes() image.tobytes()
with Image.open("Tests/images/transparent.png") as target: assert_image_similar_tofile(image, "Tests/images/transparent.png", 20.0)
assert_image_similar(image, target, 20.0)
def test_write_lossless_rgb(tmp_path): def test_write_lossless_rgb(tmp_path):

View File

@ -2,7 +2,7 @@ import pytest
from PIL import Image, WmfImagePlugin from PIL import Image, WmfImagePlugin
from .helper import assert_image_similar, hopper from .helper import assert_image_similar_tofile, hopper
def test_load_raw(): def test_load_raw():
@ -13,9 +13,7 @@ def test_load_raw():
# Currently, support for WMF/EMF is Windows-only # Currently, support for WMF/EMF is Windows-only
im.load() im.load()
# Compare to reference rendering # Compare to reference rendering
with Image.open("Tests/images/drawing_emf_ref.png") as imref: assert_image_similar_tofile(im, "Tests/images/drawing_emf_ref.png", 0)
imref.load()
assert_image_similar(im, imref, 0)
# Test basic WMF open and rendering # Test basic WMF open and rendering
with Image.open("Tests/images/drawing.wmf") as im: with Image.open("Tests/images/drawing.wmf") as im:
@ -23,9 +21,7 @@ def test_load_raw():
# Currently, support for WMF/EMF is Windows-only # Currently, support for WMF/EMF is Windows-only
im.load() im.load()
# Compare to reference rendering # Compare to reference rendering
with Image.open("Tests/images/drawing_wmf_ref.png") as imref: assert_image_similar_tofile(im, "Tests/images/drawing_wmf_ref.png", 2.0)
imref.load()
assert_image_similar(im, imref, 2.0)
def test_register_handler(tmp_path): def test_register_handler(tmp_path):
@ -66,8 +62,7 @@ def test_load_set_dpi():
im.load(144) im.load(144)
assert im.size == (164, 164) assert im.size == (164, 164)
with Image.open("Tests/images/drawing_wmf_ref_144.png") as expected: assert_image_similar_tofile(im, "Tests/images/drawing_wmf_ref_144.png", 2.1)
assert_image_similar(im, expected, 2.1)
def test_save(tmp_path): def test_save(tmp_path):

View File

@ -4,7 +4,11 @@ import pytest
from PIL import FontFile, Image, ImageDraw, ImageFont, PcfFontFile from PIL import FontFile, Image, ImageDraw, ImageFont, PcfFontFile
from .helper import assert_image_equal, assert_image_similar, skip_unless_feature from .helper import (
assert_image_equal_tofile,
assert_image_similar_tofile,
skip_unless_feature,
)
fontname = "Tests/fonts/10x20-ISO8859-1.pcf" fontname = "Tests/fonts/10x20-ISO8859-1.pcf"
@ -33,8 +37,7 @@ def save_font(request, tmp_path):
font.save(tempname) font.save(tempname)
with Image.open(tempname.replace(".pil", ".pbm")) as loaded: with Image.open(tempname.replace(".pil", ".pbm")) as loaded:
with Image.open("Tests/fonts/10x20.pbm") as target: assert_image_equal_tofile(loaded, "Tests/fonts/10x20.pbm")
assert_image_equal(loaded, target)
with open(tempname, "rb") as f_loaded: with open(tempname, "rb") as f_loaded:
with open("Tests/fonts/10x20.pil", "rb") as f_target: with open("Tests/fonts/10x20.pil", "rb") as f_target:
@ -58,8 +61,7 @@ def test_draw(request, tmp_path):
im = Image.new("L", (130, 30), "white") im = Image.new("L", (130, 30), "white")
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
draw.text((0, 0), message, "black", font=font) draw.text((0, 0), message, "black", font=font)
with Image.open("Tests/images/test_draw_pbm_target.png") as target: assert_image_similar_tofile(im, "Tests/images/test_draw_pbm_target.png", 0)
assert_image_similar(im, target, 0)
def test_textsize(request, tmp_path): def test_textsize(request, tmp_path):
@ -80,8 +82,7 @@ def _test_high_characters(request, tmp_path, message):
im = Image.new("L", (750, 30), "white") im = Image.new("L", (750, 30), "white")
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
draw.text((0, 0), message, "black", font=font) draw.text((0, 0), message, "black", font=font)
with Image.open("Tests/images/high_ascii_chars.png") as target: assert_image_similar_tofile(im, "Tests/images/high_ascii_chars.png", 0)
assert_image_similar(im, target, 0)
def test_high_characters(request, tmp_path): def test_high_characters(request, tmp_path):

View File

@ -2,7 +2,11 @@ import os
from PIL import FontFile, Image, ImageDraw, ImageFont, PcfFontFile from PIL import FontFile, Image, ImageDraw, ImageFont, PcfFontFile
from .helper import assert_image_equal, assert_image_similar, skip_unless_feature from .helper import (
assert_image_equal_tofile,
assert_image_similar_tofile,
skip_unless_feature,
)
fontname = "Tests/fonts/ter-x20b.pcf" fontname = "Tests/fonts/ter-x20b.pcf"
@ -47,8 +51,7 @@ def save_font(request, tmp_path, encoding):
font.save(tempname) font.save(tempname)
with Image.open(tempname.replace(".pil", ".pbm")) as loaded: with Image.open(tempname.replace(".pil", ".pbm")) as loaded:
with Image.open(f"Tests/fonts/ter-x20b-{encoding}.pbm") as target: assert_image_equal_tofile(loaded, f"Tests/fonts/ter-x20b-{encoding}.pbm")
assert_image_equal(loaded, target)
with open(tempname, "rb") as f_loaded: with open(tempname, "rb") as f_loaded:
with open(f"Tests/fonts/ter-x20b-{encoding}.pil", "rb") as f_target: with open(f"Tests/fonts/ter-x20b-{encoding}.pil", "rb") as f_target:
@ -79,8 +82,7 @@ def _test_draw(request, tmp_path, encoding):
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
message = charsets[encoding]["message"].encode(encoding) message = charsets[encoding]["message"].encode(encoding)
draw.text((0, 0), message, "black", font=font) draw.text((0, 0), message, "black", font=font)
with Image.open(charsets[encoding]["image1"]) as target: assert_image_similar_tofile(im, charsets[encoding]["image1"], 0)
assert_image_similar(im, target, 0)
def test_draw_iso8859_1(request, tmp_path): def test_draw_iso8859_1(request, tmp_path):

View File

@ -10,7 +10,8 @@ from PIL import Image, ImageDraw, ImagePalette, ImageShow, UnidentifiedImageErro
from .helper import ( from .helper import (
assert_image_equal, assert_image_equal,
assert_image_similar, assert_image_equal_tofile,
assert_image_similar_tofile,
assert_not_all_same, assert_not_all_same,
hopper, hopper,
is_win32, is_win32,
@ -171,8 +172,7 @@ class TestImage:
with tempfile.TemporaryFile() as fp: with tempfile.TemporaryFile() as fp:
im.save(fp, "JPEG") im.save(fp, "JPEG")
fp.seek(0) fp.seek(0)
with Image.open(fp) as reloaded: assert_image_similar_tofile(im, fp, 20)
assert_image_similar(im, reloaded, 20)
def test_unknown_extension(self, tmp_path): def test_unknown_extension(self, tmp_path):
im = hopper() im = hopper()
@ -413,8 +413,7 @@ class TestImage:
# Assert # Assert
assert im.size == (512, 512) assert im.size == (512, 512)
with Image.open("Tests/images/effect_mandelbrot.png") as im2: assert_image_equal_tofile(im, "Tests/images/effect_mandelbrot.png")
assert_image_equal(im, im2)
def test_effect_mandelbrot_bad_arguments(self): def test_effect_mandelbrot_bad_arguments(self):
# Arrange # Arrange
@ -456,8 +455,7 @@ class TestImage:
# Assert # Assert
assert im.size == (128, 128) assert im.size == (128, 128)
with Image.open("Tests/images/effect_spread.png") as im3: assert_image_similar_tofile(im2, "Tests/images/effect_spread.png", 110)
assert_image_similar(im2, im3, 110)
def test_effect_spread_zero(self): def test_effect_spread_zero(self):
# Arrange # Arrange

View File

@ -1,6 +1,11 @@
from PIL import Image from PIL import Image
from .helper import assert_image_equal, assert_image_similar, hopper from .helper import (
assert_image_equal,
assert_image_equal_tofile,
assert_image_similar,
hopper,
)
def rotate(im, mode, angle, center=None, translate=None): def rotate(im, mode, angle, center=None, translate=None):
@ -113,15 +118,13 @@ def test_center():
def test_rotate_no_fill(): def test_rotate_no_fill():
im = Image.new("RGB", (100, 100), "green") im = Image.new("RGB", (100, 100), "green")
im = im.rotate(45) im = im.rotate(45)
with Image.open("Tests/images/rotate_45_no_fill.png") as target: assert_image_equal_tofile(im, "Tests/images/rotate_45_no_fill.png")
assert_image_equal(im, target)
def test_rotate_with_fill(): def test_rotate_with_fill():
im = Image.new("RGB", (100, 100), "green") im = Image.new("RGB", (100, 100), "green")
im = im.rotate(45, fillcolor="white") im = im.rotate(45, fillcolor="white")
with Image.open("Tests/images/rotate_45_with_fill.png") as target: assert_image_equal_tofile(im, "Tests/images/rotate_45_with_fill.png")
assert_image_equal(im, target)
def test_alpha_rotate_no_fill(): def test_alpha_rotate_no_fill():

View File

@ -8,7 +8,13 @@ import pytest
from PIL import Image, ImageMode, features from PIL import Image, ImageMode, features
from .helper import assert_image, assert_image_equal, assert_image_similar, hopper from .helper import (
assert_image,
assert_image_equal,
assert_image_similar,
assert_image_similar_tofile,
hopper,
)
try: try:
from PIL import ImageCms from PIL import ImageCms
@ -240,8 +246,7 @@ def test_lab_color():
# i.save('temp.lab.tif') # visually verified vs PS. # i.save('temp.lab.tif') # visually verified vs PS.
with Image.open("Tests/images/hopper.Lab.tif") as target: assert_image_similar_tofile(i, "Tests/images/hopper.Lab.tif", 3.5)
assert_image_similar(i, target, 3.5)
def test_lab_srgb(): def test_lab_srgb():

View File

@ -6,6 +6,7 @@ from PIL import Image, ImageColor, ImageDraw, ImageFont
from .helper import ( from .helper import (
assert_image_equal, assert_image_equal,
assert_image_equal_tofile,
assert_image_similar_tofile, assert_image_similar_tofile,
hopper, hopper,
skip_unless_feature, skip_unless_feature,
@ -96,7 +97,7 @@ def test_arc_end_le_start():
draw.arc(BBOX1, start=start, end=end) draw.arc(BBOX1, start=start, end=end)
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_arc_end_le_start.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_arc_end_le_start.png")
def test_arc_no_loops(): def test_arc_no_loops():
@ -174,7 +175,7 @@ def test_arc_high():
draw.arc([110, 10, 189, 189], 20, 150, width=20, fill="white") draw.arc([110, 10, 189, 189], 20, 150, width=20, fill="white")
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_arc_high.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_arc_high.png")
def test_bitmap(): def test_bitmap():
@ -188,7 +189,7 @@ def test_bitmap():
draw.bitmap((10, 10), small) draw.bitmap((10, 10), small)
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_bitmap.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_bitmap.png")
def helper_chord(mode, bbox, start, end): def helper_chord(mode, bbox, start, end):
@ -247,8 +248,7 @@ def test_chord_zero_width():
draw.chord(BBOX1, 10, 260, fill="red", outline="yellow", width=0) draw.chord(BBOX1, 10, 260, fill="red", outline="yellow", width=0)
# Assert # Assert
with Image.open("Tests/images/imagedraw_chord_zero_width.png") as expected: assert_image_equal_tofile(im, "Tests/images/imagedraw_chord_zero_width.png")
assert_image_equal(im, expected)
def test_chord_too_fat(): def test_chord_too_fat():
@ -260,7 +260,7 @@ def test_chord_too_fat():
draw.chord([-150, -150, 99, 99], 15, 60, width=10, fill="white", outline="red") draw.chord([-150, -150, 99, 99], 15, 60, width=10, fill="white", outline="red")
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_chord_too_fat.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_chord_too_fat.png")
def helper_ellipse(mode, bbox): def helper_ellipse(mode, bbox):
@ -367,8 +367,7 @@ def test_ellipse_zero_width():
draw.ellipse(BBOX1, fill="green", outline="blue", width=0) draw.ellipse(BBOX1, fill="green", outline="blue", width=0)
# Assert # Assert
with Image.open("Tests/images/imagedraw_ellipse_zero_width.png") as expected: assert_image_equal_tofile(im, "Tests/images/imagedraw_ellipse_zero_width.png")
assert_image_equal(im, expected)
def ellipse_various_sizes_helper(filled): def ellipse_various_sizes_helper(filled):
@ -395,17 +394,15 @@ def ellipse_various_sizes_helper(filled):
def test_ellipse_various_sizes(): def test_ellipse_various_sizes():
im = ellipse_various_sizes_helper(False) im = ellipse_various_sizes_helper(False)
with Image.open("Tests/images/imagedraw_ellipse_various_sizes.png") as expected: assert_image_equal_tofile(im, "Tests/images/imagedraw_ellipse_various_sizes.png")
assert_image_equal(im, expected)
def test_ellipse_various_sizes_filled(): def test_ellipse_various_sizes_filled():
im = ellipse_various_sizes_helper(True) im = ellipse_various_sizes_helper(True)
with Image.open( assert_image_equal_tofile(
"Tests/images/imagedraw_ellipse_various_sizes_filled.png" im, "Tests/images/imagedraw_ellipse_various_sizes_filled.png"
) as expected: )
assert_image_equal(im, expected)
def helper_line(points): def helper_line(points):
@ -417,7 +414,7 @@ def helper_line(points):
draw.line(points, fill="yellow", width=2) draw.line(points, fill="yellow", width=2)
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_line.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_line.png")
def test_line1(): def test_line1():
@ -446,7 +443,7 @@ def test_shape1():
draw.shape(s, fill=1) draw.shape(s, fill=1)
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_shape1.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_shape1.png")
def test_shape2(): def test_shape2():
@ -467,7 +464,7 @@ def test_shape2():
draw.shape(s, outline="blue") draw.shape(s, outline="blue")
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_shape2.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_shape2.png")
def helper_pieslice(bbox, start, end): def helper_pieslice(bbox, start, end):
@ -526,8 +523,7 @@ def test_pieslice_zero_width():
draw.pieslice(BBOX1, 10, 260, fill="white", outline="blue", width=0) draw.pieslice(BBOX1, 10, 260, fill="white", outline="blue", width=0)
# Assert # Assert
with Image.open("Tests/images/imagedraw_pieslice_zero_width.png") as expected: assert_image_equal_tofile(im, "Tests/images/imagedraw_pieslice_zero_width.png")
assert_image_equal(im, expected)
def test_pieslice_wide(): def test_pieslice_wide():
@ -539,7 +535,7 @@ def test_pieslice_wide():
draw.pieslice([0, 0, 199, 99], 190, 170, width=10, fill="white", outline="red") draw.pieslice([0, 0, 199, 99], 190, 170, width=10, fill="white", outline="red")
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_pieslice_wide.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_pieslice_wide.png")
def helper_point(points): def helper_point(points):
@ -551,7 +547,7 @@ def helper_point(points):
draw.point(points, fill="yellow") draw.point(points, fill="yellow")
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_point.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_point.png")
def test_point1(): def test_point1():
@ -571,7 +567,7 @@ def helper_polygon(points):
draw.polygon(points, fill="red", outline="blue") draw.polygon(points, fill="red", outline="blue")
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_polygon.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_polygon.png")
def test_polygon1(): def test_polygon1():
@ -595,7 +591,7 @@ def test_polygon_kite():
draw.polygon(KITE_POINTS, fill="blue", outline="yellow") draw.polygon(KITE_POINTS, fill="blue", outline="yellow")
# Assert # Assert
assert_image_equal(im, Image.open(expected)) assert_image_equal_tofile(im, expected)
def test_polygon_1px_high(): def test_polygon_1px_high():
@ -609,7 +605,7 @@ def test_polygon_1px_high():
draw.polygon([(0, 1), (0, 1), (2, 1), (2, 1)], "#f00") draw.polygon([(0, 1), (0, 1), (2, 1), (2, 1)], "#f00")
# Assert # Assert
assert_image_equal(im, Image.open(expected)) assert_image_equal_tofile(im, expected)
def helper_rectangle(bbox): def helper_rectangle(bbox):
@ -621,7 +617,7 @@ def helper_rectangle(bbox):
draw.rectangle(bbox, fill="black", outline="green") draw.rectangle(bbox, fill="black", outline="green")
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_rectangle.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_rectangle.png")
def test_rectangle1(): def test_rectangle1():
@ -656,7 +652,7 @@ def test_rectangle_width():
draw.rectangle(BBOX1, outline="green", width=5) draw.rectangle(BBOX1, outline="green", width=5)
# Assert # Assert
assert_image_equal(im, Image.open(expected)) assert_image_equal_tofile(im, expected)
def test_rectangle_width_fill(): def test_rectangle_width_fill():
@ -669,7 +665,7 @@ def test_rectangle_width_fill():
draw.rectangle(BBOX1, fill="blue", outline="green", width=5) draw.rectangle(BBOX1, fill="blue", outline="green", width=5)
# Assert # Assert
assert_image_equal(im, Image.open(expected)) assert_image_equal_tofile(im, expected)
def test_rectangle_zero_width(): def test_rectangle_zero_width():
@ -681,8 +677,7 @@ def test_rectangle_zero_width():
draw.rectangle(BBOX1, fill="blue", outline="green", width=0) draw.rectangle(BBOX1, fill="blue", outline="green", width=0)
# Assert # Assert
with Image.open("Tests/images/imagedraw_rectangle_zero_width.png") as expected: assert_image_equal_tofile(im, "Tests/images/imagedraw_rectangle_zero_width.png")
assert_image_equal(im, expected)
def test_rectangle_I16(): def test_rectangle_I16():
@ -694,9 +689,7 @@ def test_rectangle_I16():
draw.rectangle(BBOX1, fill="black", outline="green") draw.rectangle(BBOX1, fill="black", outline="green")
# Assert # Assert
assert_image_equal( assert_image_equal_tofile(im.convert("I"), "Tests/images/imagedraw_rectangle_I.png")
im.convert("I"), Image.open("Tests/images/imagedraw_rectangle_I.png")
)
def test_floodfill(): def test_floodfill():
@ -749,7 +742,7 @@ def test_floodfill_border():
) )
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_floodfill2.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_floodfill2.png")
def test_floodfill_thresh(): def test_floodfill_thresh():
@ -765,7 +758,7 @@ def test_floodfill_thresh():
ImageDraw.floodfill(im, centre_point, ImageColor.getrgb("red"), thresh=30) ImageDraw.floodfill(im, centre_point, ImageColor.getrgb("red"), thresh=30)
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_floodfill2.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_floodfill2.png")
def test_floodfill_not_negative(): def test_floodfill_not_negative():
@ -782,9 +775,7 @@ def test_floodfill_not_negative():
ImageDraw.floodfill(im, (int(W / 4), int(H / 4)), ImageColor.getrgb("red")) ImageDraw.floodfill(im, (int(W / 4), int(H / 4)), ImageColor.getrgb("red"))
# Assert # Assert
assert_image_equal( assert_image_equal_tofile(im, "Tests/images/imagedraw_floodfill_not_negative.png")
im, Image.open("Tests/images/imagedraw_floodfill_not_negative.png")
)
def create_base_image_draw( def create_base_image_draw(
@ -816,32 +807,29 @@ def test_square():
def test_triangle_right(): def test_triangle_right():
with Image.open(os.path.join(IMAGES_PATH, "triangle_right.png")) as expected: img, draw = create_base_image_draw((20, 20))
expected.load() draw.polygon([(3, 5), (17, 5), (10, 12)], BLACK)
img, draw = create_base_image_draw((20, 20)) assert_image_equal_tofile(
draw.polygon([(3, 5), (17, 5), (10, 12)], BLACK) img, os.path.join(IMAGES_PATH, "triangle_right.png"), "triangle right failed"
assert_image_equal(img, expected, "triangle right failed") )
def test_line_horizontal(): def test_line_horizontal():
with Image.open( img, draw = create_base_image_draw((20, 20))
os.path.join(IMAGES_PATH, "line_horizontal_w2px_normal.png") draw.line((5, 5, 14, 5), BLACK, 2)
) as expected: assert_image_equal_tofile(
expected.load() img,
img, draw = create_base_image_draw((20, 20)) os.path.join(IMAGES_PATH, "line_horizontal_w2px_normal.png"),
draw.line((5, 5, 14, 5), BLACK, 2) "line straight horizontal normal 2px wide failed",
assert_image_equal( )
img, expected, "line straight horizontal normal 2px wide failed"
) img, draw = create_base_image_draw((20, 20))
with Image.open( draw.line((14, 5, 5, 5), BLACK, 2)
os.path.join(IMAGES_PATH, "line_horizontal_w2px_inverted.png") assert_image_equal_tofile(
) as expected: img,
expected.load() os.path.join(IMAGES_PATH, "line_horizontal_w2px_inverted.png"),
img, draw = create_base_image_draw((20, 20)) "line straight horizontal inverted 2px wide failed",
draw.line((14, 5, 5, 5), BLACK, 2) )
assert_image_equal(
img, expected, "line straight horizontal inverted 2px wide failed"
)
with Image.open(os.path.join(IMAGES_PATH, "line_horizontal_w3px.png")) as expected: with Image.open(os.path.join(IMAGES_PATH, "line_horizontal_w3px.png")) as expected:
expected.load() expected.load()
img, draw = create_base_image_draw((20, 20)) img, draw = create_base_image_draw((20, 20))
@ -854,45 +842,43 @@ def test_line_horizontal():
assert_image_equal( assert_image_equal(
img, expected, "line straight horizontal inverted 3px wide failed" img, expected, "line straight horizontal inverted 3px wide failed"
) )
with Image.open(
os.path.join(IMAGES_PATH, "line_horizontal_w101px.png") img, draw = create_base_image_draw((200, 110))
) as expected: draw.line((5, 55, 195, 55), BLACK, 101)
expected.load() assert_image_equal_tofile(
img, draw = create_base_image_draw((200, 110)) img,
draw.line((5, 55, 195, 55), BLACK, 101) os.path.join(IMAGES_PATH, "line_horizontal_w101px.png"),
assert_image_equal(img, expected, "line straight horizontal 101px wide failed") "line straight horizontal 101px wide failed",
)
def test_line_h_s1_w2(): def test_line_h_s1_w2():
pytest.skip("failing") pytest.skip("failing")
with Image.open( img, draw = create_base_image_draw((20, 20))
os.path.join(IMAGES_PATH, "line_horizontal_slope1px_w2px.png") draw.line((5, 5, 14, 6), BLACK, 2)
) as expected: assert_image_equal_tofile(
expected.load() img,
img, draw = create_base_image_draw((20, 20)) os.path.join(IMAGES_PATH, "line_horizontal_slope1px_w2px.png"),
draw.line((5, 5, 14, 6), BLACK, 2) "line horizontal 1px slope 2px wide failed",
assert_image_equal(img, expected, "line horizontal 1px slope 2px wide failed") )
def test_line_vertical(): def test_line_vertical():
with Image.open( img, draw = create_base_image_draw((20, 20))
os.path.join(IMAGES_PATH, "line_vertical_w2px_normal.png") draw.line((5, 5, 5, 14), BLACK, 2)
) as expected: assert_image_equal_tofile(
expected.load() img,
img, draw = create_base_image_draw((20, 20)) os.path.join(IMAGES_PATH, "line_vertical_w2px_normal.png"),
draw.line((5, 5, 5, 14), BLACK, 2) "line straight vertical normal 2px wide failed",
assert_image_equal( )
img, expected, "line straight vertical normal 2px wide failed"
) img, draw = create_base_image_draw((20, 20))
with Image.open( draw.line((5, 14, 5, 5), BLACK, 2)
os.path.join(IMAGES_PATH, "line_vertical_w2px_inverted.png") assert_image_equal_tofile(
) as expected: img,
expected.load() os.path.join(IMAGES_PATH, "line_vertical_w2px_inverted.png"),
img, draw = create_base_image_draw((20, 20)) "line straight vertical inverted 2px wide failed",
draw.line((5, 14, 5, 5), BLACK, 2) )
assert_image_equal(
img, expected, "line straight vertical inverted 2px wide failed"
)
with Image.open(os.path.join(IMAGES_PATH, "line_vertical_w3px.png")) as expected: with Image.open(os.path.join(IMAGES_PATH, "line_vertical_w3px.png")) as expected:
expected.load() expected.load()
img, draw = create_base_image_draw((20, 20)) img, draw = create_base_image_draw((20, 20))
@ -905,18 +891,21 @@ def test_line_vertical():
assert_image_equal( assert_image_equal(
img, expected, "line straight vertical inverted 3px wide failed" img, expected, "line straight vertical inverted 3px wide failed"
) )
with Image.open(os.path.join(IMAGES_PATH, "line_vertical_w101px.png")) as expected: img, draw = create_base_image_draw((110, 200))
expected.load() draw.line((55, 5, 55, 195), BLACK, 101)
img, draw = create_base_image_draw((110, 200)) assert_image_equal_tofile(
draw.line((55, 5, 55, 195), BLACK, 101) img,
assert_image_equal(img, expected, "line straight vertical 101px wide failed") os.path.join(IMAGES_PATH, "line_vertical_w101px.png"),
with Image.open( "line straight vertical 101px wide failed",
os.path.join(IMAGES_PATH, "line_vertical_slope1px_w2px.png") )
) as expected:
expected.load() img, draw = create_base_image_draw((20, 20))
img, draw = create_base_image_draw((20, 20)) draw.line((5, 5, 6, 14), BLACK, 2)
draw.line((5, 5, 6, 14), BLACK, 2) assert_image_equal_tofile(
assert_image_equal(img, expected, "line vertical 1px slope 2px wide failed") img,
os.path.join(IMAGES_PATH, "line_vertical_slope1px_w2px.png"),
"line vertical 1px slope 2px wide failed",
)
def test_line_oblique_45(): def test_line_oblique_45():
@ -1185,7 +1174,7 @@ def test_draw_regular_polygon(n_sides, rotation, polygon_name):
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
bounding_circle = ((W // 2, H // 2), 25) bounding_circle = ((W // 2, H // 2), 25)
draw.regular_polygon(bounding_circle, n_sides, rotation=rotation, fill="red") draw.regular_polygon(bounding_circle, n_sides, rotation=rotation, fill="red")
assert_image_equal(im, Image.open(filename)) assert_image_equal_tofile(im, filename)
@pytest.mark.parametrize( @pytest.mark.parametrize(

View File

@ -4,7 +4,8 @@ from PIL import Image, ImageDraw, ImageDraw2
from .helper import ( from .helper import (
assert_image_equal, assert_image_equal,
assert_image_similar, assert_image_equal_tofile,
assert_image_similar_tofile,
hopper, hopper,
skip_unless_feature, skip_unless_feature,
) )
@ -61,7 +62,7 @@ def helper_ellipse(mode, bbox):
draw.ellipse(bbox, pen, brush) draw.ellipse(bbox, pen, brush)
# Assert # Assert
assert_image_similar(im, Image.open(expected), 1) assert_image_similar_tofile(im, expected, 1)
def test_ellipse1(): def test_ellipse1():
@ -82,7 +83,7 @@ def test_ellipse_edge():
draw.ellipse(((0, 0), (W - 1, H - 1)), brush) draw.ellipse(((0, 0), (W - 1, H - 1)), brush)
# Assert # Assert
assert_image_similar(im, Image.open("Tests/images/imagedraw_ellipse_edge.png"), 1) assert_image_similar_tofile(im, "Tests/images/imagedraw_ellipse_edge.png", 1)
def helper_line(points): def helper_line(points):
@ -95,7 +96,7 @@ def helper_line(points):
draw.line(points, pen) draw.line(points, pen)
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_line.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_line.png")
def test_line1_pen(): def test_line1_pen():
@ -118,7 +119,7 @@ def test_line_pen_as_brush():
draw.line(POINTS1, pen, brush) draw.line(POINTS1, pen, brush)
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_line.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_line.png")
def helper_polygon(points): def helper_polygon(points):
@ -132,7 +133,7 @@ def helper_polygon(points):
draw.polygon(points, pen, brush) draw.polygon(points, pen, brush)
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_polygon.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_polygon.png")
def test_polygon1(): def test_polygon1():
@ -154,7 +155,7 @@ def helper_rectangle(bbox):
draw.rectangle(bbox, pen, brush) draw.rectangle(bbox, pen, brush)
# Assert # Assert
assert_image_equal(im, Image.open("Tests/images/imagedraw_rectangle.png")) assert_image_equal_tofile(im, "Tests/images/imagedraw_rectangle.png")
def test_rectangle1(): def test_rectangle1():
@ -178,7 +179,7 @@ def test_big_rectangle():
draw.rectangle(bbox, brush) draw.rectangle(bbox, brush)
# Assert # Assert
assert_image_similar(im, Image.open(expected), 1) assert_image_similar_tofile(im, expected, 1)
@skip_unless_feature("freetype2") @skip_unless_feature("freetype2")
@ -193,7 +194,7 @@ def test_text():
draw.text((5, 5), "ImageDraw2", font) draw.text((5, 5), "ImageDraw2", font)
# Assert # Assert
assert_image_similar(im, Image.open(expected), 13) assert_image_similar_tofile(im, expected, 13)
@skip_unless_feature("freetype2") @skip_unless_feature("freetype2")

View File

@ -13,7 +13,6 @@ from PIL import Image, ImageDraw, ImageFont, features
from .helper import ( from .helper import (
assert_image_equal, assert_image_equal,
assert_image_equal_tofile, assert_image_equal_tofile,
assert_image_similar,
assert_image_similar_tofile, assert_image_similar_tofile,
is_win32, is_win32,
skip_unless_feature, skip_unless_feature,
@ -130,8 +129,7 @@ class TestImageFont:
draw.text((10, 10), txt, font=ttf) draw.text((10, 10), txt, font=ttf)
target = "Tests/images/transparent_background_text.png" target = "Tests/images/transparent_background_text.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 4.09)
assert_image_similar(im, target_img, 4.09)
def test_textsize_equal(self): def test_textsize_equal(self):
im = Image.new(mode="RGB", size=(300, 100)) im = Image.new(mode="RGB", size=(300, 100))
@ -143,11 +141,10 @@ class TestImageFont:
draw.text((10, 10), txt, font=ttf) draw.text((10, 10), txt, font=ttf)
draw.rectangle((10, 10, 10 + size[0], 10 + size[1])) draw.rectangle((10, 10, 10 + size[0], 10 + size[1]))
target = "Tests/images/rectangle_surrounding_text.png" # Epsilon ~.5 fails with FreeType 2.7
with Image.open(target) as target_img: assert_image_similar_tofile(
im, "Tests/images/rectangle_surrounding_text.png", 2.5
# Epsilon ~.5 fails with FreeType 2.7 )
assert_image_similar(im, target_img, 2.5)
@pytest.mark.parametrize( @pytest.mark.parametrize(
"text, mode, font, size, length_basic, length_raqm", "text, mode, font, size, length_basic, length_raqm",
@ -191,13 +188,10 @@ class TestImageFont:
draw.text((0, y), line, font=ttf) draw.text((0, y), line, font=ttf)
y += line_spacing y += line_spacing
target = "Tests/images/multiline_text.png" # some versions of freetype have different horizontal spacing.
with Image.open(target) as target_img: # setting a tight epsilon, I'm showing the original test failure
# at epsilon = ~38.
# some versions of freetype have different horizontal spacing. assert_image_similar_tofile(im, "Tests/images/multiline_text.png", 6.2)
# setting a tight epsilon, I'm showing the original test failure
# at epsilon = ~38.
assert_image_similar(im, target_img, 6.2)
def test_render_multiline_text(self): def test_render_multiline_text(self):
ttf = self.get_font() ttf = self.get_font()
@ -208,11 +202,8 @@ class TestImageFont:
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
draw.text((0, 0), TEST_TEXT, font=ttf) draw.text((0, 0), TEST_TEXT, font=ttf)
target = "Tests/images/multiline_text.png" # Epsilon ~.5 fails with FreeType 2.7
with Image.open(target) as target_img: assert_image_similar_tofile(im, "Tests/images/multiline_text.png", 6.2)
# Epsilon ~.5 fails with FreeType 2.7
assert_image_similar(im, target_img, 6.2)
# Test that text() can pass on additional arguments # Test that text() can pass on additional arguments
# to multiline_text() # to multiline_text()
@ -227,11 +218,10 @@ class TestImageFont:
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
draw.multiline_text((0, 0), TEST_TEXT, font=ttf, align=align) draw.multiline_text((0, 0), TEST_TEXT, font=ttf, align=align)
target = "Tests/images/multiline_text" + ext + ".png" # Epsilon ~.5 fails with FreeType 2.7
with Image.open(target) as target_img: assert_image_similar_tofile(
im, "Tests/images/multiline_text" + ext + ".png", 6.2
# Epsilon ~.5 fails with FreeType 2.7 )
assert_image_similar(im, target_img, 6.2)
def test_unknown_align(self): def test_unknown_align(self):
im = Image.new(mode="RGB", size=(300, 100)) im = Image.new(mode="RGB", size=(300, 100))
@ -285,11 +275,8 @@ class TestImageFont:
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
draw.multiline_text((0, 0), TEST_TEXT, font=ttf, spacing=10) draw.multiline_text((0, 0), TEST_TEXT, font=ttf, spacing=10)
target = "Tests/images/multiline_text_spacing.png" # Epsilon ~.5 fails with FreeType 2.7
with Image.open(target) as target_img: assert_image_similar_tofile(im, "Tests/images/multiline_text_spacing.png", 6.2)
# Epsilon ~.5 fails with FreeType 2.7
assert_image_similar(im, target_img, 6.2)
def test_rotated_transposed_font(self): def test_rotated_transposed_font(self):
img_grey = Image.new("L", (100, 100)) img_grey = Image.new("L", (100, 100))
@ -423,15 +410,12 @@ class TestImageFont:
im = Image.new(mode="RGB", size=(300, 100)) im = Image.new(mode="RGB", size=(300, 100))
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
target = "Tests/images/default_font.png" # Act
with Image.open(target) as target_img: default_font = ImageFont.load_default()
draw.text((10, 10), txt, font=default_font)
# Act # Assert
default_font = ImageFont.load_default() assert_image_equal_tofile(im, "Tests/images/default_font.png")
draw.text((10, 10), txt, font=default_font)
# Assert
assert_image_equal(im, target_img)
def test_getsize_empty(self): def test_getsize_empty(self):
# issue #2614 # issue #2614
@ -680,13 +664,11 @@ class TestImageFont:
d.text((10, 10), "Text", font=font, fill="black") d.text((10, 10), "Text", font=font, fill="black")
try: try:
with Image.open(path) as expected: assert_image_similar_tofile(im, path, epsilon)
assert_image_similar(im, expected, epsilon)
except AssertionError: except AssertionError:
if "_adobe" in path: if "_adobe" in path:
path = path.replace("_adobe", "_adobe_older_harfbuzz") path = path.replace("_adobe", "_adobe_older_harfbuzz")
with Image.open(path) as expected: assert_image_similar_tofile(im, path, epsilon)
assert_image_similar(im, expected, epsilon)
else: else:
raise raise
@ -777,8 +759,7 @@ class TestImageFont:
assert d.textbbox((0, 0), text, f, anchor=anchor) == bbox_expected assert d.textbbox((0, 0), text, f, anchor=anchor) == bbox_expected
with Image.open(path) as expected: assert_image_similar_tofile(im, path, 7)
assert_image_similar(im, expected, 7)
@pytest.mark.parametrize( @pytest.mark.parametrize(
"anchor, align", "anchor, align",
@ -816,8 +797,7 @@ class TestImageFont:
(300, 200), text, fill="black", anchor=anchor, font=f, align=align (300, 200), text, fill="black", anchor=anchor, font=f, align=align
) )
with Image.open(target) as expected: assert_image_similar_tofile(im, target, 4)
assert_image_similar(im, expected, 4)
def test_anchor_invalid(self): def test_anchor_invalid(self):
font = self.get_font() font = self.get_font()
@ -875,8 +855,7 @@ class TestImageFont:
d = ImageDraw.Draw(im) d = ImageDraw.Draw(im)
d.text((10, 10), txt, font=ttf, fill="#fa6", embedded_color=True) d.text((10, 10), txt, font=ttf, fill="#fa6", embedded_color=True)
with Image.open("Tests/images/standard_embedded.png") as expected: assert_image_similar_tofile(im, "Tests/images/standard_embedded.png", 6.2)
assert_image_similar(im, expected, 6.2)
@skip_unless_feature_version("freetype2", "2.5.0") @skip_unless_feature_version("freetype2", "2.5.0")
def test_cbdt(self): def test_cbdt(self):
@ -892,8 +871,7 @@ class TestImageFont:
d.text((10, 10), "\U0001f469", embedded_color=True, font=font) d.text((10, 10), "\U0001f469", embedded_color=True, font=font)
with Image.open("Tests/images/cbdt_notocoloremoji.png") as expected: assert_image_similar_tofile(im, "Tests/images/cbdt_notocoloremoji.png", 6.2)
assert_image_similar(im, expected, 6.2)
except IOError as e: except IOError as e:
assert str(e) in ("unimplemented feature", "unknown file format") assert str(e) in ("unimplemented feature", "unknown file format")
pytest.skip("freetype compiled without libpng or unsupported") pytest.skip("freetype compiled without libpng or unsupported")
@ -912,8 +890,9 @@ class TestImageFont:
d.text((10, 10), "\U0001f469", "black", font=font) d.text((10, 10), "\U0001f469", "black", font=font)
with Image.open("Tests/images/cbdt_notocoloremoji_mask.png") as expected: assert_image_similar_tofile(
assert_image_similar(im, expected, 6.2) im, "Tests/images/cbdt_notocoloremoji_mask.png", 6.2
)
except IOError as e: except IOError as e:
assert str(e) in ("unimplemented feature", "unknown file format") assert str(e) in ("unimplemented feature", "unknown file format")
pytest.skip("freetype compiled without libpng or unsupported") pytest.skip("freetype compiled without libpng or unsupported")
@ -931,8 +910,7 @@ class TestImageFont:
d.text((15, 5), "Bungee", embedded_color=True, font=font) d.text((15, 5), "Bungee", embedded_color=True, font=font)
with Image.open("Tests/images/colr_bungee.png") as expected: assert_image_similar_tofile(im, "Tests/images/colr_bungee.png", 21)
assert_image_similar(im, expected, 21)
@skip_unless_feature_version("freetype2", "2.10.0") @skip_unless_feature_version("freetype2", "2.10.0")
def test_colr_mask(self): def test_colr_mask(self):
@ -947,8 +925,7 @@ class TestImageFont:
d.text((15, 5), "Bungee", "black", font=font) d.text((15, 5), "Bungee", "black", font=font)
with Image.open("Tests/images/colr_bungee_mask.png") as expected: assert_image_similar_tofile(im, "Tests/images/colr_bungee_mask.png", 22)
assert_image_similar(im, expected, 22)
@skip_unless_feature("raqm") @skip_unless_feature("raqm")

View File

@ -4,7 +4,7 @@ from packaging.version import parse as parse_version
from PIL import Image, ImageDraw, ImageFont, features from PIL import Image, ImageDraw, ImageFont, features
from .helper import ( from .helper import (
assert_image_similar, assert_image_similar_tofile,
skip_unless_feature, skip_unless_feature,
skip_unless_feature_version, skip_unless_feature_version,
) )
@ -31,8 +31,7 @@ def test_complex_text():
draw.text((0, 0), "اهلا عمان", font=ttf, fill=500) draw.text((0, 0), "اهلا عمان", font=ttf, fill=500)
target = "Tests/images/test_text.png" target = "Tests/images/test_text.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
def test_y_offset(): def test_y_offset():
@ -43,8 +42,7 @@ def test_y_offset():
draw.text((0, 0), "العالم العربي", font=ttf, fill=500) draw.text((0, 0), "العالم العربي", font=ttf, fill=500)
target = "Tests/images/test_y_offset.png" target = "Tests/images/test_y_offset.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 1.7)
assert_image_similar(im, target_img, 1.7)
def test_complex_unicode_text(): def test_complex_unicode_text():
@ -55,8 +53,7 @@ def test_complex_unicode_text():
draw.text((0, 0), "السلام عليكم", font=ttf, fill=500) draw.text((0, 0), "السلام عليكم", font=ttf, fill=500)
target = "Tests/images/test_complex_unicode_text.png" target = "Tests/images/test_complex_unicode_text.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
ttf = ImageFont.truetype("Tests/fonts/KhmerOSBattambang-Regular.ttf", FONT_SIZE) ttf = ImageFont.truetype("Tests/fonts/KhmerOSBattambang-Regular.ttf", FONT_SIZE)
@ -65,8 +62,7 @@ def test_complex_unicode_text():
draw.text((0, 0), "លោកុប្បត្តិ", font=ttf, fill=500) draw.text((0, 0), "លោកុប្បត្តិ", font=ttf, fill=500)
target = "Tests/images/test_complex_unicode_text2.png" target = "Tests/images/test_complex_unicode_text2.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 2.33)
assert_image_similar(im, target_img, 2.33)
def test_text_direction_rtl(): def test_text_direction_rtl():
@ -77,8 +73,7 @@ def test_text_direction_rtl():
draw.text((0, 0), "English عربي", font=ttf, fill=500, direction="rtl") draw.text((0, 0), "English عربي", font=ttf, fill=500, direction="rtl")
target = "Tests/images/test_direction_rtl.png" target = "Tests/images/test_direction_rtl.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
def test_text_direction_ltr(): def test_text_direction_ltr():
@ -89,8 +84,7 @@ def test_text_direction_ltr():
draw.text((0, 0), "سلطنة عمان Oman", font=ttf, fill=500, direction="ltr") draw.text((0, 0), "سلطنة عمان Oman", font=ttf, fill=500, direction="ltr")
target = "Tests/images/test_direction_ltr.png" target = "Tests/images/test_direction_ltr.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
def test_text_direction_rtl2(): def test_text_direction_rtl2():
@ -101,8 +95,7 @@ def test_text_direction_rtl2():
draw.text((0, 0), "Oman سلطنة عمان", font=ttf, fill=500, direction="rtl") draw.text((0, 0), "Oman سلطنة عمان", font=ttf, fill=500, direction="rtl")
target = "Tests/images/test_direction_ltr.png" target = "Tests/images/test_direction_ltr.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
def test_text_direction_ttb(): def test_text_direction_ttb():
@ -117,8 +110,7 @@ def test_text_direction_ttb():
pytest.skip("libraqm 0.7 or greater not available") pytest.skip("libraqm 0.7 or greater not available")
target = "Tests/images/test_direction_ttb.png" target = "Tests/images/test_direction_ttb.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 2.8)
assert_image_similar(im, target_img, 2.8)
def test_text_direction_ttb_stroke(): def test_text_direction_ttb_stroke():
@ -141,8 +133,7 @@ def test_text_direction_ttb_stroke():
pytest.skip("libraqm 0.7 or greater not available") pytest.skip("libraqm 0.7 or greater not available")
target = "Tests/images/test_direction_ttb_stroke.png" target = "Tests/images/test_direction_ttb_stroke.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 19.4)
assert_image_similar(im, target_img, 19.4)
def test_ligature_features(): def test_ligature_features():
@ -152,8 +143,7 @@ def test_ligature_features():
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)
draw.text((0, 0), "filling", font=ttf, fill=500, features=["-liga"]) draw.text((0, 0), "filling", font=ttf, fill=500, features=["-liga"])
target = "Tests/images/test_ligature_features.png" target = "Tests/images/test_ligature_features.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
liga_size = ttf.getsize("fi", features=["-liga"]) liga_size = ttf.getsize("fi", features=["-liga"])
assert liga_size == (13, 19) assert liga_size == (13, 19)
@ -167,8 +157,7 @@ def test_kerning_features():
draw.text((0, 0), "TeToAV", font=ttf, fill=500, features=["-kern"]) draw.text((0, 0), "TeToAV", font=ttf, fill=500, features=["-kern"])
target = "Tests/images/test_kerning_features.png" target = "Tests/images/test_kerning_features.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
def test_arabictext_features(): def test_arabictext_features():
@ -185,8 +174,7 @@ def test_arabictext_features():
) )
target = "Tests/images/test_arabictext_features.png" target = "Tests/images/test_arabictext_features.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
def test_x_max_and_y_offset(): def test_x_max_and_y_offset():
@ -197,8 +185,7 @@ def test_x_max_and_y_offset():
draw.text((0, 0), "لح", font=ttf, fill=500) draw.text((0, 0), "لح", font=ttf, fill=500)
target = "Tests/images/test_x_max_and_y_offset.png" target = "Tests/images/test_x_max_and_y_offset.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
def test_language(): def test_language():
@ -209,8 +196,7 @@ def test_language():
draw.text((0, 0), "абвг", font=ttf, fill=500, language="sr") draw.text((0, 0), "абвг", font=ttf, fill=500, language="sr")
target = "Tests/images/test_language.png" target = "Tests/images/test_language.png"
with Image.open(target) as target_img: assert_image_similar_tofile(im, target, 0.5)
assert_image_similar(im, target_img, 0.5)
@pytest.mark.parametrize("mode", ("L", "1")) @pytest.mark.parametrize("mode", ("L", "1"))
@ -287,8 +273,7 @@ def test_anchor_ttb(anchor):
if str(ex) == "libraqm 0.7 or greater required for 'ttb' direction": if str(ex) == "libraqm 0.7 or greater required for 'ttb' direction":
pytest.skip("libraqm 0.7 or greater not available") pytest.skip("libraqm 0.7 or greater not available")
with Image.open(path) as expected: assert_image_similar_tofile(im, path, 1) # fails at 5
assert_image_similar(im, expected, 1) # fails at 5
combine_tests = ( combine_tests = (
@ -351,8 +336,7 @@ def test_combine(name, text, dir, anchor, epsilon):
if str(ex) == "libraqm 0.7 or greater required for 'ttb' direction": if str(ex) == "libraqm 0.7 or greater required for 'ttb' direction":
pytest.skip("libraqm 0.7 or greater not available") pytest.skip("libraqm 0.7 or greater not available")
with Image.open(path) as expected: assert_image_similar_tofile(im, path, epsilon)
assert_image_similar(im, expected, epsilon)
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -384,8 +368,7 @@ def test_combine_multiline(anchor, align):
d.rectangle(bbox, outline="red") d.rectangle(bbox, outline="red")
d.multiline_text((200, 200), text, fill="black", anchor=anchor, font=f, align=align) d.multiline_text((200, 200), text, fill="black", anchor=anchor, font=f, align=align)
with Image.open(path) as expected: assert_image_similar_tofile(im, path, 0.015)
assert_image_similar(im, expected, 0.015)
def test_anchor_invalid_ttb(): def test_anchor_invalid_ttb():

View File

@ -3,7 +3,7 @@ import pytest
from PIL import Image, ImageMorph, _imagingmorph from PIL import Image, ImageMorph, _imagingmorph
from .helper import assert_image_equal, hopper from .helper import assert_image_equal_tofile, hopper
def string_to_img(image_string): def string_to_img(image_string):
@ -57,8 +57,7 @@ def assert_img_equal_img_string(A, Bstring):
def test_str_to_img(): def test_str_to_img():
with Image.open("Tests/images/morph_a.png") as im: assert_image_equal_tofile(A, "Tests/images/morph_a.png")
assert_image_equal(A, im)
def create_lut(): def create_lut():

View File

@ -5,6 +5,7 @@ from PIL import Image, ImageDraw, ImageOps, ImageStat, features
from .helper import ( from .helper import (
assert_image_equal, assert_image_equal,
assert_image_similar, assert_image_similar,
assert_image_similar_tofile,
assert_tuple_approx_equal, assert_tuple_approx_equal,
hopper, hopper,
) )
@ -112,10 +113,9 @@ def test_pad():
new_im = ImageOps.pad(im, new_size, color=color, centering=centering) new_im = ImageOps.pad(im, new_size, color=color, centering=centering)
assert new_im.size == new_size assert new_im.size == new_size
with Image.open( assert_image_similar_tofile(
"Tests/images/imageops_pad_" + label + "_" + str(i) + ".jpg" new_im, "Tests/images/imageops_pad_" + label + "_" + str(i) + ".jpg", 6
) as target: )
assert_image_similar(new_im, target, 6)
def test_pil163(): def test_pil163():

View File

@ -2,7 +2,7 @@ import pytest
from PIL import Image, ImagePalette from PIL import Image, ImagePalette
from .helper import assert_image_equal from .helper import assert_image_equal_tofile
def test_sanity(): def test_sanity():
@ -141,8 +141,7 @@ def test_2bit_palette(tmp_path):
img.putpalette(b"\xFF\x00\x00\x00\xFF\x00\x00\x00\xFF") # RGB img.putpalette(b"\xFF\x00\x00\x00\xFF\x00\x00\x00\xFF") # RGB
img.save(outfile, format="PNG") img.save(outfile, format="PNG")
with Image.open(outfile) as reloaded: assert_image_equal_tofile(img, outfile)
assert_image_equal(img, reloaded)
def test_invalid_palette(): def test_invalid_palette():

View File

@ -1,8 +1,8 @@
import pytest import pytest
from PIL import Image, ImageQt from PIL import ImageQt
from .helper import assert_image_equal, hopper from .helper import assert_image_equal, assert_image_equal_tofile, hopper
pytestmark = pytest.mark.skipif( pytestmark = pytest.mark.skipif(
not ImageQt.qt_is_installed, reason="Qt bindings are not installed" not ImageQt.qt_is_installed, reason="Qt bindings are not installed"
@ -40,5 +40,4 @@ def test_sanity(tmp_path):
data.save(tempfile) data.save(tempfile)
# Check that it actually worked. # Check that it actually worked.
with Image.open(tempfile) as reloaded: assert_image_equal_tofile(src, tempfile)
assert_image_equal(reloaded, src)