diff options
author | 云峤 <chensong.cs@alibaba-inc.com> | 2015-05-05 13:23:53 -0700 |
---|---|---|
committer | Reynold Xin <rxin@databricks.com> | 2015-05-05 13:23:53 -0700 |
commit | 735bc3d042b1e3e12de57b66f166af14254ad314 (patch) | |
tree | 853b8c985baf71c78ab51ddc7724dfad630523c6 /python/pyspark/sql/tests.py | |
parent | 489700c809a7c0a836538f3d0bd58bed609e8768 (diff) | |
download | spark-735bc3d042b1e3e12de57b66f166af14254ad314.tar.gz spark-735bc3d042b1e3e12de57b66f166af14254ad314.tar.bz2 spark-735bc3d042b1e3e12de57b66f166af14254ad314.zip |
[SPARK-7294][SQL] ADD BETWEEN
Author: 云峤 <chensong.cs@alibaba-inc.com>
Author: kaka1992 <kaka_1992@163.com>
Closes #5839 from kaka1992/master and squashes the following commits:
b15360d [kaka1992] Fix python unit test in sql/test. =_= I forget to commit this file last time.
f928816 [kaka1992] Fix python style in sql/test.
d2e7f72 [kaka1992] Fix python style in sql/test.
c54d904 [kaka1992] Fix empty map bug.
7e64d1e [云峤] Update
7b9b858 [云峤] undo
f080f8d [云峤] update pep8
76f0c51 [云峤] Merge remote-tracking branch 'remotes/upstream/master'
7d62368 [云峤] [SPARK-7294] ADD BETWEEN
baf839b [云峤] [SPARK-7294] ADD BETWEEN
d11d5b9 [云峤] [SPARK-7294] ADD BETWEEN
Diffstat (limited to 'python/pyspark/sql/tests.py')
-rw-r--r-- | python/pyspark/sql/tests.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/python/pyspark/sql/tests.py b/python/pyspark/sql/tests.py index 7ea6656d31..46c4c88e98 100644 --- a/python/pyspark/sql/tests.py +++ b/python/pyspark/sql/tests.py @@ -453,6 +453,14 @@ class SQLTests(ReusedPySparkTestCase): for row in rndn: assert row[1] >= -4.0 and row[1] <= 4.0, "got: %s" % row[1] + def test_between_function(self): + df = self.sc.parallelize([ + Row(a=1, b=2, c=3), + Row(a=2, b=1, c=3), + Row(a=4, b=1, c=4)]).toDF() + self.assertEqual([Row(a=2, b=1, c=3), Row(a=4, b=1, c=4)], + df.filter(df.a.between(df.b, df.c)).collect()) + def test_save_and_load(self): df = self.df tmpPath = tempfile.mkdtemp() |