Adding filtered cases for other tests

edmondop · Jun 4, 2024 · bc330b9 · bc330b9
1 parent 1f6e525
commit bc330b9
Showing 1 changed file with 63 additions and 2 deletions.
diff --git a/datafusion/core/tests/fuzz_cases/join_fuzz.rs b/datafusion/core/tests/fuzz_cases/join_fuzz.rs
@@ -107,6 +107,18 @@ async fn test_left_join_1k() {
     .await
 }
 
+#[tokio::test]
+async fn test_left_join_1k_filtered() {
+    JoinFuzzTestCase::new(
+        make_staggered_batches(1000),
+        make_staggered_batches(1000),
+        JoinType::Left,
+        Some(Box::new(less_than_10_join_filter)),
+    )
+    .run_test()
+    .await
+}
+
 #[tokio::test]
 async fn test_right_join_1k() {
     JoinFuzzTestCase::new(
@@ -118,6 +130,17 @@ async fn test_right_join_1k() {
     .run_test()
     .await
 }
+#[tokio::test]
+async fn test_right_join_1k_filtered() {
+    JoinFuzzTestCase::new(
+        make_staggered_batches(1000),
+        make_staggered_batches(1000),
+        JoinType::Right,
+        Some(Box::new(less_than_10_join_filter)),
+    )
+    .run_test()
+    .await
+}
 
 #[tokio::test]
 async fn test_full_join_1k() {
@@ -131,6 +154,18 @@ async fn test_full_join_1k() {
     .await
 }
 
+#[tokio::test]
+async fn test_full_join_1k_filtered() {
+    JoinFuzzTestCase::new(
+        make_staggered_batches(1000),
+        make_staggered_batches(1000),
+        JoinType::Full,
+        Some(Box::new(less_than_10_join_filter)),
+    )
+    .run_test()
+    .await
+}
+
 #[tokio::test]
 async fn test_semi_join_1k() {
     JoinFuzzTestCase::new(
@@ -143,6 +178,18 @@ async fn test_semi_join_1k() {
     .await
 }
 
+#[tokio::test]
+async fn test_semi_join_1k_filtered() {
+    JoinFuzzTestCase::new(
+        make_staggered_batches(1000),
+        make_staggered_batches(1000),
+        JoinType::LeftSemi,
+        Some(Box::new(less_than_10_join_filter)),
+    )
+    .run_test()
+    .await
+}
+
 #[tokio::test]
 async fn test_anti_join_1k() {
     JoinFuzzTestCase::new(
@@ -155,20 +202,34 @@ async fn test_anti_join_1k() {
     .await
 }
 
+#[tokio::test]
+async fn test_anti_join_1k_filtered() {
+    JoinFuzzTestCase::new(
+        make_staggered_batches(1000),
+        make_staggered_batches(1000),
+        JoinType::LeftAnti,
+        Some(Box::new(less_than_10_join_filter)),
+    )
+    .run_test()
+    .await
+}
+
+type JoinFilterBuilder = Box<dyn Fn(Arc<Schema>, Arc<Schema>) -> JoinFilter>;
+
 struct JoinFuzzTestCase {
     batch_sizes: &'static [usize],
     input1: Vec<RecordBatch>,
     input2: Vec<RecordBatch>,
     join_type: JoinType,
-    join_filter_builder: Option<Box<dyn Fn(Arc<Schema>, Arc<Schema>) -> JoinFilter>>,
+    join_filter_builder: Option<JoinFilterBuilder>,
 }
 
 impl JoinFuzzTestCase {
     fn new(
         input1: Vec<RecordBatch>,
         input2: Vec<RecordBatch>,
         join_type: JoinType,
-        join_filter_builder: Option<Box<dyn Fn(Arc<Schema>, Arc<Schema>) -> JoinFilter>>,
+        join_filter_builder: Option<JoinFilterBuilder>,
     ) -> Self {
         Self {
             batch_sizes: &[1, 2, 7, 49, 50, 51, 100],