From 8f5762ec5f739c5b99f295179870361588d12f23 Mon Sep 17 00:00:00 2001 From: Andrew Murray Date: Wed, 30 Jun 2021 20:32:48 +1000 Subject: [PATCH] Parametrized test --- Tests/test_imagepalette.py | 32 ++++++++++++++++++-------------- 1 file changed, 18 insertions(+), 14 deletions(-) diff --git a/Tests/test_imagepalette.py b/Tests/test_imagepalette.py index 9c4f8acfb..ecfbda1d8 100644 --- a/Tests/test_imagepalette.py +++ b/Tests/test_imagepalette.py @@ -49,23 +49,27 @@ def test_getcolor(): palette.getcolor("unknown") -def test_getcolor_not_special(): - im = Image.new("P", (1, 1)) - for index, palette in { +@pytest.mark.parametrize( + "index, palette", + [ # Test when the palette is not full - 0: ImagePalette.ImagePalette(), + (0, ImagePalette.ImagePalette()), # Test when the palette is full - 255: ImagePalette.ImagePalette("RGB", list(range(256)) * 3), - }.items(): - # Do not use transparency index as a new color - im.info["transparency"] = index - index1 = palette.getcolor((0, 0, 0), im) - assert index1 != index + (255, ImagePalette.ImagePalette("RGB", list(range(256)) * 3)), + ], +) +def test_getcolor_not_special(index, palette): + im = Image.new("P", (1, 1)) - # Do not use background index as a new color - im.info["background"] = index1 - index2 = palette.getcolor((0, 0, 1), im) - assert index2 not in (index, index1) + # Do not use transparency index as a new color + im.info["transparency"] = index + index1 = palette.getcolor((0, 0, 0), im) + assert index1 != index + + # Do not use background index as a new color + im.info["background"] = index1 + index2 = palette.getcolor((0, 0, 1), im) + assert index2 not in (index, index1) def test_file(tmp_path):