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