Mirror
/
cpython
mirror of https://github.com/python/cpython
4
1
Fork 0
Code Issues Releases Wiki Activity

gh-106602: [Enum] Add __copy__ and __deepcopy__ (GH-106666)

This commit is contained in:
Prince Roshan 2023-07-13 02:31:17 +05:30 committed by GitHub
parent e4b88c1e4a
commit 357e9e9da3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Lib/enum.py
View File

@ -1218,6 +1218,12 @@ class Enum(metaclass=EnumType):
def __reduce_ex__(self, proto): def __reduce_ex__(self, proto):
return self.__class__, (self._value_, ) return self.__class__, (self._value_, )
def __deepcopy__(self,memo):
return self
def __copy__(self):
return self
# enum.property is used to provide access to the `name` and # enum.property is used to provide access to the `name` and
# `value` attributes of enum members while keeping some measure of # `value` attributes of enum members while keeping some measure of
# protection from modification, while still allowing for an enumeration # protection from modification, while still allowing for an enumeration

8
Lib/test/test_enum.py
View File

@ -804,9 +804,17 @@ class _MinimalOutputTests:
TE = self.MainEnum TE = self.MainEnum
copied = copy.copy(TE) copied = copy.copy(TE)
self.assertEqual(copied, TE) self.assertEqual(copied, TE)
self.assertIs(copied, TE)
deep = copy.deepcopy(TE) deep = copy.deepcopy(TE)
self.assertEqual(deep, TE) self.assertEqual(deep, TE)
self.assertIs(deep, TE)
def test_copy_member(self):
TE = self.MainEnum
copied = copy.copy(TE.first)
self.assertIs(copied, TE.first)
deep = copy.deepcopy(TE.first)
self.assertIs(deep, TE.first)
class _FlagTests: class _FlagTests:

1
Misc/NEWS.d/next/Library/2023-07-12-04-58-45.gh-issue-106602.dGCcXe.rst Normal file
View File

@ -0,0 +1 @@
Add __copy__ and __deepcopy__ in :mod:`enum`