From 756e89c887455966b0afb2d79eee29fdd5c51a48 Mon Sep 17 00:00:00 2001 From: Hyukjin Kwon Date: Fri, 14 Oct 2022 13:38:58 +0900 Subject: [PATCH] Make the tests pass with ANSI enabled --- .../catalyst/optimizer/RewriteDistinctAggregatesSuite.scala | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/optimizer/RewriteDistinctAggregatesSuite.scala b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/optimizer/RewriteDistinctAggregatesSuite.scala index cb4771dd92f80..ac136dfb898ef 100644 --- a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/optimizer/RewriteDistinctAggregatesSuite.scala +++ b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/optimizer/RewriteDistinctAggregatesSuite.scala @@ -28,6 +28,7 @@ class RewriteDistinctAggregatesSuite extends PlanTest { val nullInt = Literal(null, IntegerType) val nullString = Literal(null, StringType) val testRelation = LocalRelation($"a".string, $"b".string, $"c".string, $"d".string, $"e".int) + val testRelation2 = LocalRelation($"a".double, $"b".int, $"c".int, $"d".int, $"e".int) private def checkRewrite(rewrite: LogicalPlan): Unit = rewrite match { case Aggregate(_, _, Aggregate(_, _, _: Expand)) => @@ -77,7 +78,7 @@ class RewriteDistinctAggregatesSuite extends PlanTest { } test("SPARK-40382: eliminate multiple distinct groups due to superficial differences") { - val input = testRelation + val input = testRelation2 .groupBy($"a")( countDistinct($"b" + $"c").as("agg1"), countDistinct($"c" + $"b").as("agg2"), @@ -92,7 +93,7 @@ class RewriteDistinctAggregatesSuite extends PlanTest { } test("SPARK-40382: reduce multiple distinct groups due to superficial differences") { - val input = testRelation + val input = testRelation2 .groupBy($"a")( countDistinct($"b" + $"c" + $"d").as("agg1"), countDistinct($"d" + $"c" + $"b").as("agg2"),