Kjell Tore Guttormsen
37547fe292
The context sets, the packaged knowledge bases and the example bundles are replaced by one fictitious example set about IT operations in an invented organisation: three context sets (serverrom-2027, driftsavtale-2027 and the two-base drift-og-avtale-2027), two synthetic knowledge bases under src/portfolio_optimiser/data/kunnskapsbaser and two example bundles under src/portfolio_optimiser/data/bundles. Numbers, codes and structural values in tests and fixtures are kept; names, ids and wording change. Dated measurement documents that only recorded runs on the replaced material are deleted. Gate figures measured on the new set are not comparable with earlier ones. The exclusion gate from the previous commit is green: 0 tracked files hit outside the shared/ subtree. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
"""Step 6 tests — first-class Pydantic provenance stamp.
|
|
|
|
At least one citation is mandatory (the load-bearing NFR); a valid stamp exposes the
|
|
proposal's provenance fields; the annotation adapter yields MAF citation dicts for display.
|
|
Pattern: tests/spikes/test_c_validator.py (pydantic raises + field assertions).
|
|
"""
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from portfolio_optimiser.provenance import Citation, ProvenanceStamp
|
|
from portfolio_optimiser.retrieval import TextSpan
|
|
|
|
|
|
def _stamp() -> ProvenanceStamp:
|
|
return ProvenanceStamp(
|
|
citations=[Citation(file="licence.txt", locator=TextSpan(0, 10), snippet="0123456789")],
|
|
model="qwen3:4b",
|
|
role="proposer",
|
|
validator_decision="validated",
|
|
token_usage=42,
|
|
cost_baseline_anchored=True,
|
|
bundle_id_source=None,
|
|
code_forms={},
|
|
)
|
|
|
|
|
|
def test_zero_citations_is_rejected() -> None:
|
|
with pytest.raises(ValidationError):
|
|
ProvenanceStamp(
|
|
citations=[],
|
|
model="m",
|
|
role="proposer",
|
|
validator_decision="validated",
|
|
token_usage=1,
|
|
cost_baseline_anchored=True,
|
|
bundle_id_source=None,
|
|
code_forms={},
|
|
)
|
|
|
|
|
|
def test_valid_stamp_exposes_provenance_fields() -> None:
|
|
stamp = _stamp()
|
|
assert len(stamp.citations) >= 1
|
|
assert stamp.model and stamp.role
|
|
assert stamp.validator_decision in {"validated", "rejected"}
|
|
assert isinstance(stamp.token_usage, int) and stamp.token_usage > 0
|
|
|
|
|
|
def test_to_annotations_yields_citation_dicts() -> None:
|
|
anns = _stamp().to_annotations()
|
|
assert anns and all(a["type"] == "citation" for a in anns)
|
|
assert anns[0]["annotated_regions"][0]["type"] == "text_span"
|