tests: update filters in join fuzz tests

korowa · Sep 6, 2024 · b6951bd · b6951bd
1 parent 23bdb1a
commit b6951bd
Showing 1 changed file with 4 additions and 20 deletions.
diff --git a/datafusion/core/tests/fuzz_cases/join_fuzz.rs b/datafusion/core/tests/fuzz_cases/join_fuzz.rs
@@ -67,22 +67,6 @@ async fn test_inner_join_1k() {
     .await
 }
 
-fn less_than_100_join_filter(schema1: Arc<Schema>, _schema2: Arc<Schema>) -> JoinFilter {
-    let less_than_100 = Arc::new(BinaryExpr::new(
-        Arc::new(Column::new("a", 0)),
-        Operator::Lt,
-        Arc::new(Literal::new(ScalarValue::from(100))),
-    )) as _;
-    let column_indices = vec![ColumnIndex {
-        index: 0,
-        side: JoinSide::Left,
-    }];
-    let intermediate_schema =
-        Schema::new(vec![schema1.field_with_name("a").unwrap().to_owned()]);
-
-    JoinFilter::new(less_than_100, column_indices, intermediate_schema)
-}
-
 fn col_lt_col_filter(schema1: Arc<Schema>, schema2: Arc<Schema>) -> JoinFilter {
     let less_filter = Arc::new(BinaryExpr::new(
         Arc::new(Column::new("x", 1)),
@@ -121,7 +105,7 @@ async fn test_inner_join_1k_filtered() {
         make_staggered_batches(1000),
         make_staggered_batches(1000),
         JoinType::Inner,
-        Some(Box::new(less_than_100_join_filter)),
+        Some(Box::new(col_lt_col_filter)),
     )
     .run_test(&[JoinTestType::HjSmj, JoinTestType::NljHj], false)
     .await
@@ -157,7 +141,7 @@ async fn test_left_join_1k_filtered() {
         make_staggered_batches(1000),
         make_staggered_batches(1000),
         JoinType::Left,
-        Some(Box::new(less_than_100_join_filter)),
+        Some(Box::new(col_lt_col_filter)),
     )
     .run_test(&[JoinTestType::HjSmj, JoinTestType::NljHj], false)
     .await
@@ -182,7 +166,7 @@ async fn test_right_join_1k_filtered() {
         make_staggered_batches(1000),
         make_staggered_batches(1000),
         JoinType::Right,
-        Some(Box::new(less_than_100_join_filter)),
+        Some(Box::new(col_lt_col_filter)),
     )
     .run_test(&[JoinTestType::HjSmj, JoinTestType::NljHj], false)
     .await
@@ -206,7 +190,7 @@ async fn test_full_join_1k_filtered() {
         make_staggered_batches(1000),
         make_staggered_batches(1000),
         JoinType::Full,
-        Some(Box::new(less_than_100_join_filter)),
+        Some(Box::new(col_lt_col_filter)),
     )
     .run_test(&[JoinTestType::HjSmj, JoinTestType::NljHj], false)
     .await