muse-agent-example/tests/skills/制定作品规划/test_repair_deterministic_receipt.py

32 lines
1.2 KiB
Python

#!/usr/bin/env python3
"""确定性规划回执修正的离线合同测试。"""
import pathlib
import sys
import unittest
PROJECT_ROOT = pathlib.Path(__file__).resolve().parents[3]
SCRIPT_DIR = PROJECT_ROOT / "muse" / "lifecycle" / "flow" / "skills" / "book" / "制定作品规划" / "scripts"
sys.path.insert(0, str(SCRIPT_DIR))
import repair_deterministic_receipt as repair # noqa: E402
class DeterministicReceiptRepairTest(unittest.TestCase):
def test只修正误写为_true的确定性回执(self):
summary = {"executionKind": "deterministic_persist", "sectionId": 3}
self.assertTrue(repair._is_deterministic(summary, True))
self.assertFalse(repair._is_deterministic(summary, None))
self.assertFalse(repair._is_deterministic({"executionKind": "model"}, True))
def test修正摘要保留原字段并标记证据修正(self):
corrected = repair._corrected_summary({"sectionId": 3})
self.assertEqual(corrected["sectionId"], 3)
self.assertEqual(corrected["evidenceCorrection"], "deterministic-model-match-v1")
self.assertTrue(corrected["previousModelMatch"])
if __name__ == "__main__":
unittest.main()