diff --git a/Tests/test_file_gimppalette.py b/Tests/test_file_gimppalette.py index f7913ff81..b58bb865e 100644 --- a/Tests/test_file_gimppalette.py +++ b/Tests/test_file_gimppalette.py @@ -21,19 +21,14 @@ def test_sanity(): def test_large_file_is_truncated(): - import warnings - + original_max_file_size = GimpPaletteFile._max_file_size try: - original_value = GimpPaletteFile._max_file_size GimpPaletteFile._max_file_size = 100 - with warnings.catch_warnings(): - warnings.simplefilter("error") - with pytest.raises(UserWarning): - with open("Tests/images/custom_gimp_palette.gpl", "rb") as fp: - GimpPaletteFile(fp) - + with open("Tests/images/custom_gimp_palette.gpl", "rb") as fp: + with pytest.warns(UserWarning): + GimpPaletteFile(fp) finally: - GimpPaletteFile._max_file_size = original_value + GimpPaletteFile._max_file_size = original_max_file_size def test_get_palette(): diff --git a/src/PIL/GimpPaletteFile.py b/src/PIL/GimpPaletteFile.py index b5fab2525..82121be3c 100644 --- a/src/PIL/GimpPaletteFile.py +++ b/src/PIL/GimpPaletteFile.py @@ -49,7 +49,7 @@ class GimpPaletteFile: warnings.warn( f"Palette file truncated at {self._max_file_size - len(s)} bytes" ) - break # pragma: no cover + break # skip fields and comment lines if re.match(rb"\w+:|#", s):