mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-10 11:58:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

[Enum] improve pickle support (#26666)

Browse source 
search all bases for a __reduce__ style method; if a __new__ method is
found first the enum will be made unpicklable
This commit is contained in:
Ethan Furman 2021-06-11 01:26:32 -07:00 • committed by GitHub
parent 3ce35bfbbe
commit e9726314df
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 65 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

36
Lib/test/test_enum.py
View file

@ -594,13 +594,49 @@ class TestEnum(unittest.TestCase):
def test_inherited_data_type(self):
class HexInt(int):
__qualname__ = 'HexInt'
def __repr__(self):
return hex(self)
class MyEnum(HexInt, enum.Enum):
__qualname__ = 'MyEnum'
A = 1
B = 2
C = 3
self.assertEqual(repr(MyEnum.A), '<MyEnum.A: 0x1>')
globals()['HexInt'] = HexInt
globals()['MyEnum'] = MyEnum
test_pickle_dump_load(self.assertIs, MyEnum.A)
test_pickle_dump_load(self.assertIs, MyEnum)
#
class SillyInt(HexInt):
__qualname__ = 'SillyInt'
pass
class MyOtherEnum(SillyInt, enum.Enum):
__qualname__ = 'MyOtherEnum'
D = 4
E = 5
F = 6
self.assertIs(MyOtherEnum._member_type_, SillyInt)
globals()['SillyInt'] = SillyInt
globals()['MyOtherEnum'] = MyOtherEnum
test_pickle_dump_load(self.assertIs, MyOtherEnum.E)
test_pickle_dump_load(self.assertIs, MyOtherEnum)
#
class BrokenInt(int):
__qualname__ = 'BrokenInt'
def __new__(cls, value):
return int.__new__(cls, value)
class MyBrokenEnum(BrokenInt, Enum):
__qualname__ = 'MyBrokenEnum'
G = 7
H = 8
I = 9
self.assertIs(MyBrokenEnum._member_type_, BrokenInt)
self.assertIs(MyBrokenEnum(7), MyBrokenEnum.G)
globals()['BrokenInt'] = BrokenInt
globals()['MyBrokenEnum'] = MyBrokenEnum
test_pickle_exception(self.assertRaises, TypeError, MyBrokenEnum.G)
test_pickle_exception(self.assertRaises, PicklingError, MyBrokenEnum)
def test_too_many_data_types(self):
with self.assertRaisesRegex(TypeError, 'too many data types'):