Spaces:
Sleeping
Sleeping
File size: 1,018 Bytes
fee2f22 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | """Database tests for backend initialization."""
from __future__ import annotations
from sqlalchemy import inspect
from backend.database import Base, engine, init_db
def test_database_connection() -> None:
"""Engine can produce a SQLAlchemy connection object."""
with engine.connect() as connection:
assert connection is not None
def test_model_creation() -> None:
"""ORM metadata contains required tables."""
init_db()
inspector = inspect(engine)
tables = set(inspector.get_table_names())
assert "datasets" in tables
assert "cleaning_jobs" in tables
assert "audit_logs" in tables
assert "model_versions" in tables
def test_crud_operations() -> None:
"""At minimum ensure table metadata is bound and creatable."""
# A lightweight sanity check that table definitions are valid.
assert Base.metadata.tables["datasets"] is not None
assert Base.metadata.tables["cleaning_jobs"] is not None
assert Base.metadata.tables["audit_logs"] is not None
|