Fix python style

Signed-off-by: Karen Feng <karen.feng@databricks.com>

python/pyspark/sql/functions.py

@@ -1594,7 +1594,7 @@ def xxhash64(*cols):
 
 
 @since(3.1)
-def assert_true(col, errMsg = None):
+def assert_true(col, errMsg=None):
     """
     Returns null if the input column is true; throws an exception with the provided error message
     otherwise.
@@ -1618,9 +1618,9 @@ def assert_true(col, errMsg = None):
         )
 
     errMsg = (
-            _create_column_from_literal(errMsg)
-            if isinstance(errMsg, str)
-            else _to_java_column(errMsg)
+        _create_column_from_literal(errMsg)
+        if isinstance(errMsg, str)
+        else _to_java_column(errMsg)
     )
     return Column(sc._jvm.functions.assert_true(_to_java_column(col), errMsg))
 
@@ -1637,9 +1637,9 @@ def raise_error(errMsg):
 
     sc = SparkContext._active_spark_context
     errMsg = (
-            _create_column_from_literal(errMsg)
-            if isinstance(errMsg, str)
-            else _to_java_column(errMsg)
+        _create_column_from_literal(errMsg)
+        if isinstance(errMsg, str)
+        else _to_java_column(errMsg)
     )
     return Column(sc._jvm.functions.raise_error(errMsg))
 

python/pyspark/sql/tests/test_functions.py

@@ -547,7 +547,10 @@ def test_assert_true(self):
 
         with self.assertRaises(TypeError) as cm:
             df.select(assert_true(df.id < 2, 5))
-        self.assertEquals("errMsg should be a Column or a str, got <class 'int'>", str(cm.exception))
+        self.assertEquals(
+            "errMsg should be a Column or a str, got <class 'int'>",
+            str(cm.exception)
+        )
 
     def test_raise_error(self):
         from pyspark.sql.functions import raise_error
@@ -566,7 +569,10 @@ def test_raise_error(self):
 
         with self.assertRaises(TypeError) as cm:
             df.select(raise_error(None))
-        self.assertEquals("errMsg should be a Column or a str, got <class 'NoneType'>", str(cm.exception))
+        self.assertEquals(
+            "errMsg should be a Column or a str, got <class 'NoneType'>",
+            str(cm.exception)
+        )
 
 
 if __name__ == "__main__":