25 lines
548 B
Python
25 lines
548 B
Python
|
"""导出 Markdown 中 AI 块的 [SPLIT] 规范化。"""
|
||
|
|
|
||
|
|
from app.features.evaluation.importers.user_export_markdown import (
|
||
|
|
extract_dialogue_turns_from_export_md,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_extract_dialogue_turns_replaces_split_in_ai():
|
||
|
|
md = """
|
||
|
|
#### 轮次 1
|
||
|
|
|
||
|
|
**用户:**
|
||
|
|
你好。
|
||
|
|
|
||
|
|
**AI:**
|
||
|
|
第一段[SPLIT]第二段
|
||
|
|
"""
|
||
|
|
turns = extract_dialogue_turns_from_export_md(md)
|
||
|
|
assert len(turns) == 1
|
||
|
|
u, a = turns[0]
|
||
|
|
assert u == "你好。"
|
||
|
|
assert "[SPLIT]" not in a
|
||
|
|
assert "第一段" in a and "第二段" in a
|
||
|
|
assert "\n" in a
|