From: Brett Cannon <bcannon@gmail.com>
Date: Tue, 27 Feb 2007 00:12:43 +0000 (+0000)
Subject: Tweak the fix for test_traceback since the fix for it to run on its own broke
X-Git-Tag: v3.0a1~1157
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=44c526174d9296ce358ccee9652382e7ea5377f4;p=python

Tweak the fix for test_traceback since the fix for it to run on its own broke
it under regrtest.  'traceback' likes to strip out the module name if it is
__main__ or __builtin__ but not in other cases.
---

diff --git a/Lib/test/test_traceback.py b/Lib/test/test_traceback.py
index 51fb9e64a5..1b9e2f8b6b 100644
--- a/Lib/test/test_traceback.py
+++ b/Lib/test/test_traceback.py
@@ -118,7 +118,11 @@ def test():
         err = traceback.format_exception_only(X, X())
         self.assertEqual(len(err), 1)
         str_value = '<unprintable %s object>' % X.__name__
-        self.assertEqual(err[0], "%s: %s\n" % ( X.__name__, str_value))
+        if X.__module__ in ('__main__', '__builtin__'):
+            str_name = X.__name__
+        else:
+            str_name = '.'.join([X.__module__, X.__name__])
+        self.assertEqual(err[0], "%s: %s\n" % (str_name, str_value))
 
     def test_without_exception(self):
         err = traceback.format_exception_only(None, None)