# -*- coding: utf-8 -*- """结题初稿内核金标准用例（离线、确定性；docx 用 zipfile 断言内容）。""" from __future__ import annotations import json import unittest import zipfile from _support import IsolatedDataRoot def _project(**overrides) -> dict: base = { "projectNo": "NSFC2026003", "projectName": "面向能源转化的材料设计", "piName": "王教授", "department": "化材学院", "program": "国家自然科学基金青年项目", "startDate": "2024-01-01", "endDate": "2026-12-31", "expected": [ {"type": "论文", "target": 3, "unit": "篇"}, {"type": "专利", "target": 1, "unit": "项"}, {"type": "人才培养", "target": 2, "unit": "名"}, ], "annualReportExcerpt": "完成界面改性工艺探索；发表高水平论文两篇。", "status": "待结题", } base.update(overrides) return base def _achievements() -> list[dict]: return [ {"projectNo": "NSFC2026003", "achievementType": "论文", "title": "界面调控研究A", "venueOrNo": "Applied Energy", "achDate": "2025-06-01", "isRepresentative": "1"}, {"projectNo": "NSFC2026003", "achievementType": "论文", "title": "界面调控研究B", "venueOrNo": "J. Mater. Chem. A", "achDate": "2025-11-01", "isRepresentative": "1"}, {"projectNo": "NSFC2026003", "achievementType": "论文", "title": "界面调控研究C", "venueOrNo": "", "achDate": "2026-03-01", "isRepresentative": "0"}, {"projectNo": "NSFC2026003", "achievementType": "论文", "title": "界面调控研究D", "venueOrNo": "", "achDate": "2026-05-01", "isRepresentative": "0"}, {"projectNo": "NSFC2026003", "achievementType": "专利", "title": "一种界面改性方法", "venueOrNo": "ZL2025XXXX", "achDate": "2026-01-01", "isRepresentative": "0"}, {"projectNo": "NSFC2026003", "achievementType": "人才培养", "title": "硕士研究生1名毕业", "venueOrNo": "", "achDate": "2026-06-30", "isRepresentative": "0"}, ] def _expenses() -> list[dict]: return [ {"projectNo": "NSFC2026003", "category": "差旅费", "amount": 23000.0, "occurredDate": "2025-04-10", "description": "赴外地调研"}, {"projectNo": "NSFC2026003", "category": "差旅费", "amount": 10000.0, "occurredDate": "2025-09-01", "description": "学术会议"}, {"projectNo": "NSFC2026003", "category": "设备费", "amount": 200000.0, "occurredDate": "2024-12-01", "description": "测试设备"}, {"projectNo": "NSFC2026003", "category": "劳务费", "amount": 40000.0, "occurredDate": "2026-06-01", "description": "研究生劳务"}, ] class DraftKernelTestBase(unittest.TestCase): def setUp(self) -> None: self._root = IsolatedDataRoot() self._root.__enter__() from db.connection import init_db init_db() def tearDown(self) -> None: self._root.__exit__(None, None, None) def _kernel(self, expected=None, achievements=None, expenses=None): from service.draft_service import build_draft_sections project = _project() if expected is not None: project["expected_json"] = json.dumps(expected, ensure_ascii=False) return build_draft_sections(project, achievements if achievements is not None else _achievements(), expenses if expenses is not None else _expenses()) class TestC1Completion(DraftKernelTestBase): def test_over_and_equal(self) -> None: result = self._kernel() by_type = {row["type"]: row for row in result["completion"]} self.assertEqual(by_type["论文"]["verdict"], "超额完成") # 4 > 3 self.assertEqual(by_type["专利"]["verdict"], "持平") # 1 == 1 def test_below_target(self) -> None: result = self._kernel() by_type = {row["type"]: row for row in result["completion"]} self.assertEqual(by_type["人才培养"]["verdict"], "未达标") # 1 < 2 self.assertIn("预期 2 名，实际 1 名", by_type["人才培养"]["detail"]) def test_missing_type_is_not_met_with_zero(self) -> None: result = self._kernel() by_type = {row["type"]: row for row in result["completion"]} self.assertEqual(by_type["专利"]["actual"], 1) def test_missing_target_is_pending(self) -> None: result = self._kernel(expected=[{"type": "论文", "unit": "篇"}]) self.assertEqual(result["completion"][0]["verdict"], "数据待补") class TestC3Expenses(DraftKernelTestBase): def test_ratio_sums_to_one_and_sentences(self) -> None: result = self._kernel() total_ratio = round(sum(row["ratio"] for row in result["expense_rows"]), 4) self.assertEqual(total_ratio, 1.0) top = result["expense_rows"][0] self.assertEqual(top["category"], "设备费") # 金额最大排前 self.assertIn("购置/使用科研设备", top["sentence"]) def test_empty_expenses_placeholder(self) -> None: result = self._kernel(expenses=[]) self.assertIn("尚未录入", result["expense_paragraph"]) class TestC4AndC5(DraftKernelTestBase): def test_narrative_points_with_source(self) -> None: result = self._kernel() sources = {p["source"] for p in result["narrative_points"]} self.assertIn("成果清单", sources) self.assertIn("年度报告", sources) def test_sections_order_and_placeholders(self) -> None: result = self._kernel() titles = list(result["sections"].keys()) self.assertEqual(titles[:3], ["1. 研究背景与目标", "2. 研究内容与进展", "3. 预期成果完成情况"]) self.assertIn("来源：成果清单", "\n".join(result["sections"]["5. 成果清单"])) self.assertTrue(any(p.startswith("[待补充") for p in result["sections"]["1. 研究背景与目标"])) def test_w1_empty_ledger_does_not_block(self) -> None: result = self._kernel(achievements=[], expenses=[]) self.assertIn("尚未录入", "\n".join(result["sections"]["5. 成果清单"])) self.assertIn("尚未录入", result["expense_paragraph"]) class TestDraftGeneration(unittest.TestCase): def setUp(self) -> None: self._root = IsolatedDataRoot() self._root.__enter__() from db.connection import init_db from service.ledger_service import import_expenses, import_achievements, import_projects, upsert_project init_db() upsert_project(_project()) for ach in _achievements(): from service.ledger_service import upsert_achievement upsert_achievement(ach) for expense in _expenses(): from service.ledger_service import upsert_expense upsert_expense(expense) def tearDown(self) -> None: self._root.__exit__(None, None, None) def _draft(self, project_no="NSFC2026003"): from service.draft_service import query_pending_projects, run_draft from service.ledger_service import list_achievements, list_expenses projects = query_pending_projects(project_no=project_no) assert projects, "project should exist" return run_draft(projects[0], list_achievements(project_no), list_expenses(project_no)) def test_docx_generated_with_sections(self) -> None: result = self._draft() self.assertTrue(result["draft_file"].endswith(".docx")) with zipfile.ZipFile(result["draft_file"]) as zf: xml = zf.read("word/document.xml").decode("utf-8") self.assertIn("结题报告（初稿）", xml) self.assertIn("研究背景与目标", xml) self.assertIn("经费决算说明", xml) self.assertIn("来源：成果清单", xml) self.assertIn("待补充", xml) def test_preview_and_status_update(self) -> None: result = self._draft() self.assertTrue(result["preview_file"].endswith(".html")) from service.ledger_service import list_projects row = next(p for p in list_projects() if p["project_no"] == "NSFC2026003") self.assertEqual(row["status"], "初稿已生成") def test_batch_pending(self) -> None: from service.draft_service import query_pending_projects, run_draft for project in query_pending_projects(): run_draft(project, [], []) from service.ledger_service import list_drafts, stats payload = stats() self.assertEqual(payload["draft_total"], 1) self.assertEqual(len(list_drafts()), 1) def test_project_not_found(self) -> None: from service.draft_service import query_pending_projects self.assertEqual(query_pending_projects(project_no="NOPE0001"), []) if __name__ == "__main__": unittest.main()