初始化提交

backend/tests/modules/extraction/test_extraction_extract.py

@@ -0,0 +1,110 @@
+"""大周天记忆/世界抽取纯函数测试。"""
+
+from app.modules.extraction.extract import (
+    MemoryCandidateDraft,
+    WorldEntityDraft,
+    parse_extraction,
+)
+
+
+def test_parse_standard_object() -> None:
+    raw = (
+        '{"memories":[{"type":"preference","title":"喜欢简洁","content":"用户偏好简洁",'
+        '"importance":0.8,"sensitivity":"low","confidence":0.7,"reason":"多次提及"}],'
+        '"entities":[{"type":"project","name":"项目X","attributes":{"status":"active"},'
+        '"state_type":"inference","confidence":0.6}]}'
+    )
+    memories, entities = parse_extraction(raw)
+    assert len(memories) == 1
+    assert memories[0].title == "喜欢简洁"
+    assert memories[0].type == "preference"
+    assert memories[0].importance == 0.8
+    assert len(entities) == 1
+    assert entities[0].name == "项目X"
+    assert entities[0].type == "project"
+    assert entities[0].attributes == {"status": "active"}
+    assert entities[0].state_type == "inference"
+
+
+def test_parse_markdown_block() -> None:
+    raw = '```json\n{"memories":[],"entities":[]}\n```'
+    memories, entities = parse_extraction(raw)
+    assert memories == []
+    assert entities == []
+
+
+def test_parse_text_mixed() -> None:
+    raw = '抽取结果：\n{"memories":[{"title":"t","content":"c"}],"entities":[]}\n以上'
+    memories, entities = parse_extraction(raw)
+    assert len(memories) == 1
+    assert entities == []
+
+
+def test_parse_invalid_returns_empty() -> None:
+    assert parse_extraction("不是JSON") == ([], [])
+    assert parse_extraction("[]") == ([], [])  # 数组不是对象
+
+
+def test_missing_keys_returns_empty_lists() -> None:
+    memories, entities = parse_extraction('{"memories":[]}')
+    assert memories == []
+    assert entities == []
+
+
+def test_skip_memory_missing_title_or_content() -> None:
+    raw = '{"memories":[{"title":"t"},{"content":"c"},{"title":"t","content":"c"}]}'
+    memories, _ = parse_extraction(raw)
+    assert len(memories) == 1
+
+
+def test_skip_entity_missing_name() -> None:
+    raw = '{"entities":[{"type":"project"},{"name":"有名字"}]}'
+    _, entities = parse_extraction(raw)
+    assert len(entities) == 1
+    assert entities[0].name == "有名字"
+
+
+def test_memory_safe_defaults() -> None:
+    raw = '{"memories":[{"title":"t","content":"c"}]}'
+    m = parse_extraction(raw)[0][0]
+    assert m.sensitivity == "low"
+    assert m.type == "other"
+    assert m.confidence == 0.5
+    assert m.importance == 0.5
+
+
+def test_entity_safe_defaults() -> None:
+    raw = '{"entities":[{"name":"n"}]}'
+    e = parse_extraction(raw)[1][0]
+    assert e.state_type == "inference"  # 非 fact，降权
+    assert e.type == "concept"
+    assert e.confidence == 0.5
+    assert e.attributes == {}
+
+
+def test_clamp_values() -> None:
+    raw = (
+        '{"memories":[{"title":"t","content":"c","importance":1.5,"confidence":-0.2}],'
+        '"entities":[{"name":"n","confidence":2.0}]}'
+    )
+    memories, entities = parse_extraction(raw)
+    assert memories[0].importance == 1.0
+    assert memories[0].confidence == 0.0
+    assert entities[0].confidence == 1.0
+
+
+def test_invalid_sensitivity_defaults() -> None:
+    raw = '{"memories":[{"title":"t","content":"c","sensitivity":"unknown"}]}'
+    assert parse_extraction(raw)[0][0].sensitivity == "low"
+
+
+def test_invalid_state_type_defaults() -> None:
+    raw = '{"entities":[{"name":"n","state_type":"certain"}]}'
+    assert parse_extraction(raw)[1][0].state_type == "inference"
+
+
+def test_draft_types() -> None:
+    raw = '{"memories":[{"title":"t","content":"c"}],"entities":[{"name":"n"}]}'
+    memories, entities = parse_extraction(raw)
+    assert isinstance(memories[0], MemoryCandidateDraft)
+    assert isinstance(entities[0], WorldEntityDraft)