| """Gemini structured-output validation (no network).""" |
|
|
| from svarasetu.compose.polish import _validate_structured |
| from svarasetu.shield.outbound import unsupported_claims |
|
|
|
|
| def test_validate_structured_happy_path(): |
| parsed = { |
| "answer": "The heart has four chambers.", |
| "language": "en-IN", |
| "grounded": True, |
| "confidence": 0.8, |
| "refused": False, |
| "citations": [ |
| {"chunk_id": "c1", "claim": "four chambers", "supported": True}, |
| {"chunk_id": "drop-me", "claim": "ignored", "supported": True}, |
| ], |
| } |
| out = _validate_structured(parsed, "en", ["c1", "c2"]) |
| assert out is not None |
| assert out["language"] == "en" |
| assert out["grounded"] is True |
| assert out["citations"] == [ |
| {"chunk_id": "c1", "claim": "four chambers", "supported": True} |
| ] |
|
|
|
|
| def test_validate_structured_refuse_on_insufficient(): |
| parsed = { |
| "answer": "I don't have enough grounded information to answer that.", |
| "language": "hi", |
| "grounded": True, |
| "confidence": 0.1, |
| "refused": False, |
| "citations": [], |
| } |
| out = _validate_structured(parsed, "hi", []) |
| assert out["refused"] is True |
| assert out["grounded"] is False |
|
|
|
|
| def test_validate_structured_rejects_empty(): |
| assert _validate_structured({"answer": " "}, "en", []) is None |
| assert _validate_structured({}, "en", []) is None |
|
|
|
|
| def test_unsupported_claims_flags_alien_sentence(): |
| bad = unsupported_claims( |
| "The heart has four chambers. Alien spacecraft landed on Mars in 1845.", |
| [{"text": "The human heart has four muscular chambers that pump blood."}], |
| min_overlap=0.25, |
| ) |
| assert any("Alien" in s for s in bad) |
|
|
