def test_string_removal_from_expression(self): class StringBlanker(PymTransformer): def visit_Expr(self, _node): # pylint: disable-msg=no-self-use return None transformer = StringBlanker() ast = render.parse('i=0\n"contents"') ast = transformer.visit(ast) actual = render.render(ast) self.assertEqual('i = 0', actual)
def test_string_emptying(self): class StringBlanker(PymTransformer): def visit_Str(self, _node): # pylint: disable-msg=no-self-use return 'pass' transformer = StringBlanker() ast = render.parse('"contents"') ast = transformer.visit(ast) actual = render.render(ast) self.assertEqual('pass', actual)
def test_pass_removal(self): class PassRemover(PymTransformer): def visit_Pass(self, _node): # pylint: disable-msg=no-self-use return None transformer = PassRemover() ast = render.parse('pass') ast = transformer.visit(ast) actual = render.render(ast) self.assertEqual('', actual)
def re_render(module_text): tree = parse(module_text) recast_docstrings(tree) adjust_lines(tree) return render(tree)
def test_render_empty(self): self.assertFalse(render.render(' '))
def test_render_nothing(self): self.assertIsNone(render.render(None)) self.assertIsNone(render.render(''))
def test_render(self): expected = 'i = 0\ni += 1' node = parse(expected) actual = render.render(node) self.assertEqual(actual, expected)