minor fix
Project: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/commit/8d9f0d4c Tree: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/tree/8d9f0d4c Diff: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/diff/8d9f0d4c Branch: refs/heads/JIRA-22/pr-385 Commit: 8d9f0d4c00758324029d342eb4b892e046ca4a49 Parents: 80be81e Author: amaya <g...@sapphire.in.net> Authored: Thu Sep 29 11:02:14 2016 +0900 Committer: amaya <g...@sapphire.in.net> Committed: Thu Sep 29 11:02:14 2016 +0900 ---------------------------------------------------------------------- .../test/scala/org/apache/spark/sql/hive/HivemallOpsSuite.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-hivemall/blob/8d9f0d4c/spark/spark-2.0/src/test/scala/org/apache/spark/sql/hive/HivemallOpsSuite.scala ---------------------------------------------------------------------- diff --git a/spark/spark-2.0/src/test/scala/org/apache/spark/sql/hive/HivemallOpsSuite.scala b/spark/spark-2.0/src/test/scala/org/apache/spark/sql/hive/HivemallOpsSuite.scala index 7b62b92..fe73a1b 100644 --- a/spark/spark-2.0/src/test/scala/org/apache/spark/sql/hive/HivemallOpsSuite.scala +++ b/spark/spark-2.0/src/test/scala/org/apache/spark/sql/hive/HivemallOpsSuite.scala @@ -743,8 +743,8 @@ final class HivemallOpsWithFeatureSuite extends HivemallFeatureQueryTest { val df0 = Seq((1, Seq(1, 2, 3), Seq(5, 6, 7)), (1, Seq(3, 4, 5), Seq(7, 8, 9))) .toDF("c0", "arg0", "arg1") - df0.groupby($"c0").transpose_and_dot("arg0", "arg1").collect() shouldEqual - Seq(Row(1, Seq(Seq(26.0, 30.0, 34.0), Seq(38.0, 44.0, 50.0), Seq(50.0, 58.0, 66.0)))) + checkAnswer(df0.groupby($"c0").transpose_and_dot("arg0", "arg1"), + Seq(Row(1, Seq(Seq(26.0, 30.0, 34.0), Seq(38.0, 44.0, 50.0), Seq(50.0, 58.0, 66.0))))) } }