test_impl: fix import

Signed-off-by: William Woodruff <william@trailofbits.com>
trailofbits · Mar 7, 2024 · 9020341 · 9020341
1 parent c4367df
commit 9020341
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test_impl.py b/test/test_impl.py
@@ -6,7 +6,7 @@
 import json
 import struct
 import sys
-from enum import IntEnum, StrEnum
+from enum import IntEnum
 from io import BytesIO
 
 import pytest
@@ -117,6 +117,8 @@ class X(IntEnum):
 
 @pytest.mark.skipif(sys.version_info < (3, 11), reason="StrEnum added in 3.11+")
 def test_dumps_strenum():
+    from enum import StrEnum
+
     # StrEnum is a subclass of str, so this should work transparently.
     class X(StrEnum):
         A = "foo"