From dee933617e353b2b8a922593590347b5eb5e1bae Mon Sep 17 00:00:00 2001 From: Andrew Murray Date: Wed, 19 Mar 2025 21:49:21 +1100 Subject: [PATCH] Use monkeypatch --- Tests/test_file_gimppalette.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/Tests/test_file_gimppalette.py b/Tests/test_file_gimppalette.py index 1ad7d8168..4867753bc 100644 --- a/Tests/test_file_gimppalette.py +++ b/Tests/test_file_gimppalette.py @@ -22,15 +22,11 @@ def test_sanity() -> None: GimpPaletteFile(fp) -def test_large_file_is_truncated() -> None: - original_max_file_size = GimpPaletteFile._max_file_size - try: - GimpPaletteFile._max_file_size = 100 - with open("Tests/images/custom_gimp_palette.gpl", "rb") as fp: - with pytest.warns(UserWarning): - GimpPaletteFile(fp) - finally: - GimpPaletteFile._max_file_size = original_max_file_size +def test_large_file_is_truncated(monkeypatch: pytest.MonkeyPatch) -> None: + monkeypatch.setattr(GimpPaletteFile, "_max_file_size", 100) + with open("Tests/images/custom_gimp_palette.gpl", "rb") as fp: + with pytest.warns(UserWarning): + GimpPaletteFile(fp) def test_get_palette() -> None: