From e4f04654618b1644678c18c4331418fddb2982e8 Mon Sep 17 00:00:00 2001 From: Chaerim Yeo Date: Wed, 10 Apr 2019 23:50:41 +0900 Subject: [PATCH] [SPARK-27425] Modify unit test --- .../apache/spark/sql/DataFrameAggregateSuite.scala | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameAggregateSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameAggregateSuite.scala index 0d8254ca9775d..310ee4a0afa0a 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameAggregateSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameAggregateSuite.scala @@ -903,12 +903,20 @@ class DataFrameAggregateSuite extends QueryTest with SharedSQLContext { checkAnswer( testData.agg(count_if('key % 2 === 0), count_if('key > 50), count_if('key < 50)), - Row(50, 50, 49)) + Row(50L, 50L, 49L)) checkAnswer( sql("SELECT COUNT_IF(key % 2 = 0), COUNT_IF(key > 50), COUNT_IF(key < 50) FROM testData"), - Row(50, 50, 49)) + Row(50L, 50L, 49L)) + + checkAnswer( + testData2.groupBy('a).agg(count_if('b % 2 === 0)), + Seq(Row(1, 1L), Row(2, 1L), Row(3, 1L))) + checkAnswer( + sql("SELECT a, COUNT_IF(b % 2 = 0) FROM testData2 GROUP BY a"), + Seq(Row(1, 1L), Row(2, 1L), Row(3, 1L))) checkError(testData.agg(count_if('key))) + checkError(testData.agg(count_if("key"))) checkError(sql("SELECT COUNT_IF(key) FROM testData")) } }