Function aggregate

datafusion/core/tests/dataframe/mod.rs:768–802

()

#[tokio::test]
async fn aggregate() -> Result<()> {
    // build plan using DataFrame API
    // union so some of the distincts have a clearly distinct result
    let df = test_table().await?.union(test_table().await?)?;
    let group_expr = vec![col("c1")];
    let aggr_expr = vec![
        min(col("c4")).alias("min(c4)"),
        max(col("c4")).alias("max(c4)"),
        avg(col("c4")).alias("avg(c4)"),
        avg_distinct(col("c4")).alias("avg_distinct(c4)"),
        sum(col("c4")).alias("sum(c4)"),
        sum_distinct(col("c4")).alias("sum_distinct(c4)"),
        count(col("c4")).alias("count(c4)"),
        count_distinct(col("c4")).alias("count_distinct(c4)"),
    ];

    let df: Vec<RecordBatch> = df.aggregate(group_expr, aggr_expr)?.collect().await?;

    assert_snapshot!(
        batches_to_sort_string(&df),
        @r"
    +----+---------+---------+---------------------+---------------------+---------+------------------+-----------+--------------------+
    | c1 | min(c4) | max(c4) | avg(c4)             | avg_distinct(c4)    | sum(c4) | sum_distinct(c4) | count(c4) | count_distinct(c4) |
    +----+---------+---------+---------------------+---------------------+---------+------------------+-----------+--------------------+
    | a  | -28462  | 32064   | 306.04761904761904  | 306.04761904761904  | 12854   | 6427             | 42        | 21                 |
    | b  | -28070  | 25286   | 7732.315789473684   | 7732.315789473684   | 293828  | 146914           | 38        | 19                 |
    | c  | -30508  | 29106   | -1320.5238095238096 | -1320.5238095238096 | -55462  | -27731           | 42        | 21                 |
    | d  | -24558  | 31106   | 10890.111111111111  | 10890.111111111111  | 392044  | 196022           | 36        | 18                 |
    | e  | -31500  | 32514   | -4268.333333333333  | -4268.333333333333  | -179270 | -89635           | 42        | 21                 |
    +----+---------+---------+---------------------+---------------------+---------+------------------+-----------+--------------------+
    "
    );

    Ok(())
}

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