Check Dataset result than just calling collect.

viirya · viirya · commit 5c4e7ffff5bd · 2016-07-25T12:04:46.000+08:00
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
@@ -444,7 +444,7 @@ class DatasetSuite extends QueryTest with SharedSQLContext {
     // After sampling, sampleDF doesn't contain swid=1.
     assert(!sampleDF.select("swid").collect.contains(1))
     // udfOne should not encounter swid=1.
-    sampleDF.select(udfOne($"swid")).collect
+    checkAnswer(sampleDF.select(udfOne($"swid")), List.fill(sampleDF.count.toInt)(Row(1)))
   }
 
   test("SPARK-11436: we should rebind right encoder when join 2 datasets") {

Original file line number	Diff line number	Diff line change
`@@ -444,7 +444,7 @@ class DatasetSuite extends QueryTest with SharedSQLContext {`
`444`	`444`	`// After sampling, sampleDF doesn't contain swid=1.`
`445`	`445`	`assert(!sampleDF.select("swid").collect.contains(1))`
`446`	`446`	`// udfOne should not encounter swid=1.`
`447`		`- sampleDF.select(udfOne($"swid")).collect`
	`447`	`+ checkAnswer(sampleDF.select(udfOne($"swid")), List.fill(sampleDF.count.toInt)(Row(1)))`
`448`	`448`	`}`
`449`	`449`
`450`	`450`	`test("SPARK-11436: we should rebind right encoder when join 2 datasets") {`