Inserted blank lines for PEP8 compliance

apache · Mar 31, 2015 · ea66f3d · ea66f3d
1 parent d81b02b
commit ea66f3d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/pyspark/sql/tests.py b/python/pyspark/sql/tests.py
@@ -120,10 +120,12 @@ def test_udf_with_callable(self):
         d = [Row(number=i, squared=i**2) for i in range(10)]
         rdd = self.sc.parallelize(d)
         data = self.sqlCtx.createDataFrame(rdd)
+
         class PlusFour:
             def __call__(self, col):
                 if col is not None:
                     return col + 4
+
         call = PlusFour()
         pudf = UserDefinedFunction(call, LongType())
         res = data.select(pudf(data['number']).alias('plus_four'))
@@ -133,9 +135,11 @@ def test_udf_with_partial_function(self):
         d = [Row(number=i, squared=i**2) for i in range(10)]
         rdd = self.sc.parallelize(d)
         data = self.sqlCtx.createDataFrame(rdd)
+
         def some_func(col, param):
             if col is not None:
                 return col + param
+
         pfunc = functools.partial(some_func, param=4)
         pudf = UserDefinedFunction(pfunc, LongType())
         res = data.select(pudf(data['number']).alias('plus_four'))