[3.12] gh-114149: [Enum] revert GH-114196 and add more tuple-subclass tests (GH-11421...

gh-114149: [Enum] revert GH-114160 and add more tuple-subclass tests (GH-114215)

This reverts commit 05e142b1543eb9662d6cc33722e7e16250c9219f.
(cherry picked from commit 4c7e09d0129dafddba58979ced9580f856f65efa)

Co-authored-by: Ethan Furman <ethan@stoneleaf.us>

diff --git a/Lib/enum.py b/Lib/enum.py
index 0cf88471d504a2a2943f5a72238f8bc3b217334f..1502bfe9158520ddf192c7ba09b7682f8986d11d 100644 (file)
--- a/Lib/enum.py
+++ b/Lib/enum.py
@@ -254,7 +254,7 @@ class _proto_member:
         delattr(enum_class, member_name)
         # second step: create member based on enum_class
         value = self.value
-        if type(value) is not tuple:
+        if not isinstance(value, tuple):
             args = (value, )
         else:
             args = value
@@ -1757,7 +1757,7 @@ def _simple_enum(etype=Enum, *, boundary=None, use_args=None):
                 else:
                     # create the member
                     if use_args:
-                        if type(value) is not tuple:
+                        if not isinstance(value, tuple):
                             value = (value, )
                         member = new_member(enum_class, *value)
                         value = value[0]
@@ -1807,7 +1807,7 @@ def _simple_enum(etype=Enum, *, boundary=None, use_args=None):
                 else:
                     # create the member
                     if use_args:
-                        if type(value) is not tuple:
+                        if not isinstance(value, tuple):
                             value = (value, )
                         member = new_member(enum_class, *value)
                         value = value[0]

diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index 2db3b2f95fc845fd65dce6c4f8437e02857c0190..23be142c24de0fa1e18a3c38a871bfde9cfdb5ab 100644 (file)
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -3163,19 +3163,34 @@ class TestSpecial(unittest.TestCase):
                 [x.value for x in NTEnum],
                 [TTuple(id=0, a=0, blist=[]), TTuple(id=1, a=2, blist=[4]), TTuple(id=2, a=4, blist=[0, 1, 2])],
                 )
+
+        self.assertRaises(AttributeError, getattr, NTEnum.NONE, 'id')
+        #
+        class NTCEnum(TTuple, Enum):
+            NONE = 0, 0, []
+            A = 1, 2, [4]
+            B = 2, 4, [0, 1, 2]
+        self.assertEqual(repr(NTCEnum.NONE), "<NTCEnum.NONE: TTuple(id=0, a=0, blist=[])>")
+        self.assertEqual(NTCEnum.NONE.value, TTuple(id=0, a=0, blist=[]))
+        self.assertEqual(NTCEnum.NONE.id, 0)
+        self.assertEqual(NTCEnum.A.a, 2)
+        self.assertEqual(NTCEnum.B.blist, [0, 1 ,2])
+        self.assertEqual(
+                [x.value for x in NTCEnum],
+                [TTuple(id=0, a=0, blist=[]), TTuple(id=1, a=2, blist=[4]), TTuple(id=2, a=4, blist=[0, 1, 2])],
+                )
         #
         class NTDEnum(Enum):
-            def __new__(cls, t_value):
+            def __new__(cls, id, a, blist):
                 member = object.__new__(cls)
-                member._value_ = t_value[0]
-                member.id = t_value[0]
-                member.a = t_value[1]
-                member.blist = t_value[2]
+                member.id = id
+                member.a = a
+                member.blist = blist
                 return member
             NONE = TTuple(0, 0, [])
             A = TTuple(1, 2, [4])
             B = TTuple(2, 4, [0, 1, 2])
-        self.assertEqual(repr(NTDEnum.NONE), "<NTDEnum.NONE: 0>")
+        self.assertEqual(repr(NTDEnum.NONE), "<NTDEnum.NONE: TTuple(id=0, a=0, blist=[])>")
         self.assertEqual(NTDEnum.NONE.id, 0)
         self.assertEqual(NTDEnum.A.a, 2)
         self.assertEqual(NTDEnum.B.blist, [0, 1 ,2])