Show More
@@ -609,10 +609,15 class TestModules(tt.TempFileMixin, unittest.TestCase): | |||||
609 |
|
609 | |||
610 | class Negator(ast.NodeTransformer): |
|
610 | class Negator(ast.NodeTransformer): | |
611 | """Negates all number literals in an AST.""" |
|
611 | """Negates all number literals in an AST.""" | |
|
612 | ||||
612 | def visit_Num(self, node): |
|
613 | def visit_Num(self, node): | |
613 | node.n = -node.n |
|
614 | node.n = -node.n | |
614 | return node |
|
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 | class TestAstTransform(unittest.TestCase): |
|
621 | class TestAstTransform(unittest.TestCase): | |
617 | def setUp(self): |
|
622 | def setUp(self): | |
618 | self.negator = Negator() |
|
623 | self.negator = Negator() | |
@@ -680,6 +685,11 class IntegerWrapper(ast.NodeTransformer): | |||||
680 | args=[node], keywords=[]) |
|
685 | args=[node], keywords=[]) | |
681 | return node |
|
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 | class TestAstTransform2(unittest.TestCase): |
|
693 | class TestAstTransform2(unittest.TestCase): | |
684 | def setUp(self): |
|
694 | def setUp(self): | |
685 | self.intwrapper = IntegerWrapper() |
|
695 | self.intwrapper = IntegerWrapper() | |
@@ -723,6 +733,11 class ErrorTransformer(ast.NodeTransformer): | |||||
723 | def visit_Num(self, node): |
|
733 | def visit_Num(self, node): | |
724 | raise ValueError("test") |
|
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 | class TestAstTransformError(unittest.TestCase): |
|
741 | class TestAstTransformError(unittest.TestCase): | |
727 | def test_unregistering(self): |
|
742 | def test_unregistering(self): | |
728 | err_transformer = ErrorTransformer() |
|
743 | err_transformer = ErrorTransformer() |
General Comments 0
You need to be logged in to leave comments.
Login now