Function test_grouping_sets

datafusion/core/tests/dataframe/mod.rs:3833–3872

()

#[tokio::test]
async fn test_grouping_sets() -> Result<()> {
    let grouping_set_expr = Expr::GroupingSet(GroupingSet::GroupingSets(vec![
        vec![col("a")],
        vec![col("b")],
        vec![col("a"), col("b")],
    ]));

    let df = create_test_table("test")
        .await?
        .aggregate(vec![grouping_set_expr], vec![count(col("a"))])?
        .sort(vec![
            Sort::new(col("a"), false, true),
            Sort::new(col("b"), false, true),
        ])?;

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

    assert_snapshot!(
       batches_to_string(&results),
        @r"
    +-----------+-----+---------------+
    | a         | b   | count(test.a) |
    +-----------+-----+---------------+
    |           | 100 | 1             |
    |           | 10  | 2             |
    |           | 1   | 1             |
    | abcDEF    |     | 1             |
    | abcDEF    | 1   | 1             |
    | abc123    |     | 1             |
    | abc123    | 10  | 1             |
    | CBAdef    |     | 1             |
    | CBAdef    | 10  | 1             |
    | 123AbcDef |     | 1             |
    | 123AbcDef | 100 | 1             |
    +-----------+-----+---------------+
    "
    );

    Ok(())
}

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