Update tests 3

apache · Sep 28, 2023 · b283946 · b283946
1 parent f189d9a
commit b283946
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/datafusion/optimizer/src/push_down_filter.rs b/datafusion/optimizer/src/push_down_filter.rs
@@ -1056,7 +1056,6 @@ mod tests {
         // filter is before projection
         let expected = "\
             Projection: test.a AS b, test.c\
-            \n  Filter: test.a = Int64(1)\
             \n  TableScan: test, full_filters=[test.a = Int64(1)]";
         assert_optimized_plan_eq(&plan, expected)
     }
@@ -1281,8 +1280,7 @@ mod tests {
         Filter: SUM(test.c) > Int64(10)\
         \n  Aggregate: groupBy=[[b]], aggr=[[SUM(test.c)]]\
         \n    Projection: test.a AS b, test.c\
-        \n      Filter: test.a > Int64(10)\
-        \n        TableScan: test";
+        \n      TableScan: test, full_filters=[test.a > Int64(10)]";
         assert_optimized_plan_eq(&plan, expected)
     }
 
@@ -2664,7 +2662,7 @@ Projection: a, b
         \n  Projection: test1.a, test1.b\
         \n    TableScan: test1\
         \n  Projection: test2.a, test2.b\
-        \n    TableScan: test2, full_filters=[test2.b > UInt32(2)]  ";
+        \n    TableScan: test2, full_filters=[test2.b > UInt32(2)]";
         assert_optimized_plan_eq(&plan, expected)
     }
 

diff --git a/datafusion/optimizer/src/push_down_projection.rs b/datafusion/optimizer/src/push_down_projection.rs
@@ -981,7 +981,7 @@ mod tests {
         let table_scan = table_scan_with_filters(
             Some("test"),
             &schema,
-            Some(vec![2]),
+            None,
             vec![col("b").eq(lit(1))],
         )?
         .build()?;