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

gh-106602: [Enum] Add __copy__ and __deepcopy__ (GH-106666) (cherry picked from commit 357e9e9) Co-authored-by: Prince Roshan <[email protected]>
python · Jul 12, 2023 · 77d9fdf · 77d9fdf
1 parent ec7b05a
commit 77d9fdf
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/Lib/enum.py b/Lib/enum.py
@@ -1242,6 +1242,12 @@ def __hash__(self):
     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
@@ -804,9 +804,17 @@ def test_copy(self):
         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
@@ -0,0 +1 @@
+Add __copy__ and __deepcopy__ in :mod:`enum`