[Enum] EnumMeta.__prepare__ now accepts **kwds (#23917)
This commit is contained in:
parent
7ec59d8861
commit
6ec0adefad
|
@ -178,7 +178,7 @@ class EnumMeta(type):
|
|||
Metaclass for Enum
|
||||
"""
|
||||
@classmethod
|
||||
def __prepare__(metacls, cls, bases):
|
||||
def __prepare__(metacls, cls, bases, **kwds):
|
||||
# check that previous enum members do not exist
|
||||
metacls._check_for_existing_members(cls, bases)
|
||||
# create the namespace dict
|
||||
|
|
|
@ -2119,7 +2119,7 @@ class TestEnum(unittest.TestCase):
|
|||
one = '1'
|
||||
two = b'2', 'ascii', 9
|
||||
|
||||
def test_init_subclass(self):
|
||||
def test_init_subclass_calling(self):
|
||||
class MyEnum(Enum):
|
||||
def __init_subclass__(cls, **kwds):
|
||||
super(MyEnum, cls).__init_subclass__(**kwds)
|
||||
|
@ -2155,6 +2155,16 @@ class TestEnum(unittest.TestCase):
|
|||
self.assertFalse(NeverEnum.__dict__.get('_test1', False))
|
||||
self.assertFalse(NeverEnum.__dict__.get('_test2', False))
|
||||
|
||||
def test_init_subclass_parameter(self):
|
||||
class multiEnum(Enum):
|
||||
def __init_subclass__(cls, multi):
|
||||
for member in cls:
|
||||
member._as_parameter_ = multi * member.value
|
||||
class E(multiEnum, multi=3):
|
||||
A = 1
|
||||
B = 2
|
||||
self.assertEqual(E.A._as_parameter_, 3)
|
||||
self.assertEqual(E.B._as_parameter_, 6)
|
||||
|
||||
@unittest.skipUnless(
|
||||
sys.version_info[:2] == (3, 9),
|
||||
|
|
|
@ -0,0 +1,2 @@
|
|||
`EnumMeta.__prepare__` now accepts `**kwds` to properly support
|
||||
`__init_subclass__`
|
Loading…
Reference in New Issue