Skip to content

Commit 356f56e

Browse files
committed
Test custom messages aren't overwritten
1 parent 2fd216a commit 356f56e

1 file changed

Lines changed: 11 additions & 0 deletions

File tree

Lib/test/test_exceptions.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2063,8 +2063,13 @@ class BareRaise:
20632063
def __getattr__(self, name):
20642064
raise AttributeError
20652065

2066+
class RaiseCustom:
2067+
def __getattr__(self, name):
2068+
raise AttributeError("custom")
2069+
20662070
self._test_generated_message(RaiseWithName(), "missing1")
20672071
self._test_generated_message(BareRaise(), "missing2")
2072+
self._test_generated_message(RaiseCustom(), "custom")
20682073

20692074
def test_module_getattr_attribute_error_message(self):
20702075
mod1 = ModuleType("raisewithname")
@@ -2077,8 +2082,14 @@ def bare_raise(name):
20772082
raise AttributeError
20782083
mod2.__getattr__ = bare_raise
20792084

2085+
mod3 = ModuleType("bareraise")
2086+
def raise_custom(name):
2087+
raise AttributeError("custom")
2088+
mod3.__getattr__ = raise_custom
2089+
20802090
self._test_generated_message(mod1, "missing3")
20812091
self._test_generated_message(mod2, "missing4")
2092+
self._test_generated_message(mod3, "custom")
20822093

20832094
# Note: name suggestion tests live in `test_traceback`.
20842095

0 commit comments

Comments
 (0)