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