def METHOD_NAME(con):
    table = con.table("basic_table")
    table = table.mutate(id2=table["id"] * 2)
    result1 = table[["id"]].compile()
    result2 = table[["id", "id2"]].compile()
    result3 = table[[table, (table.id + 1).name("plus1")]].compile()
    result4 = table[[(table.id + 1).name("plus1"), table]].compile()
    df = table.compile()
    tm.assert_frame_equal(result1.toPandas(), df[["id"]].toPandas())
    tm.assert_frame_equal(result2.toPandas(), df[["id", "id2"]].toPandas())
    tm.assert_frame_equal(
        result3.toPandas(),
        df[[df.columns]].withColumn("plus1", df.id + 1).toPandas(),
    )
    tm.assert_frame_equal(
        result4.toPandas(),
        df.withColumn("plus1", df.id + 1)[["plus1", *df.columns]].toPandas(),
    )