Fix empty map bug.

nemccarthy · May 5, 2015 · c54d904 · c54d904
1 parent 7e64d1e
commit c54d904
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/python/pyspark/sql/dataframe.py b/python/pyspark/sql/dataframe.py
@@ -1334,9 +1334,6 @@ def cast(self, dataType):
     def between(self, lowerBound, upperBound):
         """ A boolean expression that is evaluated to true if the value of this
         expression is between the given columns.
-
-        >>> df[df.col1.between(lowerBound, upperBound)].collect()
-        [Row(col1=5, col2=6, col3=8)]
         """
         return (self >= lowerBound) & (self <= upperBound)
 

diff --git a/python/pyspark/sql/tests.py b/python/pyspark/sql/tests.py
@@ -439,9 +439,10 @@ def test_rand_functions(self):
             assert row[1] >= -4.0 and row[1] <= 4.0, "got: %s" % row[1]
 
     def test_between_function(self):
-        df = self.sqlCtx.parallelize([Row(a=1, b=2, c=3),
-                                      Row(a=2, b=1, c=3),
-                                      Row(a=4, b=1, c=4)]).toDF()
+        df = self.sqlCtx.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([False, True, True],
                      df.select(df.a.between(df.b, df.c)).collect())