初始化提交

backend/tests/modules/drive/test_drive_extract.py

@@ -0,0 +1,95 @@
+"""大周天内驱力抽取纯函数测试。"""
+
+from app.modules.drive.extract import (
+    CuriosityDraft,
+    SubgoalDraft,
+    parse_curiosities,
+)
+
+
+def test_parse_standard_with_subgoals() -> None:
+    raw = (
+        '[{"question":"用户为何频繁提到项目X","source":"memory_gap",'
+        '"expected_value":0.7,"urgency":0.5,"risk_level":"L2",'
+        '"related_entities":["项目X"],'
+        '"subgoals":[{"title":"询问项目X目标","description":"主动询问用户",'
+        '"expected_information_gain":0.6,"interruption_cost":0.3,"risk_level":"L2"}]}]'
+    )
+    drafts = parse_curiosities(raw)
+    assert len(drafts) == 1
+    c = drafts[0]
+    assert c.question == "用户为何频繁提到项目X"
+    assert c.source == "memory_gap"
+    assert c.expected_value == 0.7
+    assert c.risk_level == "L2"
+    assert c.related_entities == ["项目X"]
+    assert len(c.subgoals) == 1
+    sg = c.subgoals[0]
+    assert sg.title == "询问项目X目标"
+    assert sg.expected_information_gain == 0.6
+    assert sg.interruption_cost == 0.3
+
+
+def test_parse_markdown_block() -> None:
+    raw = '```json\n[{"question":"q"}]\n```'
+    drafts = parse_curiosities(raw)
+    assert len(drafts) == 1
+    assert drafts[0].question == "q"
+
+
+def test_parse_invalid_returns_empty() -> None:
+    assert parse_curiosities("不是JSON") == []
+    assert parse_curiosities("[]") == []
+
+
+def test_skip_missing_question() -> None:
+    raw = '[{"source":"memory_gap"},{"question":"q"}]'
+    drafts = parse_curiosities(raw)
+    assert len(drafts) == 1
+    assert drafts[0].question == "q"
+
+
+def test_defaults_use_safe_pessimistic_values() -> None:
+    """缺省取最严格默认：risk_level=L4、source=reflection。"""
+    raw = '[{"question":"q"}]'
+    c = parse_curiosities(raw)[0]
+    assert c.source == "reflection"
+    assert c.risk_level == "L4"
+    assert c.expected_value == 0.5
+    assert c.urgency == 0.5
+    assert c.related_entities == []
+    assert c.subgoals == []
+
+
+def test_clamp_values() -> None:
+    raw = '[{"question":"q","expected_value":1.5,"urgency":-0.2}]'
+    c = parse_curiosities(raw)[0]
+    assert c.expected_value == 1.0
+    assert c.urgency == 0.0
+
+
+def test_invalid_source_defaults() -> None:
+    raw = '[{"question":"q","source":"unknown_source"}]'
+    assert parse_curiosities(raw)[0].source == "reflection"
+
+
+def test_invalid_risk_level_defaults() -> None:
+    raw = '[{"question":"q","risk_level":"L9"}]'
+    assert parse_curiosities(raw)[0].risk_level == "L4"
+
+
+def test_subgoal_defaults_and_types() -> None:
+    raw = '[{"question":"q","subgoals":[{"title":"t","description":"d"}]}]'
+    c = parse_curiosities(raw)[0]
+    assert isinstance(c, CuriosityDraft)
+    sg = c.subgoals[0]
+    assert isinstance(sg, SubgoalDraft)
+    assert sg.risk_level == "L4"  # 安全默认
+    assert sg.expected_information_gain == 0.5
+    assert sg.interruption_cost == 0.5
+
+
+def test_skip_subgoal_missing_title_or_description() -> None:
+    raw = '[{"question":"q","subgoals":[{"title":"t"},{"description":"d"},{"title":"t","description":"d"}]}]'
+    c = parse_curiosities(raw)[0]
+    assert len(c.subgoals) == 1  # 只有完整的保留