fix failed tests

Author: godfreyhe

flink-table/flink-table-planner-blink/src/test/scala/org/apache/flink/table/plan/metadata/FlinkRelMdUniqueGroupsTest.scala

@@ -313,7 +313,6 @@ class FlinkRelMdUniqueGroupsTest extends FlinkRelMdHandlerTestBase {
   @Test
   def testGetUniqueGroupsOnWindowAgg(): Unit = {
     Array(logicalWindowAgg, flinkLogicalWindowAgg,
-      batchLocalWindowAgg,
       batchGlobalWindowAggWithoutLocalAgg,
       batchGlobalWindowAggWithLocalAgg).foreach { agg =>
       assertEquals(ImmutableBitSet.of(0, 1, 2, 3, 4, 5, 6),
@@ -327,9 +326,10 @@ class FlinkRelMdUniqueGroupsTest extends FlinkRelMdHandlerTestBase {
       assertEquals(ImmutableBitSet.of(0, 1, 2),
         mq.getUniqueGroups(agg, ImmutableBitSet.of(0, 1, 2)))
     }
+    assertEquals(ImmutableBitSet.of(0, 1, 2, 3),
+      mq.getUniqueGroups(batchLocalWindowAgg, ImmutableBitSet.of(0, 1, 2, 3)))
 
     Array(logicalWindowAggWithAuxGroup, flinkLogicalWindowAggWithAuxGroup,
-      batchLocalWindowAggWithAuxGroup,
       batchGlobalWindowAggWithoutLocalAggWithAuxGroup,
       batchGlobalWindowAggWithLocalAggWithAuxGroup).foreach { agg =>
       assertEquals(ImmutableBitSet.of(1),
@@ -340,7 +340,13 @@ class FlinkRelMdUniqueGroupsTest extends FlinkRelMdHandlerTestBase {
         mq.getUniqueGroups(agg, ImmutableBitSet.of(0, 1, 2)))
       assertEquals(ImmutableBitSet.of(0, 1, 2, 3),
         mq.getUniqueGroups(agg, ImmutableBitSet.of(0, 1, 2, 3)))
+      assertEquals(ImmutableBitSet.of(0, 1, 2, 3, 4, 5, 6),
+        mq.getUniqueGroups(agg, ImmutableBitSet.of(0, 1, 2, 3, 4, 5, 6)))
     }
+    assertEquals(ImmutableBitSet.of(0),
+      mq.getUniqueGroups(batchLocalWindowAggWithAuxGroup, ImmutableBitSet.of(0, 1)))
+    assertEquals(ImmutableBitSet.of(0, 1, 2),
+      mq.getUniqueGroups(batchLocalWindowAggWithAuxGroup, ImmutableBitSet.of(0, 1, 2)))
   }
 
   @Test