diff options
author | Reynold Xin <rxin@databricks.com> | 2015-02-03 22:15:35 -0800 |
---|---|---|
committer | Reynold Xin <rxin@databricks.com> | 2015-02-03 22:15:35 -0800 |
commit | 40c4cb2fe79ceac0d656be7b72cb2ee8d7db7258 (patch) | |
tree | 581df79783cd7b5bef1c51b0156f85265689dae2 /sql/core/src/test | |
parent | eb15631854f8e8e7bcd1b3331f18d6ee81dd6d5c (diff) | |
download | spark-40c4cb2fe79ceac0d656be7b72cb2ee8d7db7258.tar.gz spark-40c4cb2fe79ceac0d656be7b72cb2ee8d7db7258.tar.bz2 spark-40c4cb2fe79ceac0d656be7b72cb2ee8d7db7258.zip |
[SPARK-5579][SQL][DataFrame] Support for project/filter using SQL expressions
```scala
df.selectExpr("abs(colA)", "colB")
df.filter("age > 21")
```
Author: Reynold Xin <rxin@databricks.com>
Closes #4348 from rxin/SPARK-5579 and squashes the following commits:
2baeef2 [Reynold Xin] Fix Python.
b416372 [Reynold Xin] [SPARK-5579][SQL][DataFrame] Support for project/filter using SQL expressions.
Diffstat (limited to 'sql/core/src/test')
-rw-r--r-- | sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala | 12 |
1 files changed, 12 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 19d4f34e56..e588555ad0 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 @@ -47,6 +47,18 @@ class DataFrameSuite extends QueryTest { testData.collect().toSeq) } + test("selectExpr") { + checkAnswer( + testData.selectExpr("abs(key)", "value"), + testData.collect().map(row => Row(math.abs(row.getInt(0)), row.getString(1))).toSeq) + } + + test("filterExpr") { + checkAnswer( + testData.filter("key > 90"), + testData.collect().filter(_.getInt(0) > 90).toSeq) + } + test("repartition") { checkAnswer( testData.select('key).repartition(10).select('key), |