Parametrized test

This commit is contained in:
Andrew Murray 2022-09-11 09:20:45 +10:00 committed by GitHub
parent beb7b4d0f6
commit ed016f8f5a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -84,22 +84,23 @@ class TestFileTiff:
with Image.open("Tests/images/multipage.tiff") as im:
im.load()
def test_get_child_images(self):
def check(ims, sizes):
assert len(ims) == len(sizes)
for i, im in enumerate(ims):
w = sizes[i]
expected = Image.new("RGB", (w, w), "#f00")
assert_image_similar(im, expected, 1)
with Image.open("Tests/images/child_ifd.tiff") as im:
@pytest.mark.parametrize(
"path, sizes",
(
("Tests/images/hopper.tif", ()),
("Tests/images/child_ifd.tiff", (16, 8)),
("Tests/images/child_ifd_jpeg.tiff", (20,)),
),
)
def test_get_child_images(self, path, sizes):
with Image.open(path) as im:
ims = im.get_child_images()
check(ims, (16, 8))
with Image.open("Tests/images/child_ifd_jpeg.tiff") as im:
ims = im.get_child_images()
check(ims, (20,))
assert len(ims) == len(sizes)
for i, im in enumerate(ims):
w = sizes[i]
expected = Image.new("RGB", (w, w), "#f00")
assert_image_similar(im, expected, 1)
def test_mac_tiff(self):
# Read RGBa images from macOS [@PIL136]