"""导出 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