Show More
@@ -438,6 +438,22 b' class TestAstTransform(unittest.TestCase):' | |||
|
438 | 438 | with tt.AssertNotPrints('-55'): |
|
439 | 439 | ip.run_cell('print (n)') |
|
440 | 440 | |
|
441 | class ErrorTransformer(ast.NodeTransformer): | |
|
442 | """Throws an error when it sees a number.""" | |
|
443 | def visit_Num(self): | |
|
444 | raise ValueError("test") | |
|
445 | ||
|
446 | class TestAstTransformError(unittest.TestCase): | |
|
447 | def test_unregistering(self): | |
|
448 | err_transformer = ErrorTransformer() | |
|
449 | ip.ast_transformers.append(err_transformer) | |
|
450 | ||
|
451 | with tt.AssertPrints("unregister", channel='stderr'): | |
|
452 | ip.run_cell("1 + 2") | |
|
453 | ||
|
454 | # This should have been removed. | |
|
455 | nt.assert_not_in(err_transformer, ip.ast_transformers) | |
|
456 | ||
|
441 | 457 | def test__IPYTHON__(): |
|
442 | 458 | # This shouldn't raise a NameError, that's all |
|
443 | 459 | __IPYTHON__ |
General Comments 0
You need to be logged in to leave comments.
Login now