28 lines
1.0 KiB
Python
28 lines
1.0 KiB
Python
from pathlib import Path
|
|
|
|
import yaml
|
|
|
|
from didactopus.multilingual_qa_seed import generate_multilingual_qa_seed, write_multilingual_qa_seed
|
|
|
|
|
|
def test_generate_multilingual_qa_seed_uses_pack_content() -> None:
|
|
payload = generate_multilingual_qa_seed("domain-packs/mit-ocw-information-entropy", languages=["es"])
|
|
assert payload["source_language"] == "en"
|
|
assert payload["review_status"] == "draft-seed"
|
|
assert "es" in payload["targets"]
|
|
target = payload["targets"]["es"]
|
|
assert target["required_terms"]
|
|
assert any(item["id"] == "shannon-entropy" for item in target["required_terms"])
|
|
assert target["required_caveats"]
|
|
|
|
|
|
def test_write_multilingual_qa_seed_writes_yaml(tmp_path: Path) -> None:
|
|
out = write_multilingual_qa_seed(
|
|
"domain-packs/mit-ocw-information-entropy",
|
|
out_path=tmp_path / "multilingual_qa.seed.yaml",
|
|
languages=["es", "fr"],
|
|
)
|
|
assert out.exists()
|
|
written = yaml.safe_load(out.read_text(encoding="utf-8"))
|
|
assert set(written["targets"]) == {"es", "fr"}
|