diff --git a/Tests/test_file_wmf.py b/Tests/test_file_wmf.py index 71b6012c3..03444eb9d 100644 --- a/Tests/test_file_wmf.py +++ b/Tests/test_file_wmf.py @@ -35,6 +35,7 @@ def test_register_handler(tmp_path): self.methodCalled = True handler = TestHandler() + original_handler = WmfImagePlugin._handler WmfImagePlugin.register_handler(handler) im = hopper() @@ -43,7 +44,7 @@ def test_register_handler(tmp_path): assert handler.methodCalled # Restore the state before this test - WmfImagePlugin.register_handler(None) + WmfImagePlugin.register_handler(original_handler) def test_load_dpi_rounding():