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

gh-106602: [Enum] Add __copy__ and __deepcopy__ (GH-106666)
(cherry picked from commit 357e9e9da3929cb9d55ea31896e66f488e44e8f2)

Co-authored-by: Prince Roshan <princekrroshan01@gmail.com>

diff --git a/Lib/enum.py b/Lib/enum.py
index 1184b5b13e6d706ce87273dd2cefa61e58714b26..1f447c878c1f84b411ea6ceb79d6b931da2c93d6 100644 (file)
--- a/Lib/enum.py
+++ b/Lib/enum.py
@@ -1227,6 +1227,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 442802823e7d0bbdc3e0042608b8f557b6de60d1..cc66875bde02a8857722e7c2705d2b6d7dfec021 100644 (file)
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -775,9 +775,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 (file)
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`