add test

Author: aditanase

datafusion/optimizer/src/push_down_filter.rs

@@ -4127,4 +4127,37 @@ mod tests {
         "
         )
     }
+
+    /// Create a test table scan with uppercase column names for case sensitivity testing
+    fn test_table_scan_with_uppercase_columns() -> Result<LogicalPlan> {
+        let schema = Schema::new(vec![
+            Field::new("A", DataType::UInt32, false),
+            Field::new("B", DataType::UInt32, false),
+            Field::new("C", DataType::UInt32, false),
+        ]);
+        table_scan(Some("test"), &schema, None)?.build()
+    }
+
+    #[test]
+    fn filter_agg_case_insensitive() -> Result<()> {
+        // Create a table with uppercase column names to test case sensitivity
+        let table_scan = test_table_scan_with_uppercase_columns()?;
+        let plan = LogicalPlanBuilder::from(table_scan)
+            .aggregate(
+                vec![col(r#""A""#)],
+                vec![sum(col(r#""B""#)).alias("total_salary")],
+            )?
+            // Use a lowercase column name in the filter to test case insensitivity
+            .filter(col(r#""A""#).gt(lit(10i64)))?
+            .build()?;
+
+        // With the fix, the filter should be pushed down through the aggregation
+        assert_optimized_plan_equal!(
+            plan,
+            @r"
+        Aggregate: groupBy=[[test.A]], aggr=[[sum(test.B) AS total_salary]]
+          TableScan: test, full_filters=[test.A > Int64(10)]
+        "
+        )
+    }
 }