63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
from io import BytesIO
|
||
|
||
from app.schemas.risk_prediction import AiSolutionReport, ReportCard, RiskAssessmentResult
|
||
from app.services.reports import (
|
||
build_risk_report_docx,
|
||
build_risk_report_pdf,
|
||
effective_report_cards,
|
||
report_card_lines,
|
||
)
|
||
from docx import Document
|
||
|
||
|
||
def _result() -> RiskAssessmentResult:
|
||
return RiskAssessmentResult(
|
||
subject_id="洗发水新品",
|
||
score=72,
|
||
level="medium",
|
||
model_version="test",
|
||
factors=[],
|
||
report_cards=[
|
||
ReportCard(
|
||
type="supplier",
|
||
title="供应商深度画像",
|
||
name="佳供应商",
|
||
party_type="供应商",
|
||
supplier_score=88.5,
|
||
frequency_coefficient=1.2,
|
||
top_problems=[{"name": "泄漏", "count": 3}],
|
||
suggestions=["安排首单全检"],
|
||
)
|
||
],
|
||
)
|
||
|
||
|
||
def test_report_card_lines_are_canonical() -> None:
|
||
lines = dict(report_card_lines(_result().report_cards[0]))
|
||
assert lines["分析对象"] == "佳供应商"
|
||
assert lines["供应商得分"] == "88.50 分"
|
||
assert lines["高频问题"] == "泄漏（3次）"
|
||
assert lines["建议方案"] == "安排首单全检"
|
||
|
||
|
||
def test_old_ai_solution_payload_is_supported() -> None:
|
||
result = _result().model_copy(
|
||
update={"report_cards": [], "ai_solution_cards": [AiSolutionReport(title="旧方案")]}
|
||
)
|
||
cards = effective_report_cards(result)
|
||
assert len(cards) == 1
|
||
assert cards[0].title == "旧方案"
|
||
|
||
|
||
def test_pdf_and_word_include_appended_card_content() -> None:
|
||
result = _result()
|
||
pdf = build_risk_report_pdf(result)
|
||
word = build_risk_report_docx(result)
|
||
assert pdf.startswith(b"%PDF")
|
||
document = Document(BytesIO(word))
|
||
text = "\n".join(paragraph.text for paragraph in document.paragraphs)
|
||
text += "\n" + "\n".join(cell.text for table in document.tables for row in table.rows for cell in row.cells)
|
||
assert "佳供应商" in text
|
||
assert "88.50 分" in text
|
||
assert "安排首单全检" in text
|