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)