diff options
author | Reynold Xin <rxin@databricks.com> | 2015-02-16 20:42:57 -0800 |
---|---|---|
committer | Reynold Xin <rxin@databricks.com> | 2015-02-16 20:42:57 -0800 |
commit | d380f324c6d38ffacfda83a525a1a7e23347e5b8 (patch) | |
tree | e74bd8ad25966fc86b69438937e1c654d29b2789 /sql/core | |
parent | a51d51ffac00931c80ce93889a98c2f77aef8953 (diff) | |
download | spark-d380f324c6d38ffacfda83a525a1a7e23347e5b8.tar.gz spark-d380f324c6d38ffacfda83a525a1a7e23347e5b8.tar.bz2 spark-d380f324c6d38ffacfda83a525a1a7e23347e5b8.zip |
[SPARK-5853][SQL] Schema support in Row.
Author: Reynold Xin <rxin@databricks.com>
Closes #4640 from rxin/SPARK-5853 and squashes the following commits:
9c6f569 [Reynold Xin] [SPARK-5853][SQL] Schema support in Row.
Diffstat (limited to 'sql/core')
-rw-r--r-- | sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala | 6 |
1 files changed, 6 insertions, 0 deletions
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 524571d9cc..0da619def1 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 @@ -89,6 +89,12 @@ class DataFrameSuite extends QueryTest { testData.collect().toSeq) } + test("head and take") { + assert(testData.take(2) === testData.collect().take(2)) + assert(testData.head(2) === testData.collect().take(2)) + assert(testData.head(2).head.schema === testData.schema) + } + test("self join") { val df1 = testData.select(testData("key")).as('df1) val df2 = testData.select(testData("key")).as('df2) |