PDC / backend /tests /test_data_source_audit.py
borndeveloper's picture
Deploy: case-study recommendation engine + ply yarn fix + weighted percentile removal
967454e
Raw
History Blame Contribute Delete
554 Bytes
from app.services.data_service import DataService
def test_data_source_audit_has_expected_counts() -> None:
ds = DataService()
ds.load_data()
audit = ds.get_data_source_audit()
assert "row_counts" in audit
rc = audit["row_counts"]
assert rc["raw_working"] > 9000
assert rc["raw_piece_dyed"] > 1500
assert rc["used_total"] > 10000
assert "column_coverage" in audit
cov = audit["column_coverage"]
assert "Greige EPI" in cov
assert "FINISH EPI" in cov
assert cov["Greige EPI"]["non_null_rows"] > 9000