hanging bigquery ?

Author: euri10

docs/examples/queries/users.parquet

@@ -1,6 +1,3 @@
-pytest_plugins = ["pytest_databases.docker.postgres", "pytest_databases.docker.mysql",
-                  "pytest_databases.docker.oracle", "pytest_databases.docker.bigquery"]
-
 from collections.abc import Generator
 from typing import Any
 
@@ -12,6 +9,13 @@
 from sqlspec.adapters.bigquery.config import BigQueryConfig
 from sqlspec.adapters.bigquery.driver import BigQueryDriver
 
+pytest_plugins = [
+    "pytest_databases.docker.postgres",
+    "pytest_databases.docker.mysql",
+    "pytest_databases.docker.oracle",
+    "pytest_databases.docker.bigquery",
+]
+
 
 @pytest.fixture
 def table_schema_prefix(bigquery_service: "BigQueryService") -> str:
@@ -39,3 +43,17 @@ def bigquery_session(bigquery_config: BigQueryConfig) -> Generator[BigQueryDrive
     with bigquery_config.provide_session() as session:
         yield session
 
+@pytest.fixture
+async def postgres_session_async(postgres_service: "PostgresService") -> Generator[Any, Any, None]:
+    """Create a Postgres async session."""
+    from sqlspec.adapters.asyncpg import AsyncpgConfig
+
+    config = AsyncpgConfig(
+        pool_config={
+            "dsn": f"postgresql://{postgres_service.user}:{postgres_service.password}@{postgres_service.host}:{postgres_service.port}/{postgres_service.database}",
+            "min_size": 1,
+            "max_size": 5,
+        }
+    )
+    async with config.provide_session() as session:
+        yield session

docs/examples/usage/conftest.py

@@ -1,6 +1,8 @@
 # Test module converted from docs example - code-block 10
 """Minimal smoke test for drivers_and_querying example 10."""
 
+from pathlib import Path
+
 from sqlspec.adapters.duckdb import DuckDBConfig
 
 
@@ -16,9 +18,9 @@ def test_example_10_duckdb_config() -> None:
 
     with spec.provide_session(config) as session:
         # Create table from Parquet
-        session.execute("""
-           CREATE TABLE users AS
-           SELECT * FROM read_parquet('users.parquet')
+        session.execute(f"""
+           CREATE TABLE if not exists users AS
+           SELECT * FROM read_parquet('{Path(__file__).parent.parent / "queries/users.parquet"}')
        """)
 
         # Analytical query

docs/examples/usage/test_drivers_and_querying_10.py

@@ -20,7 +20,7 @@ def test_example_11_oracledb_config(oracle_service: OracleService) -> None:
     )
 
     with spec.provide_session(config) as session:
-        create_table_sql = """CREATE TABLE employees (
+        create_table_sql = """CREATE TABLE if not exists employees (
            employee_id NUMBER PRIMARY KEY,
            first_name VARCHAR2(50),
            last_name VARCHAR2(50)

docs/examples/usage/test_drivers_and_querying_11.py

@@ -1,32 +1,35 @@
 # Test module converted from docs example - code-block 12
 """Minimal smoke test for drivers_and_querying example 12."""
 
-
 from sqlspec.adapters.bigquery import BigQueryDriver
 
 
-def test_example_12_bigquery_config(bigquery_session: BigQueryDriver) -> None:
-
+def test_example_12_bigquery_config(bigquery_session: "BigQueryDriver") -> None:
+    import datetime
 
     from sqlspec import SQLSpec
-    spec = SQLSpec()
 
-    result = bigquery_session.execute("SELECT 1 AS value")
+    SQLSpec()
+
+    bigquery_session.execute("SELECT 1 AS value")
 
     # Create the test table
 
     create_table_query = """
-    CREATE TABLE events (
+    CREATE TABLE if not exists events (
         timestamp TIMESTAMP,
         event_type STRING
     )
     """
     bigquery_session.execute_script(create_table_query)
 
-    result = bigquery_session.execute("""
+    bigquery_session.execute(
+        """
        SELECT DATE(timestamp) as date,
               COUNT(*) as events
        FROM events
        WHERE timestamp >= @start_date
        GROUP BY date
-   """, start_date=datetime.date(2025, 1, 1))
+   """,
+        start_date=datetime.date(2025, 1, 1),
+    )

docs/examples/usage/test_drivers_and_querying_12.py

@@ -16,14 +16,14 @@ async def test_example_5_construct_config(postgres_service: PostgresService) ->
     )
     assert config is not None
     async with spec.provide_session(config) as session:
-        create_table_query = """        CREATE TABLE IF NOT EXISTS users (
+        create_table_query = """CREATE TABLE IF NOT EXISTS users (
             id SERIAL PRIMARY KEY,
             name VARCHAR(100),
             email VARCHAR(100) UNIQUE
         );        """
         await session.execute(create_table_query)
         # Insert with RETURNING
         await session.execute(
-            "INSERT INTO users (name, email) VALUES ($1, $2) RETURNING id", "Alice", "alice@example.com"
+            "INSERT INTO users (name, email) VALUES ($1, $2) ETURNING id", "Alice", "alice@example.com"
         )
         await session.execute("SELECT * FROM users WHERE id = $1", 1)