From 2305ab854470ba70a0ba624dd68b039a58294b03 Mon Sep 17 00:00:00 2001 From: Adam Turner <9087854+aa-turner@users.noreply.github.com> Date: Wed, 2 Apr 2025 20:14:53 +0100 Subject: [PATCH] Use `enum._simple_enum` for `annotationlib.Format` --- Lib/annotationlib.py | 3 ++- Lib/test/test_annotationlib.py | 9 +++++++++ 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/Lib/annotationlib.py b/Lib/annotationlib.py index 42f1f3877514d9..d51369d0cc057d 100644 --- a/Lib/annotationlib.py +++ b/Lib/annotationlib.py @@ -20,7 +20,8 @@ ] -class Format(enum.IntEnum): +@enum._simple_enum(enum.IntEnum) +class Format: VALUE = 1 VALUE_WITH_FAKE_GLOBALS = 2 FORWARDREF = 3 diff --git a/Lib/test/test_annotationlib.py b/Lib/test/test_annotationlib.py index 495606b48ed2e8..f757b12233f965 100644 --- a/Lib/test/test_annotationlib.py +++ b/Lib/test/test_annotationlib.py @@ -3,6 +3,7 @@ import annotationlib import builtins import collections +import enum import functools import itertools import pickle @@ -51,6 +52,14 @@ def test_enum(self): self.assertEqual(Format.STRING.value, 4) self.assertEqual(Format.STRING, 4) + def test_simple_enum(self): + class Format(enum.IntEnum): + VALUE = 1 + VALUE_WITH_FAKE_GLOBALS = 2 + FORWARDREF = 3 + STRING = 4 + enum._test_simple_enum(Format, annotationlib.Format) + class TestForwardRefFormat(unittest.TestCase): def test_closure(self):