Skip to content

Commit 8b2ac55

Browse files
fix: Apply black formatting to resolve CI code-check failure
Co-Authored-By: [email protected] <[email protected]>
1 parent ef14caf commit 8b2ac55

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

src/firebolt_db/firebolt_dialect.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def create_connect_args(self, url: URL) -> Tuple[List, Dict]:
169169

170170
def _validate_core_connection(self, url: URL, parameters: Dict[str, str]) -> None:
171171
"""Validate that Core connection parameters are correct.
172-
172+
173173
Only validates credentials since FireboltCore auth handles other parameters.
174174
"""
175175
if url.username or url.password:
@@ -185,7 +185,7 @@ def _build_connection_kwargs(
185185
self, url: URL, parameters: Dict[str, str], auth: Auth, is_core_connection: bool
186186
) -> Dict[str, Union[str, Auth, Dict[str, Any], None]]:
187187
"""Build connection kwargs for the SDK.
188-
188+
189189
SQLAlchemy URL mapping:
190190
- url.host -> database (Firebolt database name)
191191
- url.database -> engine_name (Firebolt engine name)

tests/unit/test_firebolt_dialect.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -352,13 +352,15 @@ def test_create_connect_args_core_with_credentials_error(
352352
):
353353
dialect.create_connect_args(u)
354354

355-
def test_create_connect_args_core_with_engine_allowed(self, dialect: FireboltDialect):
355+
def test_create_connect_args_core_with_engine_allowed(
356+
self, dialect: FireboltDialect
357+
):
356358
"""Test that Core connections now allow engine_name parameter."""
357359
connection_url = (
358360
"test_engine://test_db_name/test_engine?url=http://localhost:8080"
359361
)
360362
u = url.make_url(connection_url)
361-
363+
362364
result_list, result_dict = dialect.create_connect_args(u)
363365
assert result_dict["engine_name"] == "test_engine"
364366
assert result_dict["url"] == "http://localhost:8080"
@@ -371,7 +373,7 @@ def test_create_connect_args_core_with_account_allowed(
371373
"test_engine://test_db_name?url=http://localhost:8080&account_name=test"
372374
)
373375
u = url.make_url(connection_url)
374-
376+
375377
result_list, result_dict = dialect.create_connect_args(u)
376378
assert result_dict["account_name"] == "test"
377379
assert result_dict["url"] == "http://localhost:8080"

0 commit comments

Comments
 (0)