Formatting

Author: tim-u

google/cloud/dataproc_spark_connect/session.py

@@ -596,7 +596,7 @@ def getOrCreate(self) -> "DataprocSparkSession":
                     raise DataprocSparkConnectException(
                         f"Error while creating Dataproc Session: project ID is not set"
                     )
-                
+
                 if self._region is None:
                     raise DataprocSparkConnectException(
                         f"Error while creating Dataproc Session: location is not set"

tests/integration/test_session.py

@@ -791,19 +791,3 @@ def test_create_local_spark_session(batch_workload_env, local_spark_session):
         assert dataproc_spark_session == local_spark_session
     finally:
         dataproc_spark_session.stop()
-
-def test_create_session_without_project_id(test_region):
-    """Tests that an exception is raised when project ID is not provided."""
-    with pytest.raises(ValueError, match="Project ID must be specified"):
-        (
-            DataprocSparkSession.builder.location(test_region)
-            .getOrCreate()
-        )
-
-def test_create_session_without_location(test_project):
-    """Tests that an exception is raised when location is not provided."""
-    with pytest.raises(ValueError, match="Location must be specified"):
-        (
-            DataprocSparkSession.builder.projectId(test_project)
-            .getOrCreate()
-        )

tests/unit/test_session.py

@@ -1479,7 +1479,6 @@ def test_create_session_without_project_id(self):
         except DataprocSparkConnectException as e:
             self.assertIn("project ID is not set", str(e))
 
-
     def test_create_session_without_location(self):
         """Tests that an exception is raised when location is not provided."""
         os.environ.clear()
@@ -1488,6 +1487,7 @@ def test_create_session_without_location(self):
         except DataprocSparkConnectException as e:
             self.assertIn("location is not set", str(e))
 
+
 class DataprocSparkConnectClientTest(unittest.TestCase):
 
     def setUp(self):