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