dota2tuned / tests /test_storage.py
b3hr0uz's picture
feat: add Dota 2 hackathon scaffold
0caf2ba
Raw
History Blame Contribute Delete
530 Bytes
from pathlib import Path
from dota2tuned.storage import connect_duckdb, refresh_views, write_parquet
def test_refresh_views_creates_duckdb_views(tmp_path: Path):
parquet_dir = tmp_path / "parquet"
duckdb_path = tmp_path / "db.duckdb"
write_parquet(parquet_dir / "example.parquet", [{"id": 1, "name": "ok"}])
refresh_views(duckdb_path, parquet_dir)
conn = connect_duckdb(duckdb_path)
try:
assert conn.execute("select count(*) from example").fetchone()[0] == 1
finally:
conn.close()