style: fix pre-commit errors (license header and formatting)

Author: Subham-KRLX

python/sedona/spark/__init__.py

@@ -74,6 +74,7 @@
 
 from pyspark.sql import DataFrame
 
+
 def to_sedonadb(self, connection=None):
     """
     Converts a SedonaSpark DataFrame to a SedonaDB DataFrame.
@@ -83,11 +84,14 @@ def to_sedonadb(self, connection=None):
     try:
         import sedona.db
     except ImportError:
-        raise ImportError("SedonaDB is not installed. Please install it using `pip install sedona-db`.")
+        raise ImportError(
+            "SedonaDB is not installed. Please install it using `pip install sedona-db`."
+        )
 
     if connection is None:
         connection = sedona.db.connect()
 
     return connection.create_data_frame(dataframe_to_arrow(self))
 
+
 DataFrame.to_sedonadb = to_sedonadb

python/tests/test_to_sedonadb.py

@@ -1,42 +1,61 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
 
 import unittest
 import sys
 from unittest.mock import MagicMock, patch
 from pyspark.sql import SparkSession, DataFrame
 from sedona.spark import *
 
+
 class TestToSedonaDB(unittest.TestCase):
 
     def setUp(self):
         # Mock sedona.db to avoid ImportError
         self.mock_sedona_db = MagicMock()
         sys.modules["sedona.db"] = self.mock_sedona_db
         import sedona
+
         sedona.db = self.mock_sedona_db
         self.spark = SparkSession.builder.getOrCreate()
 
     def tearDown(self):
         if "sedona.db" in sys.modules:
             del sys.modules["sedona.db"]
         import sedona
+
         if hasattr(sedona, "db"):
             del sedona.db
 
-    @patch('sedona.spark.dataframe_to_arrow')
+    @patch("sedona.spark.dataframe_to_arrow")
     def test_to_sedonadb_no_connection(self, mock_dataframe_to_arrow):
         # Mock dependencies
         mock_arrow_table = MagicMock()
         mock_dataframe_to_arrow.return_value = mock_arrow_table
-        
+
         mock_connection = MagicMock()
         self.mock_sedona_db.connect.return_value = mock_connection
-        
+
         mock_sedonadb_df = MagicMock()
         mock_connection.create_data_frame.return_value = mock_sedonadb_df
 
         # Create a dummy Spark DataFrame
         df = self.spark.range(1)
-        
+
         # Call the method
         result = df.to_sedonadb()
 
@@ -46,19 +65,19 @@ def test_to_sedonadb_no_connection(self, mock_dataframe_to_arrow):
         mock_connection.create_data_frame.assert_called_once_with(mock_arrow_table)
         self.assertEqual(result, mock_sedonadb_df)
 
-    @patch('sedona.spark.dataframe_to_arrow')
+    @patch("sedona.spark.dataframe_to_arrow")
     def test_to_sedonadb_with_connection(self, mock_dataframe_to_arrow):
         # Mock dependencies
         mock_arrow_table = MagicMock()
         mock_dataframe_to_arrow.return_value = mock_arrow_table
-        
+
         mock_connection = MagicMock()
         mock_sedonadb_df = MagicMock()
         mock_connection.create_data_frame.return_value = mock_sedonadb_df
 
         # Create a dummy Spark DataFrame
         df = self.spark.range(1)
-        
+
         # Call the method
         result = df.to_sedonadb(connection=mock_connection)
 
@@ -67,5 +86,6 @@ def test_to_sedonadb_with_connection(self, mock_dataframe_to_arrow):
         mock_connection.create_data_frame.assert_called_once_with(mock_arrow_table)
         self.assertEqual(result, mock_sedonadb_df)
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     unittest.main()