CUD-Traffic-AI / tests /test_eval.py
Rajeev Ranjan Pandey
Refine UI and fix model selection bugs
e078b1d
raw
history blame contribute delete
522 Bytes
import pandas as pd
from src.evaluate.eval import evaluate_outputs
def test_evaluate_outputs_adds_metrics():
df = pd.DataFrame([{
"Description": "Crash blocked one lane on I-95 northbound.",
"reference_summary": "One lane was blocked after a crash on I-95 northbound.",
"generated_summary": "A crash blocked one lane on I-95 northbound.",
"model_name": "lead1",
}])
out = evaluate_outputs(df)
assert "compression_ratio" in out.columns
assert "rouge1_f1" in out.columns