Show More
@@ -609,10 +609,15 b' class TestModules(tt.TempFileMixin, unittest.TestCase):' | |||
|
609 | 609 | |
|
610 | 610 | class Negator(ast.NodeTransformer): |
|
611 | 611 | """Negates all number literals in an AST.""" |
|
612 | ||
|
612 | 613 | def visit_Num(self, node): |
|
613 | 614 | node.n = -node.n |
|
614 | 615 | return node |
|
615 | 616 | |
|
617 | if sys.version_info > (3,8): | |
|
618 | def visit_Constant(self, node): | |
|
619 | return self.visit_Num(node) | |
|
620 | ||
|
616 | 621 | class TestAstTransform(unittest.TestCase): |
|
617 | 622 | def setUp(self): |
|
618 | 623 | self.negator = Negator() |
@@ -680,6 +685,11 b' class IntegerWrapper(ast.NodeTransformer):' | |||
|
680 | 685 | args=[node], keywords=[]) |
|
681 | 686 | return node |
|
682 | 687 | |
|
688 | if sys.version_info > (3,8): | |
|
689 | def visit_Constant(self, node): | |
|
690 | return self.visit_Num(node) | |
|
691 | ||
|
692 | ||
|
683 | 693 | class TestAstTransform2(unittest.TestCase): |
|
684 | 694 | def setUp(self): |
|
685 | 695 | self.intwrapper = IntegerWrapper() |
@@ -723,6 +733,11 b' class ErrorTransformer(ast.NodeTransformer):' | |||
|
723 | 733 | def visit_Num(self, node): |
|
724 | 734 | raise ValueError("test") |
|
725 | 735 | |
|
736 | if sys.version_info > (3,8): | |
|
737 | def visit_Constant(self, node): | |
|
738 | return self.visit_Num(node) | |
|
739 | ||
|
740 | ||
|
726 | 741 | class TestAstTransformError(unittest.TestCase): |
|
727 | 742 | def test_unregistering(self): |
|
728 | 743 | err_transformer = ErrorTransformer() |
General Comments 0
You need to be logged in to leave comments.
Login now