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

View File

@ -1218,6 +1218,12 @@ class Enum(metaclass=EnumType):
def __reduce_ex__(self, proto):
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
# `value` attributes of enum members while keeping some measure of
# protection from modification, while still allowing for an enumeration

View File

@ -804,9 +804,17 @@ class _MinimalOutputTests:
TE = self.MainEnum
copied = copy.copy(TE)
self.assertEqual(copied, TE)
self.assertIs(copied, TE)
deep = copy.deepcopy(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:

View File

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