2323import org .apache .doris .nereids .trees .expressions .functions .agg .Sum ;
2424import org .apache .doris .nereids .trees .expressions .functions .scalar .GroupingId ;
2525import org .apache .doris .nereids .trees .plans .Plan ;
26+ import org .apache .doris .nereids .trees .plans .algebra .Repeat .RepeatType ;
2627import org .apache .doris .nereids .trees .plans .logical .LogicalOlapScan ;
2728import org .apache .doris .nereids .trees .plans .logical .LogicalRepeat ;
2829import org .apache .doris .nereids .util .PlanConstructor ;
@@ -71,6 +72,7 @@ public void testToShapes() {
7172 Repeat <Plan > repeat = new LogicalRepeat <>(
7273 groupingSets ,
7374 ImmutableList .of (id , name , alias ),
75+ RepeatType .GROUPING_SETS ,
7476 scan
7577 );
7678
@@ -106,6 +108,7 @@ public void testToShapesWithGroupingFunction() {
106108 Repeat <Plan > repeat = new LogicalRepeat <>(
107109 groupingSets ,
108110 ImmutableList .of (name , groupingAlias ),
111+ RepeatType .GROUPING_SETS ,
109112 scan
110113 );
111114
@@ -160,6 +163,7 @@ public void testGetGroupingSetsIndexesInOutput() {
160163 Repeat <Plan > repeat = new LogicalRepeat <>(
161164 groupingSets ,
162165 ImmutableList .of (id , name , gender , groupingId ),
166+ RepeatType .GROUPING_SETS ,
163167 scan
164168 );
165169 List <Slot > outputSlots = ImmutableList .of (id , name , gender , groupingId .toSlot ());
@@ -185,6 +189,7 @@ public void testComputeRepeatSlotIdList() {
185189 Repeat <Plan > repeat = new LogicalRepeat <>(
186190 groupingSets ,
187191 ImmutableList .of (id , name ),
192+ RepeatType .GROUPING_SETS ,
188193 scan
189194 );
190195 List <Slot > outputSlots = ImmutableList .of (id , name );
0 commit comments