agentic-search / tests /unit /test_aggregate.py
minhhien0811's picture
Publish code and Fair Protocol 500 results
ec1ecb9 verified
Raw
History Blame Contribute Delete
530 Bytes
from agentic_search.evaluation.aggregate import latest_records
def test_latest_records_replaces_a_failed_attempt_with_a_retry() -> None:
records = [
{"example_id": "q1", "system": "sac", "status": "failed"},
{"example_id": "q1", "system": "sac", "status": "completed"},
{"example_id": "q2", "system": "fixed", "status": "completed"},
]
latest = latest_records(records)
assert len(latest) == 2
assert next(row for row in latest if row["example_id"] == "q1")["status"] == "completed"