Skip to content
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 9 additions & 9 deletions tools/scripts/bench.py
Original file line number Diff line number Diff line change
Expand Up @@ -666,15 +666,15 @@ def raw_sqlite_read_heavy() -> None:
def sqlspec_sqlite_initialization() -> None:
with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
spec = SQLSpec()
config = SqliteConfig(database=tmp.name)
config = SqliteConfig(connection_config={"database": tmp.name})
with spec.provide_session(config) as session:
session.execute(CREATE_TEST_TABLE)


def sqlspec_sqlite_write_heavy() -> None:
with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
spec = SQLSpec()
config = SqliteConfig(database=tmp.name)
config = SqliteConfig(connection_config={"database": tmp.name})
with spec.provide_session(config) as session:
session.execute(CREATE_TEST_TABLE)
# Use execute_many for bulk inserts
Expand All @@ -685,7 +685,7 @@ def sqlspec_sqlite_write_heavy() -> None:
def sqlspec_sqlite_read_heavy() -> None:
with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
spec = SQLSpec()
config = SqliteConfig(database=tmp.name)
config = SqliteConfig(connection_config={"database": tmp.name})
with spec.provide_session(config) as session:
session.execute(CREATE_TEST_TABLE)
data: Sequence[tuple[str]] = [(f"value_{i}",) for i in range(ROWS_TO_INSERT)]
Expand Down Expand Up @@ -1369,7 +1369,7 @@ def sqlspec_sqlite_iterative_inserts() -> None:
"""Individual inserts in a loop - shows per-call overhead."""
with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
spec = SQLSpec()
config = SqliteConfig(database=tmp.name)
config = SqliteConfig(connection_config={"database": tmp.name})
with spec.provide_session(config) as session:
session.execute(CREATE_TEST_TABLE)
for i in range(ROWS_TO_INSERT):
Expand Down Expand Up @@ -1417,7 +1417,7 @@ def sqlspec_sqlite_repeated_queries() -> None:
"""Repeated single-row queries - tests query cache effectiveness."""
with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
spec = SQLSpec()
config = SqliteConfig(database=tmp.name)
config = SqliteConfig(connection_config={"database": tmp.name})
with spec.provide_session(config) as session:
session.execute(CREATE_TEST_TABLE)
data: Sequence[tuple[str]] = [(f"value_{i}",) for i in range(ROWS_TO_INSERT)]
Expand Down Expand Up @@ -2476,7 +2476,7 @@ def sqlspec_sqlite_dict_key_transform() -> None:
"""SELECT 10K rows with 10 columns, then transform keys to camelCase (sqlspec)."""
with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
spec = SQLSpec()
config = SqliteConfig(database=tmp.name)
config = SqliteConfig(connection_config={"database": tmp.name})
with spec.provide_session(config) as session:
session.execute(CREATE_WIDE_TABLE)
data: Sequence[tuple[str, ...]] = [_generate_wide_row(i) for i in range(ROWS_TO_INSERT)]
Expand Down Expand Up @@ -2513,7 +2513,7 @@ def sqlspec_sqlite_schema_mapping() -> None:
"""SELECT 10K rows and map to TypedDict using schema_type= parameter."""
with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
spec = SQLSpec()
config = SqliteConfig(database=tmp.name)
config = SqliteConfig(connection_config={"database": tmp.name})
with spec.provide_session(config) as session:
session.execute(CREATE_WIDE_TABLE)
data: Sequence[tuple[str, ...]] = [_generate_wide_row(i) for i in range(ROWS_TO_INSERT)]
Expand Down Expand Up @@ -2567,7 +2567,7 @@ def sqlspec_sqlite_complex_parameters() -> None:
"""INSERT rows with complex parameter types: UUID, datetime, JSON (sqlspec)."""
with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
spec = SQLSpec()
config = SqliteConfig(database=tmp.name)
config = SqliteConfig(connection_config={"database": tmp.name})
with spec.provide_session(config) as session:
session.execute(CREATE_COMPLEX_TABLE)
data: Sequence[tuple[str, str, str]] = [_generate_complex_row(i) for i in range(ROWS_TO_INSERT)]
Expand Down Expand Up @@ -2598,7 +2598,7 @@ def sqlspec_sqlite_thin_path_stress() -> None:
stress_count = ROWS_TO_INSERT * 10
with tempfile.NamedTemporaryFile(suffix=".db") as tmp:
spec = SQLSpec()
config = SqliteConfig(database=tmp.name)
config = SqliteConfig(connection_config={"database": tmp.name})
with spec.provide_session(config) as session:
session.execute(CREATE_TEST_TABLE)
for i in range(stress_count):
Expand Down
Loading