formats code

dlt-hub · Sep 2, 2024 · 21d54b0 · 21d54b0
1 parent 336def8
commit 21d54b0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tests/sources/sql_database/test_arrow_helpers.py b/tests/sources/sql_database/test_arrow_helpers.py
@@ -84,6 +84,8 @@ def test_row_tuples_to_arrow_unknown_types(all_unknown: bool) -> None:
 
 
 pytest.importorskip("sqlalchemy", minversion="2.0")
+
+
 def test_row_tuples_to_arrow_detects_range_type() -> None:
     from sqlalchemy.dialects.postgresql import Range
 
@@ -95,7 +97,9 @@ def test_row_tuples_to_arrow_detects_range_type() -> None:
         (IntRange(2, 20),),
         (IntRange(3, 30),),
     ]
-    result = row_tuples_to_arrow(rows=rows, columns={"range_col": {"name": "range_col", "nullable": False}}, tz="UTC")
+    result = row_tuples_to_arrow(
+        rows=rows, columns={"range_col": {"name": "range_col", "nullable": False}}, tz="UTC"
+    )
     assert result.num_columns == 1
     assert pa.types.is_struct(result[0].type)
 

diff --git a/tests/sources/sql_database/test_sql_database_source.py b/tests/sources/sql_database/test_sql_database_source.py
@@ -49,6 +49,7 @@ def reset_os_environ():
     os.environ.clear()
     os.environ.update(original_environ)
 
+
 def make_pipeline(destination_name: str) -> dlt.Pipeline:
     return dlt.pipeline(
         pipeline_name="sql_database",