fix erro on Count(Expr:Wildcard) with DataFrame API

Author: jiangzhx

datafusion/core/tests/dataframe.rs

@@ -35,6 +35,59 @@ use datafusion::{assert_batches_eq, assert_batches_sorted_eq};
 use datafusion_expr::expr::{GroupingSet, Sort};
 use datafusion_expr::{avg, col, count, lit, max, sum, Expr, ExprSchemable};
 
+#[tokio::test]
+async fn count_wildcard() -> Result<()> {
+    let ctx = SessionContext::new();
+
+    let testdata = datafusion::test_util::parquet_test_data();
+    let df = ctx
+        .read_parquet(
+            &format!("{testdata}/alltypes_tiny_pages.parquet"),
+            ParquetReadOptions::default(),
+        )
+        .await?;
+
+    let results = df
+        .clone()
+        .aggregate(vec![], vec![count(Expr::Wildcard)])?
+        .explain(false, false)
+        .unwrap()
+        .collect()
+        .await?;
+
+    let expected = vec![
+        "+---------------+---------------------------------------------------+",
+        "| plan_type     | plan                                              |",
+        "+---------------+---------------------------------------------------+",
+        "|               |                                                   |",
+        "|               |   EmptyExec: produce_one_row=true                 |",
+        "|               |   TableScan: ?table? projection=[id]              |",
+        "| logical_plan  | Aggregate: groupBy=[[]], aggr=[[COUNT(UInt8(1))]] |",
+        "| physical_plan | ProjectionExec: expr=[7300 as COUNT(UInt8(1))]    |",
+        "+---------------+---------------------------------------------------+",
+    ];
+    assert_batches_sorted_eq!(expected, &results);
+
+    let results = df
+        .clone()
+        .aggregate(vec![], vec![count(Expr::Wildcard)])?
+        .collect()
+        .await?;
+
+    print_batches(&results);
+
+    let expected = vec![
+        "+-----------------+",
+        "| COUNT(UInt8(1)) |",
+        "+-----------------+",
+        "| 7300            |",
+        "+-----------------+",
+    ];
+    assert_batches_sorted_eq!(expected, &results);
+
+    Ok(())
+}
+
 #[tokio::test]
 async fn describe() -> Result<()> {
     let ctx = SessionContext::new();