Function register_table

datafusion/core/tests/dataframe/mod.rs:1749–1803

()

#[tokio::test]
async fn register_table() -> Result<()> {
    let df = test_table().await?.select_columns(&["c1", "c12"])?;
    let ctx = SessionContext::new();
    let df_impl = DataFrame::new(ctx.state(), df.logical_plan().clone());

    // register a dataframe as a table
    let table_provider = df_impl.clone().into_view();
    assert_eq!(table_provider.table_type(), TableType::View);
    ctx.register_table("test_table", table_provider)?;

    // pull the table out
    let table = ctx.table("test_table").await?;

    let group_expr = vec![col("c1")];
    let aggr_expr = vec![sum(col("c12"))];

    // check that we correctly read from the table
    let df_results = df_impl
        .aggregate(group_expr.clone(), aggr_expr.clone())?
        .collect()
        .await?;
    let table_results = &table.aggregate(group_expr, aggr_expr)?.collect().await?;

    assert_snapshot!(
        batches_to_sort_string(&df_results),
        @r"
    +----+-----------------------------+
    | c1 | sum(aggregate_test_100.c12) |
    +----+-----------------------------+
    | a  | 10.238448667882977          |
    | b  | 7.797734760124923           |
    | c  | 13.860958726523545          |
    | d  | 8.793968289758968           |
    | e  | 10.206140546981722          |
    +----+-----------------------------+
    "
    );

    // the results are the same as the results from the view, modulo the leaf table name
    assert_snapshot!(
        batches_to_sort_string(table_results),
        @r"
    +----+---------------------+
    | c1 | sum(test_table.c12) |
    +----+---------------------+
    | a  | 10.238448667882977  |
    | b  | 7.797734760124923   |
    | c  | 13.860958726523545  |
    | d  | 8.793968289758968   |
    | e  | 10.206140546981722  |
    +----+---------------------+
    "
    );
    Ok(())
}

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