revert the utils changes

apache · Jan 26, 2025 · a2df38d · a2df38d
1 parent 6454540
commit a2df38d
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/paimon-core/src/test/java/org/apache/paimon/mergetree/compact/MergeFunctionTestUtils.java b/paimon-core/src/test/java/org/apache/paimon/mergetree/compact/MergeFunctionTestUtils.java
@@ -81,10 +81,18 @@ public static List<ReusingTestData> getExpectedForAggSum(List<ReusingTestData> i
 
         List<ReusingTestData> expected = new ArrayList<>();
         for (List<ReusingTestData> group : groups.values()) {
-            long sum =
-                    group.stream().mapToLong(d -> d.valueKind.isAdd() ? d.value : -d.value).sum();
-            ReusingTestData last = group.get(group.size() - 1);
-            expected.add(new ReusingTestData(last.key, last.sequenceNumber, RowKind.INSERT, sum));
+            if (group.size() == 1) {
+                // due to ReducerMergeFunctionWrapper
+                expected.add(group.get(0));
+            } else {
+                long sum =
+                        group.stream()
+                                .mapToLong(d -> d.valueKind.isAdd() ? d.value : -d.value)
+                                .sum();
+                ReusingTestData last = group.get(group.size() - 1);
+                expected.add(
+                        new ReusingTestData(last.key, last.sequenceNumber, RowKind.INSERT, sum));
+            }
         }
         return expected;
     }