File size: 522 Bytes
e078b1d
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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