55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
import pytest
|
|
from types import SimpleNamespace
|
|
from pathlib import Path
|
|
|
|
from docx import Document
|
|
|
|
from genesis.data_models import StructuredSource
|
|
from genesis.parsers.word_template_parser import WordTemplateParser
|
|
from genesis.qa.qa_loop import QALoop
|
|
|
|
|
|
class FakeEngine:
|
|
def chat_structured(self, *, session_id, prompt, variables, schema, retry_count=2):
|
|
cid = variables.get("title", "x")
|
|
return SimpleNamespace(
|
|
data={
|
|
"title": cid,
|
|
"blocks": [
|
|
{"type": "heading", "level": 2, "text": f"{cid} 小节"},
|
|
{"type": "paragraph", "text": "由 FakeLLM 生成的充分说明内容，满足写入规则要求。"},
|
|
{"type": "table", "caption": "示例表", "rows": [["列1", "列2"], ["值1", "值2"]]},
|
|
],
|
|
},
|
|
status="ok",
|
|
)
|
|
|
|
|
|
def _all_text(doc: Document) -> str:
|
|
parts = [p.text for p in doc.paragraphs]
|
|
for tbl in doc.tables:
|
|
for row in tbl.rows:
|
|
for cell in row.cells:
|
|
parts.append(cell.text)
|
|
return "\n".join(parts)
|
|
|
|
|
|
def test_phase5_e2e_fake_llm(tmp_path):
|
|
template = Path("samples/phase5-slice/template.docx")
|
|
if not template.exists():
|
|
pytest.skip("样本模板缺失，跳过 e2e")
|
|
parsed = WordTemplateParser().parse(str(template))
|
|
ss = StructuredSource(
|
|
template=parsed, tables=[], rule_docs=[], image_analyses=[], existing_system=None, comments=[]
|
|
)
|
|
out = tmp_path / "e2e_out.docx"
|
|
loop = QALoop(max_rounds=2)
|
|
report = loop.run(
|
|
ss, str(out), samples_dir="nonexistent_dir_xyz", engine=FakeEngine(), template_path=str(template)
|
|
)
|
|
assert report.passed is True
|
|
loaded = Document(str(out))
|
|
text = _all_text(loaded)
|
|
assert "由 FakeLLM 生成的充分说明内容" in text
|
|
assert "示例表" in text
|