##// END OF EJS Templates
Return current node instead of None
Matthias Bussonnier -
Show More
@@ -618,6 +618,7 b' class Negator(ast.NodeTransformer):'
618 618 def visit_Constant(self, node):
619 619 if isinstance(node.value, int):
620 620 return self.visit_Num(node)
621 return node
621 622
622 623 class TestAstTransform(unittest.TestCase):
623 624 def setUp(self):
@@ -690,6 +691,7 b' class IntegerWrapper(ast.NodeTransformer):'
690 691 def visit_Constant(self, node):
691 692 if isinstance(node.value, int):
692 693 return self.visit_Num(node)
694 return node
693 695
694 696
695 697 class TestAstTransform2(unittest.TestCase):
@@ -739,6 +741,7 b' class ErrorTransformer(ast.NodeTransformer):'
739 741 def visit_Constant(self, node):
740 742 if isinstance(node.value, int):
741 743 return self.visit_Num(node)
744 return node
742 745
743 746
744 747 class TestAstTransformError(unittest.TestCase):
@@ -767,6 +770,7 b' class StringRejector(ast.NodeTransformer):'
767 770 def visit_Constant(self, node):
768 771 if isinstance(node.value, str):
769 772 raise InputRejected("test")
773 return node
770 774
771 775
772 776 class TestAstTransformInputRejection(unittest.TestCase):
General Comments 0
You need to be logged in to leave comments. Login now