debug windows

Signed-off-by: Adam Gutglick <[email protected]>

Author: AdamGS

vortex-datafusion/src/persistent/mod.rs

@@ -79,6 +79,10 @@ mod tests {
     #[case(None)]
     #[tokio::test]
     async fn query_file(#[case] limit: Option<usize>) -> anyhow::Result<()> {
+        use datafusion::functions_aggregate::count::count;
+        use datafusion_expr::Expr;
+        use datafusion_expr::utils::COUNT_STAR_EXPANSION;
+
         let session = VortexSession::default();
         let temp_dir = tempdir()?;
         let strings = ChunkedArray::from_iter([
@@ -116,12 +120,13 @@ mod tests {
 
         let ctx = SessionContext::default();
         let format = Arc::new(VortexFormat::new(session));
-        let table_url = ListingTableUrl::parse(
-            temp_dir
-                .path()
-                .to_str()
-                .ok_or_else(|| vortex_err!("Path is not valid UTF-8"))?,
-        )?;
+
+        let table_url = temp_dir
+            .path()
+            .to_str()
+            .ok_or_else(|| vortex_err!("Path is not valid UTF-8"))?;
+        dbg!(table_url);
+        let table_url = ListingTableUrl::parse(table_url)?;
         assert!(table_url.is_collection());
 
         let config = ListingTableConfig::new(table_url)
@@ -134,6 +139,18 @@ mod tests {
         let listing_table = Arc::new(ListingTable::try_new(config)?);
 
         ctx.register_table("vortex_tbl", listing_table as _)?;
+        let count_df = ctx.table("vortex_tbl").await?.aggregate(
+            vec![],
+            vec![count(Expr::Literal(COUNT_STAR_EXPANSION, None))],
+        )?;
+
+        let plan = count_df.create_physical_plan().await?;
+        println!(
+            "{}",
+            DisplayableExecutionPlan::new(plan.as_ref())
+                .set_show_schema(true)
+                .tree_render()
+        );
         let total_row_count = ctx.table("vortex_tbl").await?.count().await?;
         assert_eq!(total_row_count, 8);
 

vortex-datafusion/src/persistent/opener.rs

@@ -335,7 +335,7 @@ impl FileOpener for VortexOpener {
                 })
                 .try_flatten()
                 .map(move |batch| {
-                    if projector.projection().column_indices().is_empty() {
+                    if projector.projection().as_ref().is_empty() {
                         batch
                     } else {
                         batch.and_then(|b| projector.project_batch(&b))