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)
