diff options
author | Benjamin Peterson <benjamin@python.org> | 2015-02-19 17:58:19 -0500 |
---|---|---|
committer | Benjamin Peterson <benjamin@python.org> | 2015-02-19 17:58:19 -0500 |
commit | 21db0b50addd267c7f0627c3d53b01bbba3eb2a6 (patch) | |
tree | 946b263c5d294fb9868cad2340cd58f57760d0e4 /Lib/test/test_code_module.py | |
parent | fa0d2fdb9dcbe2353af87e445313cdcebc898a67 (diff) | |
parent | 25d7d132b630f88298ffe921a9479f66db2f8bde (diff) | |
download | cpython-21db0b50addd267c7f0627c3d53b01bbba3eb2a6.tar.gz |
merge 3.4 (#23481)
Diffstat (limited to 'Lib/test/test_code_module.py')
-rw-r--r-- | Lib/test/test_code_module.py | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/Lib/test/test_code_module.py b/Lib/test/test_code_module.py index 7a80a808b1..9b177672b4 100644 --- a/Lib/test/test_code_module.py +++ b/Lib/test/test_code_module.py @@ -1,6 +1,7 @@ "Test InteractiveConsole and InteractiveInterpreter from code module" import sys import unittest +from textwrap import dedent from contextlib import ExitStack from unittest import mock from test import support @@ -78,6 +79,40 @@ class TestInteractiveConsole(unittest.TestCase): self.console.interact(banner='') self.assertEqual(len(self.stderr.method_calls), 1) + def test_cause_tb(self): + self.infunc.side_effect = ["raise ValueError('') from AttributeError", + EOFError('Finished')] + self.console.interact() + output = ''.join(''.join(call[1]) for call in self.stderr.method_calls) + expected = dedent(""" + AttributeError + + The above exception was the direct cause of the following exception: + + Traceback (most recent call last): + File "<console>", line 1, in <module> + ValueError + """) + self.assertIn(expected, output) + + def test_context_tb(self): + self.infunc.side_effect = ["try: ham\nexcept: eggs\n", + EOFError('Finished')] + self.console.interact() + output = ''.join(''.join(call[1]) for call in self.stderr.method_calls) + expected = dedent(""" + Traceback (most recent call last): + File "<console>", line 1, in <module> + NameError: name 'ham' is not defined + + During handling of the above exception, another exception occurred: + + Traceback (most recent call last): + File "<console>", line 2, in <module> + NameError: name 'eggs' is not defined + """) + self.assertIn(expected, output) + def test_main(): support.run_unittest(TestInteractiveConsole) |