Show More
@@ -434,6 +434,21 class InteractiveShellTestCase(unittest.TestCase): | |||
|
434 | 434 | found = ip._ofind('a.foo', [('locals', locals())]) |
|
435 | 435 | nt.assert_is(found['obj'], A.foo) |
|
436 | 436 | |
|
437 | def test_custom_syntaxerror_exception(self): | |
|
438 | called = [] | |
|
439 | def my_handler(shell, etype, value, tb, tb_offset=None): | |
|
440 | called.append(etype) | |
|
441 | shell.showtraceback((etype, value, tb), tb_offset=tb_offset) | |
|
442 | ||
|
443 | ip.set_custom_exc((SyntaxError,), my_handler) | |
|
444 | try: | |
|
445 | ip.run_cell("1f") | |
|
446 | # Check that this was called, and only once. | |
|
447 | self.assertEqual(called, [SyntaxError]) | |
|
448 | finally: | |
|
449 | # Reset the custom exception hook | |
|
450 | ip.set_custom_exc((), None) | |
|
451 | ||
|
437 | 452 | def test_custom_exception(self): |
|
438 | 453 | called = [] |
|
439 | 454 | def my_handler(shell, etype, value, tb, tb_offset=None): |
General Comments 0
You need to be logged in to leave comments.
Login now