diff options
author | Wenchen Fan <wenchen@databricks.com> | 2016-04-13 11:41:09 +0800 |
---|---|---|
committer | Wenchen Fan <wenchen@databricks.com> | 2016-04-13 11:41:09 +0800 |
commit | a5f8c9b15b6181f04c1314c638017adb8c88c7df (patch) | |
tree | d338c416f166dc8ab5c02d3a83ccd7eaed3fe9af /sql/core/src | |
parent | 587cd554af24601d332e9ce5c74e98b62d0fd830 (diff) | |
download | spark-a5f8c9b15b6181f04c1314c638017adb8c88c7df.tar.gz spark-a5f8c9b15b6181f04c1314c638017adb8c88c7df.tar.bz2 spark-a5f8c9b15b6181f04c1314c638017adb8c88c7df.zip |
[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.
Diffstat (limited to 'sql/core/src')
-rw-r--r-- | sql/core/src/test/scala/org/apache/spark/sql/DatasetSuite.scala | 2 |
1 files changed, 1 insertions, 1 deletions
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 : _*) } } |