diff --git a/Tests/test_file_gimppalette.py b/Tests/test_file_gimppalette.py index 23bb8aaabfc..ab45664af6a 100644 --- a/Tests/test_file_gimppalette.py +++ b/Tests/test_file_gimppalette.py @@ -23,6 +23,7 @@ def test_sanity(): def test_large_file_is_truncated(): import warnings from unittest.mock import patch + try: original_value = GimpPaletteFile._max_file_size GimpPaletteFile._max_file_size = 100 @@ -35,6 +36,7 @@ def test_large_file_is_truncated(): finally: GimpPaletteFile._max_file_size = original_value + def test_get_palette(): # Arrange with open("Tests/images/custom_gimp_palette.gpl", "rb") as fp: diff --git a/src/PIL/GimpPaletteFile.py b/src/PIL/GimpPaletteFile.py index ff5d6e10f19..2d7104b202a 100644 --- a/src/PIL/GimpPaletteFile.py +++ b/src/PIL/GimpPaletteFile.py @@ -28,7 +28,7 @@ class GimpPaletteFile: #: override if reading larger palettes is needed max_colors = 256 _max_line_size = 100 - _max_file_size = 2 ** 20 + _max_file_size = 2**20 def __init__(self, fp): @@ -47,7 +47,8 @@ def __init__(self, fp): read += len(s) if read >= self._max_file_size: warnings.warn( - f"Palette file truncated at {self._max_file_size - len(s)} bytes") + f"Palette file truncated at {self._max_file_size - len(s)} bytes" + ) break # skip fields and comment lines