Ruff format

Author: evertlammerts

scripts/generate_import_cache_cpp.py

@@ -1,5 +1,5 @@
-import os
 import json
+import os
 
 script_dir = os.path.dirname(__file__)
 

scripts/generate_import_cache_json.py

@@ -1,5 +1,5 @@
-import os
 import json
+import os
 from typing import Union
 
 script_dir = os.path.dirname(__file__)

tests/fast/adbc/test_adbc.py

@@ -133,11 +133,14 @@ def test_commit(tmp_path):
             cur.adbc_ingest("ingest", table, "create")
 
     # This now works because we enabled autocommit
-    with adbc_driver_manager.connect(
-        driver=driver_path,
-        entrypoint="duckdb_adbc_init",
-        db_kwargs=db_kwargs,
-    ) as conn, conn.cursor() as cur:
+    with (
+        adbc_driver_manager.connect(
+            driver=driver_path,
+            entrypoint="duckdb_adbc_init",
+            db_kwargs=db_kwargs,
+        ) as conn,
+        conn.cursor() as cur,
+    ):
         cur.execute("SELECT count(*) from ingest")
         assert cur.fetch_arrow_table().to_pydict() == {"count_star()": [4]}
 
@@ -302,12 +305,15 @@ def test_large_chunk(tmp_path):
     if os.path.exists(db):
         os.remove(db)
     db_kwargs = {"path": f"{db}"}
-    with adbc_driver_manager.connect(
-        driver=driver_path,
-        entrypoint="duckdb_adbc_init",
-        db_kwargs=db_kwargs,
-        autocommit=True,
-    ) as conn, conn.cursor() as cur:
+    with (
+        adbc_driver_manager.connect(
+            driver=driver_path,
+            entrypoint="duckdb_adbc_init",
+            db_kwargs=db_kwargs,
+            autocommit=True,
+        ) as conn,
+        conn.cursor() as cur,
+    ):
         cur.adbc_ingest("ingest", table, "create")
         cur.execute("SELECT count(*) from ingest")
         assert cur.fetch_arrow_table().to_pydict() == {"count_star()": [30_000]}
@@ -325,12 +331,15 @@ def test_dictionary_data(tmp_path):
     if os.path.exists(db):
         os.remove(db)
     db_kwargs = {"path": f"{db}"}
-    with adbc_driver_manager.connect(
-        driver=driver_path,
-        entrypoint="duckdb_adbc_init",
-        db_kwargs=db_kwargs,
-        autocommit=True,
-    ) as conn, conn.cursor() as cur:
+    with (
+        adbc_driver_manager.connect(
+            driver=driver_path,
+            entrypoint="duckdb_adbc_init",
+            db_kwargs=db_kwargs,
+            autocommit=True,
+        ) as conn,
+        conn.cursor() as cur,
+    ):
         cur.adbc_ingest("ingest", table, "create")
         cur.execute("from ingest")
         assert cur.fetch_arrow_table().to_pydict() == {
@@ -350,12 +359,15 @@ def test_ree_data(tmp_path):
     if os.path.exists(db):
         os.remove(db)
     db_kwargs = {"path": f"{db}"}
-    with adbc_driver_manager.connect(
-        driver=driver_path,
-        entrypoint="duckdb_adbc_init",
-        db_kwargs=db_kwargs,
-        autocommit=True,
-    ) as conn, conn.cursor() as cur:
+    with (
+        adbc_driver_manager.connect(
+            driver=driver_path,
+            entrypoint="duckdb_adbc_init",
+            db_kwargs=db_kwargs,
+            autocommit=True,
+        ) as conn,
+        conn.cursor() as cur,
+    ):
         cur.adbc_ingest("ingest", table, "create")
         cur.execute("from ingest")
         assert cur.fetch_arrow_table().to_pydict() == {

tests/fast/api/test_native_tz.py

@@ -2,9 +2,9 @@
 import os
 
 import pytest
+from packaging.version import Version
 
 import duckdb
-from packaging.version import Version
 
 pd = pytest.importorskip("pandas")
 pa = pytest.importorskip("pyarrow")

tests/fast/arrow/test_14344.py

@@ -1,6 +1,7 @@
-import pytest
 import hashlib
 
+import pytest
+
 pa = pytest.importorskip("pyarrow")
 
 

tests/fast/arrow/test_9443.py

@@ -1,11 +1,12 @@
-import pytest
-
 from datetime import time
 from pathlib import PurePosixPath
 
+import pytest
+
 pq = pytest.importorskip("pyarrow.parquet")
 pa = pytest.importorskip("pyarrow")
 
+
 class Test9443:
     def test_9443(self, tmp_path, duckdb_cursor):
         arrow_table = pa.Table.from_pylist(

tests/fast/arrow/test_buffer_size_option.py

@@ -1,11 +1,11 @@
 import pytest
 
 import duckdb
-
 from duckdb.typing import VARCHAR
 
 pa = pytest.importorskip("pyarrow")
 
+
 class TestArrowBufferSize:
     def test_arrow_buffer_size(self):
         con = duckdb.connect()

tests/fast/arrow/test_dictionary_arrow.py

@@ -1,6 +1,7 @@
-import pytest
 import datetime
 
+import pytest
+
 pa = pytest.importorskip("pyarrow")
 pq = pytest.importorskip("pyarrow.parquet")
 ds = pytest.importorskip("pyarrow.dataset")

tests/fast/pandas/test_copy_on_write.py

@@ -1,4 +1,5 @@
 import datetime
+
 import pytest
 
 import duckdb

tests/fast/pandas/test_fetch_nested.py

@@ -1,5 +1,6 @@
-import pytest
 import numpy as np
+import pytest
+
 import duckdb
 
 pd = pytest.importorskip("pandas")