Skip to content

Commit ad23135

Browse files
authored
Merge pull request #8747 from radarhere/monkeypatch
Use monkeypatch
2 parents 3c2310e + 8020d42 commit ad23135

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

Tests/test_file_gif.py

Lines changed: 13 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1345,7 +1345,7 @@ def test_save_I(tmp_path: Path) -> None:
13451345
assert_image_equal(reloaded.convert("L"), im.convert("L"))
13461346

13471347

1348-
def test_getdata() -> None:
1348+
def test_getdata(monkeypatch: pytest.MonkeyPatch) -> None:
13491349
# Test getheader/getdata against legacy values.
13501350
# Create a 'P' image with holes in the palette.
13511351
im = Image._wedge().resize((16, 16), Image.Resampling.NEAREST)
@@ -1354,23 +1354,21 @@ def test_getdata() -> None:
13541354

13551355
passed_palette = bytes(255 - i // 3 for i in range(768))
13561356

1357-
GifImagePlugin._FORCE_OPTIMIZE = True
1358-
try:
1359-
h = GifImagePlugin.getheader(im, passed_palette)
1360-
d = GifImagePlugin.getdata(im)
1357+
monkeypatch.setattr(GifImagePlugin, "_FORCE_OPTIMIZE", True)
13611358

1362-
import pickle
1359+
h = GifImagePlugin.getheader(im, passed_palette)
1360+
d = GifImagePlugin.getdata(im)
13631361

1364-
# Enable to get target values on pre-refactor version
1365-
# with open('Tests/images/gif_header_data.pkl', 'wb') as f:
1366-
# pickle.dump((h, d), f, 1)
1367-
with open("Tests/images/gif_header_data.pkl", "rb") as f:
1368-
(h_target, d_target) = pickle.load(f)
1362+
import pickle
13691363

1370-
assert h == h_target
1371-
assert d == d_target
1372-
finally:
1373-
GifImagePlugin._FORCE_OPTIMIZE = False
1364+
# Enable to get target values on pre-refactor version
1365+
# with open('Tests/images/gif_header_data.pkl', 'wb') as f:
1366+
# pickle.dump((h, d), f, 1)
1367+
with open("Tests/images/gif_header_data.pkl", "rb") as f:
1368+
(h_target, d_target) = pickle.load(f)
1369+
1370+
assert h == h_target
1371+
assert d == d_target
13741372

13751373

13761374
def test_lzw_bits() -> None:

0 commit comments

Comments
 (0)