Skip to content

Commit 523c6f3

Browse files
viiryaclaude
andcommitted
style: fix rustfmt formatting in sum_int and sum_decimal
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent f8a1210 commit 523c6f3

2 files changed

Lines changed: 21 additions & 17 deletions

File tree

native/spark-expr/src/agg_funcs/sum_decimal.rs

Lines changed: 9 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -544,7 +544,10 @@ impl GroupsAccumulator for SumDecimalGroupsAccumulator {
544544
opt_filter: Option<&BooleanArray>,
545545
total_num_groups: usize,
546546
) -> DFResult<()> {
547-
debug_assert!(opt_filter.is_none(), "opt_filter is not supported in merge_batch");
547+
debug_assert!(
548+
opt_filter.is_none(),
549+
"opt_filter is not supported in merge_batch"
550+
);
548551

549552
self.resize_helper(total_num_groups);
550553

@@ -733,18 +736,14 @@ mod tests {
733736

734737
// values: [100, 200, 300, 400], filter: [T, F, T, F] => sum = 100+300 = 400
735738
let values: ArrayRef = Arc::new(
736-
Decimal128Array::from(vec![100i128, 200, 300, 400])
737-
.with_data_type(data_type.clone()),
739+
Decimal128Array::from(vec![100i128, 200, 300, 400]).with_data_type(data_type.clone()),
738740
);
739741
let filter = BooleanArray::from(vec![true, false, true, false]);
740742
acc.update_batch(&[values], &[0, 0, 0, 0], Some(&filter), 1)
741743
.unwrap();
742744

743745
let result = acc.evaluate(EmitTo::All).unwrap();
744-
let result = result
745-
.as_any()
746-
.downcast_ref::<Decimal128Array>()
747-
.unwrap();
746+
let result = result.as_any().downcast_ref::<Decimal128Array>().unwrap();
748747
assert_eq!(result.value(0), 400);
749748
}
750749

@@ -762,18 +761,14 @@ mod tests {
762761
crate::create_query_context_map(),
763762
);
764763

765-
let values: ArrayRef = Arc::new(
766-
Decimal128Array::from(vec![10i128, 20, 30]).with_data_type(data_type.clone()),
767-
);
764+
let values: ArrayRef =
765+
Arc::new(Decimal128Array::from(vec![10i128, 20, 30]).with_data_type(data_type.clone()));
768766
let filter = BooleanArray::from(vec![Some(true), None, Some(true)]);
769767
acc.update_batch(&[values], &[0, 0, 0], Some(&filter), 1)
770768
.unwrap();
771769

772770
let result = acc.evaluate(EmitTo::All).unwrap();
773-
let result = result
774-
.as_any()
775-
.downcast_ref::<Decimal128Array>()
776-
.unwrap();
771+
let result = result.as_any().downcast_ref::<Decimal128Array>().unwrap();
777772
assert_eq!(result.value(0), 40); // 10 + 30 = 40
778773
}
779774
}

native/spark-expr/src/agg_funcs/sum_int.rs

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -543,7 +543,10 @@ impl GroupsAccumulator for SumIntGroupsAccumulatorLegacy {
543543
opt_filter: Option<&BooleanArray>,
544544
total_num_groups: usize,
545545
) -> DFResult<()> {
546-
debug_assert!(opt_filter.is_none(), "opt_filter is not supported in merge_batch");
546+
debug_assert!(
547+
opt_filter.is_none(),
548+
"opt_filter is not supported in merge_batch"
549+
);
547550

548551
if values.len() != 1 {
549552
return Err(DataFusionError::Internal(format!(
@@ -687,7 +690,10 @@ impl GroupsAccumulator for SumIntGroupsAccumulatorAnsi {
687690
opt_filter: Option<&BooleanArray>,
688691
total_num_groups: usize,
689692
) -> DFResult<()> {
690-
debug_assert!(opt_filter.is_none(), "opt_filter is not supported in merge_batch");
693+
debug_assert!(
694+
opt_filter.is_none(),
695+
"opt_filter is not supported in merge_batch"
696+
);
691697

692698
if values.len() != 1 {
693699
return Err(DataFusionError::Internal(format!(
@@ -868,7 +874,10 @@ impl GroupsAccumulator for SumIntGroupsAccumulatorTry {
868874
opt_filter: Option<&BooleanArray>,
869875
total_num_groups: usize,
870876
) -> DFResult<()> {
871-
debug_assert!(opt_filter.is_none(), "opt_filter is not supported in merge_batch");
877+
debug_assert!(
878+
opt_filter.is_none(),
879+
"opt_filter is not supported in merge_batch"
880+
);
872881

873882
if values.len() != 2 {
874883
return Err(DataFusionError::Internal(format!(

0 commit comments

Comments
 (0)