python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-05-16 13:53:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

parametrize tests

Browse Source
This commit is contained in:
Yay295 2022-08-24 07:43:49 -05:00 committed by GitHub
parent fa591e1198
commit a7f7f6ac05
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 78 additions and 81 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

119
Tests/test_image_transform.py
View File

@ -75,12 +75,15 @@ class TestImageTransform:
assert_image_equal(transformed, scaled) assert_image_equal(transformed, scaled)
def test_fill(self): @pytest.mark.parametrize(
for mode, pixel in [ "mode,pixel",
["RGB", (255, 0, 0)], (
["RGBA", (255, 0, 0, 255)], ("RGB", (255, 0, 0)),
["LA", (76, 0)], ("RGBA", (255, 0, 0, 255)),
]: ("LA", (76, 0)),
),
)
def test_fill(self, mode, pixel):
im = hopper(mode) im = hopper(mode)
(w, h) = im.size (w, h) = im.size
transformed = im.transform( transformed = im.transform(
@ -90,7 +93,6 @@ class TestImageTransform:
Image.Resampling.BILINEAR, Image.Resampling.BILINEAR,
fillcolor="red", fillcolor="red",
) )
assert transformed.getpixel((w - 1, h - 1)) == pixel assert transformed.getpixel((w - 1, h - 1)) == pixel
def test_mesh(self): def test_mesh(self):
@ -222,14 +224,12 @@ class TestImageTransform:
with pytest.raises(ValueError): with pytest.raises(ValueError):
im.transform((100, 100), None) im.transform((100, 100), None)
def test_unknown_resampling_filter(self): @pytest.mark.parametrize("resample", (Image.Resampling.BOX, "unknown"))
def test_unknown_resampling_filter(self, resample):
with hopper() as im: with hopper() as im:
(w, h) = im.size (w, h) = im.size
for resample in (Image.Resampling.BOX, "unknown"):
with pytest.raises(ValueError): with pytest.raises(ValueError):
im.transform( im.transform((100, 100), Image.Transform.EXTENT, (0, 0, w, h), resample)
(100, 100), Image.Transform.EXTENT, (0, 0, w, h), resample
)
class TestImageTransformAffine: class TestImageTransformAffine:
@ -239,7 +239,16 @@ class TestImageTransformAffine:
im = hopper("RGB") im = hopper("RGB")
return im.crop((10, 20, im.width - 10, im.height - 20)) return im.crop((10, 20, im.width - 10, im.height - 20))
def _test_rotate(self, deg, transpose): @pytest.mark.parametrize(
"deg,transpose",
(
(0, None),
(90, Image.Transpose.ROTATE_90),
(180, Image.Transpose.ROTATE_180),
(270, Image.Transpose.ROTATE_270),
),
)
def test_rotate(self, deg, transpose):
im = self._test_image() im = self._test_image()
angle = -math.radians(deg) angle = -math.radians(deg)
@ -271,78 +280,66 @@ class TestImageTransformAffine:
) )
assert_image_equal(transposed, transformed) assert_image_equal(transposed, transformed)
def test_rotate_0_deg(self): @pytest.mark.parametrize(
self._test_rotate(0, None) "scale,epsilonscale",
(
def test_rotate_90_deg(self): (1.1, 6.9),
self._test_rotate(90, Image.Transpose.ROTATE_90) (1.5, 5.5),
(2.0, 5.5),
def test_rotate_180_deg(self): (2.3, 3.7),
self._test_rotate(180, Image.Transpose.ROTATE_180) (2.5, 3.7),
),
def test_rotate_270_deg(self): )
self._test_rotate(270, Image.Transpose.ROTATE_270) @pytest.mark.parametrize(
"resample,epsilon",
def _test_resize(self, scale, epsilonscale): (
(Image.Resampling.NEAREST, 0),
(Image.Resampling.BILINEAR, 2),
(Image.Resampling.BICUBIC, 1),
),
)
def test_resize(self, scale, epsilonscale, resample, epsilon):
im = self._test_image() im = self._test_image()
size_up = int(round(im.width * scale)), int(round(im.height * scale)) size_up = int(round(im.width * scale)), int(round(im.height * scale))
matrix_up = [1 / scale, 0, 0, 0, 1 / scale, 0, 0, 0] matrix_up = [1 / scale, 0, 0, 0, 1 / scale, 0, 0, 0]
matrix_down = [scale, 0, 0, 0, scale, 0, 0, 0] matrix_down = [scale, 0, 0, 0, scale, 0, 0, 0]
for resample, epsilon in [
(Image.Resampling.NEAREST, 0),
(Image.Resampling.BILINEAR, 2),
(Image.Resampling.BICUBIC, 1),
]:
transformed = im.transform(size_up, self.transform, matrix_up, resample) transformed = im.transform(size_up, self.transform, matrix_up, resample)
transformed = transformed.transform( transformed = transformed.transform(
im.size, self.transform, matrix_down, resample im.size, self.transform, matrix_down, resample
) )
assert_image_similar(transformed, im, epsilon * epsilonscale) assert_image_similar(transformed, im, epsilon * epsilonscale)
def test_resize_1_1x(self): @pytest.mark.parametrize(
self._test_resize(1.1, 6.9) "x,y,epsilonscale",
(
def test_resize_1_5x(self): (0.1, 0, 3.7),
self._test_resize(1.5, 5.5) (0.6, 0, 9.1),
(50, 50, 0),
def test_resize_2_0x(self): ),
self._test_resize(2.0, 5.5) )
@pytest.mark.parametrize(
def test_resize_2_3x(self): "resample,epsilon",
self._test_resize(2.3, 3.7) (
(Image.Resampling.NEAREST, 0),
def test_resize_2_5x(self): (Image.Resampling.BILINEAR, 1.5),
self._test_resize(2.5, 3.7) (Image.Resampling.BICUBIC, 1),
),
def _test_translate(self, x, y, epsilonscale): )
def test_translate(self, x, y, epsilonscale, resample, epsilon):
im = self._test_image() im = self._test_image()
size_up = int(round(im.width + x)), int(round(im.height + y)) size_up = int(round(im.width + x)), int(round(im.height + y))
matrix_up = [1, 0, -x, 0, 1, -y, 0, 0] matrix_up = [1, 0, -x, 0, 1, -y, 0, 0]
matrix_down = [1, 0, x, 0, 1, y, 0, 0] matrix_down = [1, 0, x, 0, 1, y, 0, 0]
for resample, epsilon in [
(Image.Resampling.NEAREST, 0),
(Image.Resampling.BILINEAR, 1.5),
(Image.Resampling.BICUBIC, 1),
]:
transformed = im.transform(size_up, self.transform, matrix_up, resample) transformed = im.transform(size_up, self.transform, matrix_up, resample)
transformed = transformed.transform( transformed = transformed.transform(
im.size, self.transform, matrix_down, resample im.size, self.transform, matrix_down, resample
) )
assert_image_similar(transformed, im, epsilon * epsilonscale) assert_image_similar(transformed, im, epsilon * epsilonscale)
def test_translate_0_1(self):
self._test_translate(0.1, 0, 3.7)
def test_translate_0_6(self):
self._test_translate(0.6, 0, 9.1)
def test_translate_50(self):
self._test_translate(50, 50, 0)
class TestImageTransformPerspective(TestImageTransformAffine): class TestImageTransformPerspective(TestImageTransformAffine):
# Repeat all tests for AFFINE transformations with PERSPECTIVE # Repeat all tests for AFFINE transformations with PERSPECTIVE