summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_enum.py
diff options
context:
space:
mode:
authorEthan Furman <ethan@stoneleaf.us>2024-01-18 00:31:00 (GMT)
committerGitHub <noreply@github.com>2024-01-18 00:31:00 (GMT)
commit4c7e09d0129dafddba58979ced9580f856f65efa (patch)
tree5c84b14e48f05019974558d35a8179a2432b9a35 /Lib/test/test_enum.py
parent945540306c12116154d2e4cc6c17a8efd2290537 (diff)
downloadcpython-4c7e09d0129dafddba58979ced9580f856f65efa.zip
cpython-4c7e09d0129dafddba58979ced9580f856f65efa.tar.gz
cpython-4c7e09d0129dafddba58979ced9580f856f65efa.tar.bz2
gh-114149: [Enum] revert #114160 and add more tuple-subclass tests (GH-114215)
This reverts commit 05e142b1543eb9662d6cc33722e7e16250c9219f.
Diffstat (limited to 'Lib/test/test_enum.py')
-rw-r--r--Lib/test/test_enum.py27
1 files changed, 21 insertions, 6 deletions
diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index 48e0b02..d045739 100644
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -3201,19 +3201,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])