From f6f3dab6724d72a60ebf0244843a82418f7d7740 Mon Sep 17 00:00:00 2001
From: Andrew Murray <radarhere@users.noreply.github.com>
Date: Wed, 27 Dec 2023 17:42:17 +1100
Subject: [PATCH] Import UnidentifiedImageError directly

---
 Tests/oss-fuzz/test_fuzzers.py | 4 ++--
 Tests/test_file_eps.py         | 4 ++--
 Tests/test_file_psd.py         | 6 +++---
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/Tests/oss-fuzz/test_fuzzers.py b/Tests/oss-fuzz/test_fuzzers.py
index 68834045a..186a0efd3 100644
--- a/Tests/oss-fuzz/test_fuzzers.py
+++ b/Tests/oss-fuzz/test_fuzzers.py
@@ -6,7 +6,7 @@ import fuzzers
 import packaging
 import pytest
 
-from PIL import Image, features
+from PIL import Image, UnidentifiedImageError, features
 from Tests.helper import skip_unless_feature
 
 if sys.platform.startswith("win32"):
@@ -42,7 +42,7 @@ def test_fuzz_images(path):
     except (
         Image.DecompressionBombError,
         Image.DecompressionBombWarning,
-        Image.UnidentifiedImageError,
+        UnidentifiedImageError,
     ):
         # Known Image.* exceptions
         assert True
diff --git a/Tests/test_file_eps.py b/Tests/test_file_eps.py
index c479c384a..360ae11b5 100644
--- a/Tests/test_file_eps.py
+++ b/Tests/test_file_eps.py
@@ -3,7 +3,7 @@ import io
 
 import pytest
 
-from PIL import EpsImagePlugin, Image, features
+from PIL import EpsImagePlugin, Image, UnidentifiedImageError, features
 
 from .helper import (
     assert_image_similar,
@@ -417,7 +417,7 @@ def test_emptyline():
 )
 def test_timeout(test_file):
     with open(test_file, "rb") as f:
-        with pytest.raises(Image.UnidentifiedImageError):
+        with pytest.raises(UnidentifiedImageError):
             with Image.open(f):
                 pass
 
diff --git a/Tests/test_file_psd.py b/Tests/test_file_psd.py
index 8b06ce2b1..d98f23356 100644
--- a/Tests/test_file_psd.py
+++ b/Tests/test_file_psd.py
@@ -3,7 +3,7 @@ import warnings
 
 import pytest
 
-from PIL import Image, PsdImagePlugin
+from PIL import Image, PsdImagePlugin, UnidentifiedImageError
 
 from .helper import assert_image_equal_tofile, assert_image_similar, hopper, is_pypy
 
@@ -146,11 +146,11 @@ def test_combined_larger_than_size():
     [
         (
             "Tests/images/timeout-1ee28a249896e05b83840ae8140622de8e648ba9.psd",
-            Image.UnidentifiedImageError,
+            UnidentifiedImageError,
         ),
         (
             "Tests/images/timeout-598843abc37fc080ec36a2699ebbd44f795d3a6f.psd",
-            Image.UnidentifiedImageError,
+            UnidentifiedImageError,
         ),
         ("Tests/images/timeout-c8efc3fded6426986ba867a399791bae544f59bc.psd", OSError),
         ("Tests/images/timeout-dedc7a4ebd856d79b4359bbcc79e8ef231ce38f6.psd", OSError),