From 27ab774dea4daee44c44c7befe029c13cedc7a26 Mon Sep 17 00:00:00 2001 From: Ethan Furman Date: Wed, 17 Jan 2024 09:47:11 -0800 Subject: [PATCH] gh-114149: [Enum] fix tuple subclass handling when using custom __new__ (GH-114160) (cherry picked from commit 33b47a2c2853066b549f242065f6c2e12e18b33b) Co-authored-by: Ethan Furman --- Lib/enum.py | 6 +++--- Lib/test/test_enum.py | 16 ++++++++++++++++ ...024-01-16-15-59-06.gh-issue-114149.LJ8IPm.rst | 1 + 3 files changed, 20 insertions(+), 3 deletions(-) create mode 100644 Misc/NEWS.d/next/Library/2024-01-16-15-59-06.gh-issue-114149.LJ8IPm.rst diff --git a/Lib/enum.py b/Lib/enum.py index 63833aaa717feb..95ea23cba8f986 100644 --- a/Lib/enum.py +++ b/Lib/enum.py @@ -247,7 +247,7 @@ def __set_name__(self, enum_class, member_name): delattr(enum_class, member_name) # second step: create member based on enum_class value = self.value - if not isinstance(value, tuple): + if type(value) is not tuple: args = (value, ) else: args = value @@ -1726,7 +1726,7 @@ def convert_class(cls): else: # create the member if use_args: - if not isinstance(value, tuple): + if type(value) is not tuple: value = (value, ) member = new_member(enum_class, *value) value = value[0] @@ -1773,7 +1773,7 @@ def convert_class(cls): else: # create the member if use_args: - if not isinstance(value, tuple): + if type(value) is not 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 ab2b45853c2214..d96159cf75dcb7 100644 --- a/Lib/test/test_enum.py +++ b/Lib/test/test_enum.py @@ -2865,6 +2865,22 @@ class NTEnum(Enum): [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])], ) + # + class NTDEnum(Enum): + def __new__(cls, t_value): + member = object.__new__(cls) + member._value_ = t_value[0] + member.id = t_value[0] + member.a = t_value[1] + member.blist = t_value[2] + return member + NONE = TTuple(0, 0, []) + A = TTuple(1, 2, [4]) + B = TTuple(2, 4, [0, 1, 2]) + self.assertEqual(repr(NTDEnum.NONE), "") + self.assertEqual(NTDEnum.NONE.id, 0) + self.assertEqual(NTDEnum.A.a, 2) + self.assertEqual(NTDEnum.B.blist, [0, 1 ,2]) def test_flag_with_custom_new(self): class FlagFromChar(IntFlag): diff --git a/Misc/NEWS.d/next/Library/2024-01-16-15-59-06.gh-issue-114149.LJ8IPm.rst b/Misc/NEWS.d/next/Library/2024-01-16-15-59-06.gh-issue-114149.LJ8IPm.rst new file mode 100644 index 00000000000000..1403d78d0d4905 --- /dev/null +++ b/Misc/NEWS.d/next/Library/2024-01-16-15-59-06.gh-issue-114149.LJ8IPm.rst @@ -0,0 +1 @@ +Enum: correctly handle tuple subclasses in custom ``__new__``.