Function csv_explain_plans

datafusion/core/tests/sql/explain_analyze.rs:329–495

()

#[tokio::test]
async fn csv_explain_plans() {
    // This test verify the look of each plan in its full cycle plan creation

    let ctx = SessionContext::new();
    register_aggregate_csv_by_sql(&ctx).await;
    let sql = "EXPLAIN SELECT c1 FROM aggregate_test_100 where c2 > 10";

    // Logical plan
    // Create plan
    let msg = format!("Creating logical plan for '{sql}'");
    let dataframe = ctx.sql(sql).await.expect(&msg);
    let logical_schema = dataframe.schema();
    let plan = dataframe.logical_plan();

    //
    println!("SQL: {sql}");
    //
    // Verify schema
    let formatted = plan.display_indent_schema().to_string();
    let actual = formatted.trim();
    assert_snapshot!(
        actual,
        @r"
    Explain [plan_type:Utf8, plan:Utf8]
      Projection: aggregate_test_100.c1 [c1:Utf8View]
        Filter: aggregate_test_100.c2 > Int64(10) [c1:Utf8View, c2:Int8, c3:Int16, c4:Int16, c5:Int32, c6:Int64, c7:Int16, c8:Int32, c9:UInt32, c10:UInt64, c11:Float32, c12:Float64, c13:Utf8View]
          TableScan: aggregate_test_100 [c1:Utf8View, c2:Int8, c3:Int16, c4:Int16, c5:Int32, c6:Int64, c7:Int16, c8:Int32, c9:UInt32, c10:UInt64, c11:Float32, c12:Float64, c13:Utf8View]
    "
    );
    //
    // Verify the text format of the plan
    let formatted = plan.display_indent().to_string();
    let actual = formatted.trim();
    assert_snapshot!(
        actual,
        @r"
    Explain
      Projection: aggregate_test_100.c1
        Filter: aggregate_test_100.c2 > Int64(10)
          TableScan: aggregate_test_100
    "
    );
    //
    // verify the grahviz format of the plan
    let formatted = plan.display_graphviz().to_string();
    let actual = formatted.trim();
    assert_snapshot!(
        actual,
        @r#"
    // Begin DataFusion GraphViz Plan,
    // display it online here: https://dreampuf.github.io/GraphvizOnline

    digraph {
      subgraph cluster_1
      {
        graph[label="LogicalPlan"]
        2[shape=box label="Explain"]
        3[shape=box label="Projection: aggregate_test_100.c1"]