fix test

apache · Jul 14, 2023 · f8a01c1 · f8a01c1
1 parent 2da356c
commit f8a01c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datafusion/physical-expr/src/aggregate/groups_accumulator/accumulate.rs b/datafusion/physical-expr/src/aggregate/groups_accumulator/accumulate.rs
@@ -708,7 +708,7 @@ mod test {
             // Validate the final buffer (one value per group)
             let expected_null_buffer = mock.expected_null_buffer(total_num_groups);
 
-            let null_buffer = null_state.build();
+            let null_buffer = null_state.build(EmitTo::All);
 
             assert_eq!(null_buffer, expected_null_buffer);
         }
@@ -825,7 +825,7 @@ mod test {
             // Validate the final buffer (one value per group)
             let expected_null_buffer = mock.expected_null_buffer(total_num_groups);
 
-            let null_buffer = null_state.build();
+            let null_buffer = null_state.build(EmitTo::All);
 
             assert_eq!(null_buffer, expected_null_buffer);
         }