moxiaonian
5889d7a150
- 新增 memory/ 命主档案：memory_save/memory_lookup 建档检索， 反馈三态（已核实/已证伪/未核实），档案只存本机不进副本 - SKILL.md description 改写为按问题域与使用场景触发； 输出契约补 R-禁-22 偏离须声明（编号+理由+书证）联动 - R-运-31 事类映射表改开放清单措辞 - 规则库 541→542 条（新增 R-禁-22），库版本 3.5 - 测试全套 69 项通过
120 lines
5.0 KiB
Python
120 lines
5.0 KiB
Python
"""档案（memory/）：save → lookup 往返、同主键更新保留正文小节、无匹配输出。
|
||
|
||
脚本以 --memory-root / ZIPING_MEMORY_ROOT 覆盖档案根目录，测试用 tmp_path 隔离。
|
||
"""
|
||
import os
|
||
import re
|
||
import subprocess
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
SKILL_ROOT = Path(__file__).resolve().parents[1]
|
||
SCRIPTS_DIR = SKILL_ROOT / "scripts"
|
||
SAVE = SCRIPTS_DIR / "memory_save.py"
|
||
LOOKUP = SCRIPTS_DIR / "memory_lookup.py"
|
||
|
||
ZHANGSAN = ["--name", "张三", "--gender", "男",
|
||
"--birth", "1990-05-14 08:30", "--place", "四川省成都市"]
|
||
|
||
|
||
def run(script, mem_root, *args, use_env=False):
|
||
env = dict(os.environ)
|
||
env["PYTHONPATH"] = str(SCRIPTS_DIR) + os.pathsep + env.get("PYTHONPATH", "")
|
||
cmd = [sys.executable, str(script), *args]
|
||
if use_env:
|
||
env["ZIPING_MEMORY_ROOT"] = str(mem_root)
|
||
else:
|
||
cmd += ["--memory-root", str(mem_root)]
|
||
return subprocess.run(cmd, cwd=str(SCRIPTS_DIR), env=env,
|
||
capture_output=True, text=True, timeout=120)
|
||
|
||
|
||
def read_meta(text: str) -> dict:
|
||
"""受限 frontmatter 解析：仅取测试断言所需的标量字段。"""
|
||
lines = text.splitlines()
|
||
assert lines[0] == "---"
|
||
meta = {}
|
||
for ln in lines[1:]:
|
||
if ln == "---":
|
||
break
|
||
k, _, v = ln.partition(":")
|
||
meta[k.strip()] = v.strip().strip('"')
|
||
return meta
|
||
|
||
|
||
class TestSaveLookupRoundtrip:
|
||
def test_save_then_lookup_by_name_and_pillars(self, tmp_path):
|
||
r = run(SAVE, tmp_path, *ZHANGSAN)
|
||
assert r.returncode == 0, f"stderr:\n{r.stderr}"
|
||
path = Path(r.stdout.strip())
|
||
assert path.is_file()
|
||
assert path.parent == tmp_path / "archives"
|
||
text = path.read_text(encoding="utf-8")
|
||
# schema：frontmatter + 固定小节 + 嵌入命盘
|
||
meta = read_meta(text)
|
||
assert meta["name"] == "张三" and meta["gender"] == "男"
|
||
assert meta["birth_clock"] == "1990-05-14 08:30"
|
||
assert meta["birthplace"] == "四川省成都市"
|
||
assert meta["created_at"] and meta["updated_at"]
|
||
for section in ("## 盘面", "## 主要关注方向", "## 归纳",
|
||
"## 流年大事", "## 反馈与澄清"):
|
||
assert section in text
|
||
assert "## 四柱" in text # 盘面嵌入命盘全文
|
||
assert "- 性格：未断" in text
|
||
|
||
# 按姓名检索命中（参数覆盖 memory 根目录）
|
||
r = run(LOOKUP, tmp_path, "--name", "张三")
|
||
assert r.returncode == 0, f"stderr:\n{r.stderr}"
|
||
assert str(path) in r.stdout and "张三" in r.stdout
|
||
|
||
# 按四柱检索命中（环境变量覆盖 memory 根目录）
|
||
pillars = meta["pillars"].strip("[]").replace('"', "").replace(",", "")
|
||
r = run(LOOKUP, tmp_path, "--pillars", pillars, use_env=True)
|
||
assert r.returncode == 0, f"stderr:\n{r.stderr}"
|
||
assert str(path) in r.stdout
|
||
|
||
# 出生信息组合检索，空白容错
|
||
r = run(LOOKUP, tmp_path, "--birth", "1990-05-14 08:30",
|
||
"--place", " 四川省成都市 ", "--gender", "男")
|
||
assert r.returncode == 0 and str(path) in r.stdout
|
||
|
||
|
||
class TestSameKeyUpdate:
|
||
def test_update_preserves_body_sections(self, tmp_path):
|
||
r1 = run(SAVE, tmp_path, *ZHANGSAN)
|
||
assert r1.returncode == 0, f"stderr:\n{r1.stderr}"
|
||
path = Path(r1.stdout.strip())
|
||
|
||
# 模拟解释完成后的回写：改归纳、加反馈、固定 created_at 以便断言
|
||
text = path.read_text(encoding="utf-8")
|
||
text = text.replace("- 性格：未断", "- 性格：沉稳内敛【已核实】")
|
||
text = text.replace("（无）", "- 【已证伪】断 2024 年升职；实际为调岗未升。")
|
||
text = re.sub(r'^created_at:.*$', 'created_at: "2000-01-01 00:00"',
|
||
text, count=1, flags=re.M)
|
||
path.write_text(text, encoding="utf-8")
|
||
|
||
# 同主键再次建档：同一文件、盘面刷新、其余小节保留、created_at 不动
|
||
r2 = run(SAVE, tmp_path, *ZHANGSAN)
|
||
assert r2.returncode == 0, f"stderr:\n{r2.stderr}"
|
||
assert Path(r2.stdout.strip()) == path
|
||
assert list((tmp_path / "archives").glob("*.md")) == [path]
|
||
text = path.read_text(encoding="utf-8")
|
||
assert "- 性格：沉稳内敛【已核实】" in text
|
||
assert "【已证伪】断 2024 年升职；实际为调岗未升。" in text
|
||
assert "## 四柱" in text # 盘面仍为完整命盘
|
||
assert text.count("# 命盘") == 1 # 旧盘面整体替换，无残留重复
|
||
meta = read_meta(text)
|
||
assert meta["created_at"] == "2000-01-01 00:00"
|
||
assert meta["updated_at"] != meta["created_at"]
|
||
|
||
|
||
class TestLookupNoMatch:
|
||
def test_no_match_exit_0(self, tmp_path):
|
||
r = run(LOOKUP, tmp_path, "--name", "不存在的人")
|
||
assert r.returncode == 0, f"stderr:\n{r.stderr}"
|
||
assert "无档案" in r.stdout
|
||
|
||
def test_no_condition_exit_2(self, tmp_path):
|
||
r = run(LOOKUP, tmp_path)
|
||
assert r.returncode == 2
|