From 7e2d8d9cb9428b4f7274efbdc3ba5f7f461b335a Mon Sep 17 00:00:00 2001 From: Wenchen Fan Date: Sat, 15 Aug 2015 23:41:09 +0800 Subject: [PATCH 1/2] add test --- .../src/test/scala/org/apache/spark/sql/DataFrameSuite.scala | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala index cf22797752b97..205cb8ada473c 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala @@ -873,4 +873,9 @@ class DataFrameSuite extends QueryTest with SharedSQLContext { """{"a": {"b": 1}}""" :: Nil)) checkAnswer(df.orderBy("a.b"), Row(Row(1))) } + + test("SPARK-9950: correctly analyze grouping/aggregating on struct fields") { + val df = Seq(("x", (1,1)), ("y", (2, 2))).toDF("a", "b") + checkAnswer(df.groupBy("b._1").agg(sum("b._2")), Row(1, 1) :: Row(2, 2) :: Nil) + } } From 0bbfb1c55a198f6466a973f501be89203e09ac6c Mon Sep 17 00:00:00 2001 From: Wenchen Fan Date: Sun, 16 Aug 2015 11:18:42 +0800 Subject: [PATCH 2/2] fix style... --- .../src/test/scala/org/apache/spark/sql/DataFrameSuite.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala index 205cb8ada473c..1e2aaae52c9e8 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala @@ -875,7 +875,7 @@ class DataFrameSuite extends QueryTest with SharedSQLContext { } test("SPARK-9950: correctly analyze grouping/aggregating on struct fields") { - val df = Seq(("x", (1,1)), ("y", (2, 2))).toDF("a", "b") + val df = Seq(("x", (1, 1)), ("y", (2, 2))).toDF("a", "b") checkAnswer(df.groupBy("b._1").agg(sum("b._2")), Row(1, 1) :: Row(2, 2) :: Nil) } }