fix

apache · Jul 20, 2020 · cd93b70 · cd93b70
1 parent 8aee8a5
commit cd93b70
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...catalyst/src/test/scala/org/apache/spark/sql/catalyst/optimizer/EliminateSortsSuite.scala b/...catalyst/src/test/scala/org/apache/spark/sql/catalyst/optimizer/EliminateSortsSuite.scala
@@ -325,9 +325,9 @@ class EliminateSortsSuite extends PlanTest {
     Seq(MaxBy(Symbol("a"), Symbol("b")), MinBy(Symbol("a"), Symbol("b"))).foreach { agg =>
       val projectPlan = testRelation.select(Symbol("a"), Symbol("b"))
       val unnecessaryOrderByPlan = projectPlan.orderBy(Symbol("a").asc)
-      val groupByPlan = unnecessaryOrderByPlan.groupBy(Symbol("a"))(agg)
+      val groupByPlan = unnecessaryOrderByPlan.groupBy(Symbol("a"), Symbol("b"))(agg)
       val optimized = Optimize.execute(groupByPlan.analyze)
-      val correctAnswer = projectPlan.groupBy(Symbol("a"))(agg).analyze
+      val correctAnswer = projectPlan.groupBy(Symbol("a"), Symbol("b"))(agg).analyze
       comparePlans(optimized, correctAnswer)
     }
   }