64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
"""章后抽取的机械绑定测试，不连接数据库、不调用模型。"""
|
|
import pathlib
|
|
import sys
|
|
|
|
PROJECT_ROOT = pathlib.Path(__file__).resolve().parents[3]
|
|
SCRIPT_DIR = PROJECT_ROOT / "muse" / "content" / "entity" / "skills" / "extract" / "抽取章节知识" / "scripts"
|
|
sys.path.insert(0, str(SCRIPT_DIR))
|
|
from extract_knowledge import ( # noqa: E402
|
|
ExtractionContractError,
|
|
normalize_extraction,
|
|
salvage_extraction,
|
|
)
|
|
|
|
BODY = "林深走进舰桥，何岚把黑色钥匙交给他。"
|
|
|
|
|
|
def test_normalize_binds_evidence_and_aliases():
|
|
result = normalize_extraction({
|
|
"entities": [{"type": "人物", "name": "林深", "brief": "机师", "fields": {}, "evidence": "林深走进舰桥"}],
|
|
"relations": [],
|
|
"state": {"handoff": "继续调查"},
|
|
}, BODY)
|
|
assert result["entities"][0]["type"] == "character"
|
|
assert result["entities"][0]["evidence"] in BODY
|
|
|
|
|
|
def test_normalize_rejects_fabricated_evidence():
|
|
try:
|
|
normalize_extraction({
|
|
"entities": [{"type": "item", "name": "黑钥匙", "brief": "钥匙", "fields": {}, "evidence": "正文没有这句"}],
|
|
"relations": [], "state": {},
|
|
}, BODY)
|
|
except ExtractionContractError as exc:
|
|
assert "证据不在正文" in str(exc)
|
|
else:
|
|
raise AssertionError("编造证据必须失败关闭")
|
|
|
|
|
|
def test_salvage_drops_unbound_relation_and_keeps_exact_name_evidence():
|
|
result = salvage_extraction({
|
|
"entities": [
|
|
{"type": "item", "name": "黑色钥匙", "brief": "钥匙", "fields": {}, "evidence": "模型编造"},
|
|
{"type": "item", "name": "不存在的物件", "brief": "?", "fields": {}, "evidence": "模型编造"},
|
|
],
|
|
"relations": [{"source": "林深", "target": "何岚", "type": "伙伴", "description": "?", "evidence": "模型编造"}],
|
|
"state": {},
|
|
}, "林深走进舰桥，何岚把黑色钥匙交给他。")
|
|
assert [item["name"] for item in result["entities"]] == ["黑色钥匙"]
|
|
assert result["entities"][0]["evidence"] == "黑色钥匙"
|
|
assert result["relations"] == []
|
|
assert result["mechanicalDrops"] == {"entities": 1, "relations": 1}
|
|
|
|
|
|
def main():
|
|
test_normalize_binds_evidence_and_aliases()
|
|
test_normalize_rejects_fabricated_evidence()
|
|
test_salvage_drops_unbound_relation_and_keeps_exact_name_evidence()
|
|
print("全部通过")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|