]> granicus.if.org Git - python/commitdiff
Tweak the fix for test_traceback since the fix for it to run on its own broke
authorBrett Cannon <bcannon@gmail.com>
Tue, 27 Feb 2007 00:12:43 +0000 (00:12 +0000)
committerBrett Cannon <bcannon@gmail.com>
Tue, 27 Feb 2007 00:12:43 +0000 (00:12 +0000)
it under regrtest.  'traceback' likes to strip out the module name if it is
__main__ or __builtin__ but not in other cases.

Lib/test/test_traceback.py

index 51fb9e64a598cf1907a544b645daeb74121d3fdc..1b9e2f8b6b2685c40aa86ba7b2b0ef87090cf328 100644 (file)
@@ -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)