Merge pull request #16 from WindSoilder/suppress_copy

remember to copy __suppress_context__ for ExceptionGroup.

Author: WindSoilder

exceptiongroup/__init__.py

@@ -50,6 +50,10 @@ def __copy__(self):
         new_group.__traceback__ = self.__traceback__
         new_group.__context__ = self.__context__
         new_group.__cause__ = self.__cause__
+        # Setting __cause__ also implicitly sets the __suppress_context__
+        # attribute to True.  So we should copy __suppress_context__ attribute
+        # last, after copying __cause__.
+        new_group.__suppress_context__ = self.__suppress_context__
         return new_group
 
     def __str__(self):

exceptiongroup/_tests/test_exceptiongroup.py

@@ -1,8 +1,16 @@
+import copy
 import pytest
 
 from exceptiongroup import ExceptionGroup
 
 
+def raise_group():
+    try:
+        1 / 0
+    except Exception as e:
+        raise ExceptionGroup("ManyError", [e], [str(e)]) from e
+
+
 def test_exception_group_init():
     memberA = ValueError("A")
     memberB = RuntimeError("B")
@@ -47,3 +55,30 @@ def test_exception_group_str():
     assert "ExceptionGroup: " in repr(group)
     assert "memberA" in repr(group)
     assert "memberB" in repr(group)
+
+
+def test_exception_group_copy():
+    try:
+        raise_group()  # the exception is raise by `raise...from..`
+    except ExceptionGroup as e:
+        group = e
+
+    another_group = copy.copy(group)
+    assert another_group.message == group.message
+    assert another_group.exceptions == group.exceptions
+    assert another_group.sources == group.sources
+    assert another_group.__traceback__ is group.__traceback__
+    assert another_group.__cause__ is group.__cause__
+    assert another_group.__context__ is group.__context__
+    assert another_group.__suppress_context__ is group.__suppress_context__
+    assert another_group.__cause__ is not None
+    assert another_group.__context__ is not None
+    assert another_group.__suppress_context__ is True
+
+    # doing copy when __suppress_context__ is False
+    group.__suppress_context__ = False
+    another_group = copy.copy(group)
+    assert another_group.__cause__ is group.__cause__
+    assert another_group.__context__ is group.__context__
+    assert another_group.__suppress_context__ is group.__suppress_context__
+    assert another_group.__suppress_context__ is False

exceptiongroup/_tests/test_tools.py

@@ -112,6 +112,8 @@ def test_split_and_check_attributes_same():
     assert matched.__traceback__ is new_group.__traceback__
     assert matched.__cause__ is new_group.__cause__
     assert matched.__context__ is new_group.__context__
+    assert matched.__suppress_context__ is new_group.__suppress_context__
     assert unmatched.__traceback__ is new_group.__traceback__
     assert unmatched.__cause__ is new_group.__cause__
     assert unmatched.__context__ is new_group.__context__
+    assert unmatched.__suppress_context__ is new_group.__suppress_context__