46 lines
2.1 KiB
Python
46 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
|
"""范式选择投影的离线测试，不调用向量服务。"""
|
|
|
|
import pathlib
|
|
import sys
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
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 select_patterns as selector # noqa: E402
|
|
|
|
|
|
class SelectPatternsContractTest(unittest.TestCase):
|
|
def test检索结果投影成严格范式来源指针(self):
|
|
cards = [{
|
|
"sourceId": "draft:677", "sourceVersion": "draft-revision:1",
|
|
"name": "借尸试人", "summary": "借外在突袭验证立场",
|
|
"visibleFields": {"写法": "先试探后收手"},
|
|
}]
|
|
with patch.object(selector, "search_cards", return_value=cards), \
|
|
patch.object(selector, "write_section", return_value={"id": 9, "status": "written"}) as write, \
|
|
patch.object(selector, "record_section", return_value={"status": "recorded"}):
|
|
result = selector.select_patterns(12, "深空机甲微恐战斗", top=5)
|
|
ref = result["section"]
|
|
self.assertEqual(result["selected"], 1)
|
|
self.assertEqual(ref["id"], 9)
|
|
write.assert_called_once()
|
|
payload = write.call_args.kwargs["payload"]
|
|
self.assertEqual(payload["patternReferences"][0]["sourceId"], "draft:677")
|
|
self.assertEqual(payload["patternReferences"][0]["sourceType"], "public_pattern")
|
|
|
|
def test无命中记录明确原因(self):
|
|
with patch.object(selector, "search_cards", return_value=[]), \
|
|
patch.object(selector, "write_section", return_value={"id": 10, "status": "written"}) as write, \
|
|
patch.object(selector, "record_section", return_value={"status": "recorded"}):
|
|
selector.select_patterns(12, "无命中意图")
|
|
payload = write.call_args.kwargs["payload"]
|
|
self.assertEqual(payload["patternSelection"]["emptyReason"], "公共范式检索没有命中可授权卡")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|