cleanup

Signed-off-by: jayzhan211 <[email protected]>
jayzhan211 · Oct 15, 2024 · 8be731a · 8be731a
1 parent e07006d
commit 8be731a
Show file tree

Hide file tree

Showing 6 changed files with 56 additions and 361 deletions.
diff --git a/datafusion/core/tests/fuzz_cases/aggregate_fuzz.rs b/datafusion/core/tests/fuzz_cases/aggregate_fuzz.rs
@@ -142,21 +142,20 @@ async fn test_basic_prim_aggr_group_by_single_string() {
     let data_gen_config = DatasetGeneratorConfig {
         columns,
         rows_num_range: (512, 1024),
-        // rows_num_range: (4, 12),
         sort_keys_set,
     };
 
     // Build fuzzer
     let fuzzer = builder
         .data_gen_config(data_gen_config)
-        .data_gen_rounds(1)
-        // .add_sql("SELECT b, sum(a) FROM fuzz_table GROUP BY b")
+        .data_gen_rounds(16)
+        .add_sql("SELECT b, sum(a) FROM fuzz_table GROUP BY b")
         .add_sql("SELECT b, sum(distinct a) FROM fuzz_table GROUP BY b")
-        // .add_sql("SELECT b, max(a) FROM fuzz_table GROUP BY b")
-        // .add_sql("SELECT b, min(a) FROM fuzz_table GROUP BY b")
-        // .add_sql("SELECT b, count(a) FROM fuzz_table GROUP BY b")
-        // .add_sql("SELECT b, count(distinct a) FROM fuzz_table GROUP BY b")
-        // .add_sql("SELECT b, avg(a) FROM fuzz_table GROUP BY b")
+        .add_sql("SELECT b, max(a) FROM fuzz_table GROUP BY b")
+        .add_sql("SELECT b, min(a) FROM fuzz_table GROUP BY b")
+        .add_sql("SELECT b, count(a) FROM fuzz_table GROUP BY b")
+        .add_sql("SELECT b, count(distinct a) FROM fuzz_table GROUP BY b")
+        .add_sql("SELECT b, avg(a) FROM fuzz_table GROUP BY b")
         .table_name("fuzz_table")
         .build();