Function test_aggregate_alias

datafusion/core/tests/dataframe/mod.rs:1013–1042

()

#[tokio::test]
async fn test_aggregate_alias() -> Result<()> {
    let df = test_table().await?;

    let df = df
        // GROUP BY `c2 + 1`
        .aggregate(vec![col("c2") + lit(1)], vec![])?
        // SELECT `c2 + 1` as c2
        .select(vec![(col("c2") + lit(1)).alias("c2")])?
        // GROUP BY c2 as "c2" (alias in expr is not supported by SQL)
        .aggregate(vec![col("c2").alias("c2")], vec![])?;

    let df_results = df.collect().await?;

    assert_snapshot!(
        batches_to_sort_string(&df_results),
        @r"
    +----+
    | c2 |
    +----+
    | 2  |
    | 3  |
    | 4  |
    | 5  |
    | 6  |
    +----+
    "
    );

    Ok(())
}

#[tokio::test]
async fn test_aggregate_with_union() -> Result<()> {