From a5f8c9b15b6181f04c1314c638017adb8c88c7df Mon Sep 17 00:00:00 2001
From: Wenchen Fan <wenchen@databricks.com>
Date: Wed, 13 Apr 2016 11:41:09 +0800
Subject: [PATCH] [SPARK-14554][SQL][FOLLOW-UP] use checkDataset to check the
result
## What changes were proposed in this pull request?
address this comment: https://github.com/apache/spark/pull/12322#discussion_r59417359
## How was this patch tested?
N/A
Author: Wenchen Fan <wenchen@databricks.com>
Closes #12346 from cloud-fan/tmp.
---
sql/core/src/test/scala/org/apache/spark/sql/DatasetSuite.scala | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DatasetSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DatasetSuite.scala
index 47251681e3..d074535bf6 100644
--- a/sql/core/src/test/scala/org/apache/spark/sql/DatasetSuite.scala
+++ b/sql/core/src/test/scala/org/apache/spark/sql/DatasetSuite.scala
@@ -624,7 +624,7 @@ class DatasetSuite extends QueryTest with SharedSQLContext {
test("SPARK-14554: Dataset.map may generate wrong java code for wide table") {
val wideDF = sqlContext.range(10).select(Seq.tabulate(1000) {i => ('id + i).as(s"c$i")} : _*)
// Make sure the generated code for this plan can compile and execute.
- wideDF.map(_.getLong(0)).collect()
+ checkDataset(wideDF.map(_.getLong(0)), 0L until 10 : _*)
}
}
--
GitLab