File size: 1,854 Bytes
ba54ea9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | """Tests for the inference gateway. These exercise citation parsing and
backend routing without loading any model — the mock backend is enough.
"""
import os
from datetime import datetime
import pytest
import recap.inference.gateway as gw
from recap.models import Event
def _ev(src, date_iso="2022-03-14"):
return Event(
id=src,
date=datetime.fromisoformat(date_iso),
category="lab",
title=f"Record from {src}",
source=src,
)
def test_parses_citations_from_model_output():
text = (
"Creatinine first crossed normal in March 2022 [src:lab_2022.pdf#p1]. "
"eGFR was 52 [src:lab_2022.pdf]."
)
cites = gw._parse_citations(text, [_ev("lab_2022.pdf")])
assert len(cites) == 2
assert cites[0].page == 1
assert cites[1].page is None
def test_dedupes_repeated_citations():
text = "[src:a.pdf] said X [src:a.pdf]"
cites = gw._parse_citations(text, [_ev("a.pdf")])
assert len(cites) == 1
def test_drops_citations_to_unknown_sources():
text = "[src:hallucinated.pdf]"
cites = gw._parse_citations(text, [])
assert cites == []
def test_dedupe_treats_different_pages_as_different_citations():
text = "[src:a.pdf#p1] earlier [src:a.pdf#p2] later"
cites = gw._parse_citations(text, [_ev("a.pdf")])
assert len(cites) == 2
assert {c.page for c in cites} == {1, 2}
def test_answer_end_to_end_with_mock_backend(monkeypatch):
"""Full pipeline: question -> retrieve -> mock -> cited answer."""
monkeypatch.setenv("RECAP_BACKEND", "mock")
events = [
_ev("lab_2022.pdf", "2022-03-14"),
_ev("visit_2023.pdf", "2023-01-01"),
]
a = gw.answer("when did the lab change", events)
assert "[mock answer]" in a.text
assert any(c.source_id in {"lab_2022.pdf", "visit_2023.pdf"} for c in a.citations)
|