diff options
author | Prince Roshan <princekrroshan01@gmail.com> | 2023-07-12 21:01:17 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-07-12 21:01:17 (GMT) |
commit | 357e9e9da3929cb9d55ea31896e66f488e44e8f2 (patch) | |
tree | 6e1d540f81d5d3e3bbd4cc4e4e4b83fcbb4f2d3b | |
parent | e4b88c1e4ac129b36f99a534387d64f7b8cda8ef (diff) | |
download | cpython-357e9e9da3929cb9d55ea31896e66f488e44e8f2.zip cpython-357e9e9da3929cb9d55ea31896e66f488e44e8f2.tar.gz cpython-357e9e9da3929cb9d55ea31896e66f488e44e8f2.tar.bz2 |
gh-106602: [Enum] Add __copy__ and __deepcopy__ (GH-106666)
-rw-r--r-- | Lib/enum.py | 6 | ||||
-rw-r--r-- | Lib/test/test_enum.py | 8 | ||||
-rw-r--r-- | Misc/NEWS.d/next/Library/2023-07-12-04-58-45.gh-issue-106602.dGCcXe.rst | 1 |
3 files changed, 15 insertions, 0 deletions
diff --git a/Lib/enum.py b/Lib/enum.py index 202f0da..0c985b2 100644 --- a/Lib/enum.py +++ b/Lib/enum.py @@ -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 diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py index adb1e0e..a286411 100644 --- a/Lib/test/test_enum.py +++ b/Lib/test/test_enum.py @@ -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: diff --git a/Misc/NEWS.d/next/Library/2023-07-12-04-58-45.gh-issue-106602.dGCcXe.rst b/Misc/NEWS.d/next/Library/2023-07-12-04-58-45.gh-issue-106602.dGCcXe.rst new file mode 100644 index 0000000..d9c122f --- /dev/null +++ b/Misc/NEWS.d/next/Library/2023-07-12-04-58-45.gh-issue-106602.dGCcXe.rst @@ -0,0 +1 @@ +Add __copy__ and __deepcopy__ in :mod:`enum` |