Function test_aggregate_subexpr

datafusion/core/tests/dataframe/mod.rs:1091–1125

()

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

    let group_expr = col("c2") + lit(1);
    let aggr_expr = sum(col("c3") + lit(2));

    let df = df
        // GROUP BY `c2 + 1`
        .aggregate(vec![group_expr.clone()], vec![aggr_expr.clone()])?
        // SELECT `c2 + 1` as c2 + 10, sum(c3 + 2) + 20
        // SELECT expressions contain aggr_expr and group_expr as subexpressions
        .select(vec![
            group_expr.alias("c2") + lit(10),
            (aggr_expr + lit(20)).alias("sum"),
        ])?;

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

    assert_snapshot!(
        batches_to_sort_string(&df_results),
        @r"
    +----------------+------+
    | c2 + Int32(10) | sum  |
    +----------------+------+
    | 12             | 431  |
    | 13             | 248  |
    | 14             | 453  |
    | 15             | 95   |
    | 16             | -146 |
    +----------------+------+
    "
    );

    Ok(())
}

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