feat/ eval

api/tests/evaluation/test_importers.py

@@ -5,6 +5,8 @@ import pytest
 
 from app.features.evaluation.importers.user_export_markdown import (
     extract_dialogue_turns_from_export_md,
+    extract_memoir_chapter_sections_from_export_md,
+    extract_source_user_id_from_export_md,
     extract_user_utterances_from_export_md,
 )
 
@@ -72,3 +74,32 @@ def test_extract_dialogue_turns_from_repo_user_export() -> None:
     turns = extract_dialogue_turns_from_export_md(text)
     assert len(turns) >= 5
     assert "你好" in turns[0][0]
+
+
+def test_extract_source_user_id_from_export_md() -> None:
+    md = "- **User ID:** `e27fcd97-fefa-43b8-a7a3-3ecd49ebf5f0`\n"
+    assert (
+        extract_source_user_id_from_export_md(md)
+        == "e27fcd97-fefa-43b8-a7a3-3ecd49ebf5f0"
+    )
+
+
+def test_extract_memoir_chapter_sections_from_export_md() -> None:
+    md = """
+## 回忆录章节（生成正文）
+
+### First chapter
+
+Line a.
+{{IMAGE:foo}}
+
+### Second title
+
+Line b.
+"""
+    sections = extract_memoir_chapter_sections_from_export_md(md)
+    assert len(sections) == 2
+    assert sections[0][0] == "First chapter"
+    assert "Line a." in sections[0][1]
+    assert "{{IMAGE" not in sections[0][1]
+    assert sections[1][0] == "Second title"