def test_replace_ast(self):
new_ast = AST.from_string("y = 2\n",
language=ASTLanguage.Python,
deepest=True)
new_root = AST.replace(self.root, self.statement, new_ast)
self.assertNotEqual(new_root.oid, self.root.oid)
self.assertEqual(1, len(new_root.children))
self.assertEqual("y = 2\n", new_root.source_text)
def test_replace_literal(self):
lhs = self.statement.children[0].children[0]
new_root = AST.replace(self.root, lhs, "y")
self.assertNotEqual(new_root.oid, self.root.oid)
self.assertEqual("y = 88\n", new_root.source_text)