summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_traceback.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_traceback.py')
-rw-r--r--Lib/test/test_traceback.py17
1 files changed, 17 insertions, 0 deletions
diff --git a/Lib/test/test_traceback.py b/Lib/test/test_traceback.py
index 5bd969d..e9df1ce 100644
--- a/Lib/test/test_traceback.py
+++ b/Lib/test/test_traceback.py
@@ -1378,6 +1378,23 @@ class TestTracebackException(unittest.TestCase):
exc = traceback.TracebackException(Exception, Exception("haven"), None)
self.assertEqual(list(exc.format()), ["Exception: haven\n"])
+ def test_print(self):
+ def f():
+ x = 12
+ try:
+ x/0
+ except Exception:
+ return sys.exc_info()
+ exc = traceback.TracebackException(*f(), capture_locals=True)
+ output = StringIO()
+ exc.print(file=output)
+ self.assertEqual(
+ output.getvalue().split('\n')[-4:],
+ [' x/0',
+ ' x = 12',
+ 'ZeroDivisionError: division by zero',
+ ''])
+
class MiscTest(unittest.TestCase):