Files
life-echo/api/tests/test_oral_normalize.py
Kevin 53d9e003af feat(api): 叙事 prompt、职业上下文、读路径章节、WS 解耦与错误脱敏
- 回忆录:事实边界补充允许清单;传记文体示例与 JSON 叙事要求对齐
- default 职业提示 occupation_context;cadre/military 退休语境
- GET 章节读路径零写入,prepare_chapter_read_view + markdown_for_response
- 文本归一抽到 core/text_normalize;移除弃用 reply 策略与 recompose_chapters_for_story
- ConversationService:WS 连接/用户段落/结束对话;对外错误固定文案
- 测试:HTTP 脱敏契约、章节读视图、occupation 与 background_voice
2026-04-01 11:55:52 +08:00

49 lines
1.7 KiB
Python

"""口述规则归一与 memoir 入口行为。"""
from unittest.mock import patch
from app.core.text_normalize import apply_oral_rules
from app.features.memoir.oral_normalize import normalize_oral_for_memoir
def test_apply_rules_mei_kanshang_wo() -> None:
assert "没看上我" in apply_oral_rules("我去试镜了 美看上我 张伟")
def test_apply_rules_mei_kanshang_ni() -> None:
assert apply_oral_rules("美看上你") == "没看上你"
def test_apply_rules_no_false_positive_rong() -> None:
"""「美容」等不应被误替换。"""
s = "我去了解美容项目"
assert apply_oral_rules(s) == s
def test_normalize_respects_global_off() -> None:
raw = "美看上我"
with patch("app.features.memoir.oral_normalize.settings") as m:
m.memoir_oral_normalize_enabled = False
m.memoir_oral_normalize_mode = "rules"
assert normalize_oral_for_memoir(raw, llm=None) == raw
def test_normalize_rules_mode_no_llm() -> None:
raw = "美看上我"
with patch("app.features.memoir.oral_normalize.settings") as m:
m.memoir_oral_normalize_enabled = True
m.memoir_oral_normalize_mode = "rules"
m.memoir_oral_normalize_llm_max_tokens = 512
m.memoir_oral_normalize_llm_max_input_chars = 8000
assert normalize_oral_for_memoir(raw, llm=None) == "没看上我"
def test_normalize_mode_off_string() -> None:
raw = "美看上我"
with patch("app.features.memoir.oral_normalize.settings") as m:
m.memoir_oral_normalize_enabled = True
m.memoir_oral_normalize_mode = "off"
m.memoir_oral_normalize_llm_max_tokens = 512
m.memoir_oral_normalize_llm_max_input_chars = 8000
assert normalize_oral_for_memoir(raw, llm=None) == raw