"""CosyVoice EMOTION_MAP / normalize_emotion / build_emotion_instruction 单测（P1 修复 #1898）. 覆盖： - 7 种标准英文枚举 neutral/happy/sad/angry/surprised/fearful/disgusted （CosyVoice v3 官方 emotion 值） - 大小写不敏感 - 前端中文 7 标签（中立/开心/难过/生气/惊讶/恐惧/厌恶）→ 英文枚举 - 灵应指定别名（中性/伤心/愤怒/吃惊）→ 英文枚举 - 常见中文别名与旧英文 4 枚举兼容 - 空串/空白/None 边界 - 未知值默认 neutral（warning 日志） - build_emotion_instruction 三路分支： · 克隆/设计音色 → "Speak in a {emotion} tone." · 支持 emotion Instruct 的系统音色（longanyang/longanhuan/longhuhu_v3）→ "你说话的情感是{emotion}。" · 默认系统音色（含 longxiaochun_v3）→ 返回空串（不传 instruction） """ from __future__ import annotations import logging import pytest from packages.application.cosyvoice_service import ( EMOTION_MAP, build_emotion_instruction, normalize_emotion, ) # 7 种官方英文枚举 SEVEN_STANDARD_ENUMS = [ "neutral", "happy", "sad", "angry", "surprised", "fearful", "disgusted", ] # 前端中文 7 标签 → 期望英文枚举 FRONTEND_CN_LABELS = [ ("中立", "neutral"), ("开心", "happy"), ("难过", "sad"), ("生气", "angry"), ("惊讶", "surprised"), ("恐惧", "fearful"), ("厌恶", "disgusted"), ] # 灵应派任务补充的中文别名 LINGYING_CN_ALIASES = [ ("中性", "neutral"), ("伤心", "sad"), ("愤怒", "angry"), ("吃惊", "surprised"), ] # 其他常见中文别名 CN_ALIASES = [ ("自然", "neutral"), ("愉快", "happy"), ("高兴", "happy"), ("快乐", "happy"), ("兴奋", "happy"), ("悲伤", "sad"), ("惊奇", "surprised"), ("害怕", "fearful"), ("讨厌", "disgusted"), ] # 旧英文 4 枚举 → 最接近的标准枚举 OLD_FOUR_ENUMS = [ ("natural", "neutral"), ("excited", "happy"), ("calm", "neutral"), ("friendly", "happy"), ] # 支持 emotion Instruct 的系统音色（白名单） SYSTEM_EMOTION_VOICES = ["longanyang", "longanhuan", "longhuhu_v3"] # 不支持 Instruct 的典型系统音色（含默认音色 longxiaochun_v3） NON_INSTRUCT_SYSTEM_VOICES = [ "longxiaochun_v3", "longxiaoxia_v3", "longsanshu_v3", "longyue_v3", "longyingjing_v3", "loongabby_v3", "loongandy_v3", "longfei_v3", ] # 克隆/设计音色（非 long/loong 前缀） CLONED_VOICE_IDS = [ "myclone_abc123", "xiaoming_20260915", "clone_voice_42", "custom_voice_test", ] class TestEmotionMapSevenStandard: @pytest.mark.parametrize("enum_val", SEVEN_STANDARD_ENUMS) def test_standard_enum_maps_to_self(self, enum_val: str) -> None: assert enum_val in EMOTION_MAP assert EMOTION_MAP[enum_val] == enum_val @pytest.mark.parametrize("enum_val", SEVEN_STANDARD_ENUMS) def test_normalize_standard_enum(self, enum_val: str) -> None: assert normalize_emotion(enum_val) == enum_val @pytest.mark.parametrize("enum_val", SEVEN_STANDARD_ENUMS) def test_normalize_case_insensitive(self, enum_val: str) -> None: assert normalize_emotion(enum_val.upper()) == enum_val assert normalize_emotion(enum_val.capitalize()) == enum_val assert normalize_emotion(f" {enum_val} ") == enum_val def test_neutral_maps_to_neutral(self) -> None: assert normalize_emotion("neutral") == "neutral" class TestFrontendCnLabels: @pytest.mark.parametrize("cn,expected", FRONTEND_CN_LABELS) def test_cn_label_normalizes_to_enum(self, cn: str, expected: str) -> None: assert normalize_emotion(cn) == expected class TestLingyingSpecAliases: @pytest.mark.parametrize("cn,expected", LINGYING_CN_ALIASES) def test_lingying_aliases(self, cn: str, expected: str) -> None: assert normalize_emotion(cn) == expected class TestBackwardCompatAliases: @pytest.mark.parametrize("old_key,expected", OLD_FOUR_ENUMS) def test_old_four_enums(self, old_key: str, expected: str) -> None: assert normalize_emotion(old_key) == expected @pytest.mark.parametrize("cn_key,expected", CN_ALIASES) def test_chinese_aliases(self, cn_key: str, expected: str) -> None: assert normalize_emotion(cn_key) == expected class TestNormalizeEmotionEdgeCases: @pytest.mark.parametrize("empty_val", ["", None]) def test_empty_or_none_returns_empty(self, empty_val) -> None: assert normalize_emotion(empty_val) == "" @pytest.mark.parametrize("ws", [" ", "\t", "\n", " \n "]) def test_whitespace_only_returns_empty(self, ws: str) -> None: assert normalize_emotion(ws) == "" def test_unknown_value_defaults_to_neutral_with_warning(self, caplog) -> None: caplog.set_level(logging.WARNING) result = normalize_emotion("not_a_real_emotion_xyz") assert result == "neutral" assert any("未知的 emotion" in r.message for r in caplog.records) def test_strips_leading_trailing_whitespace(self) -> None: assert normalize_emotion(" happy ") == "happy" assert normalize_emotion(" 生气 ") == "angry" assert normalize_emotion(" 中性 ") == "neutral" class TestBuildEmotionInstructionClonedVoice: @pytest.mark.parametrize("voice_id", CLONED_VOICE_IDS) @pytest.mark.parametrize("enum_val", SEVEN_STANDARD_ENUMS) def test_cloned_voice_uses_english_tone_format(self, voice_id: str, enum_val: str) -> None: inst = build_emotion_instruction(voice_id, enum_val) assert inst == f"Speak in a {enum_val} tone." assert inst.isascii(), f"克隆音色 instruction 必须是纯 ASCII 英文: {inst!r}" @pytest.mark.parametrize( "cn,expected", FRONTEND_CN_LABELS + LINGYING_CN_ALIASES + CN_ALIASES, ) def test_cloned_voice_chinese_input_english_output(self, cn: str, expected: str) -> None: norm = normalize_emotion(cn) inst = build_emotion_instruction("myclone_voice", norm) assert inst == f"Speak in a {expected} tone." assert inst.isascii() def test_cloned_voice_empty_emotion_returns_empty(self) -> None: assert build_emotion_instruction("myclone", "") == "" @pytest.mark.parametrize("voice_id", CLONED_VOICE_IDS) def test_cloned_voice_unknown_emotion_falls_back_neutral(self, voice_id: str) -> None: norm = normalize_emotion("unknown_xyz") assert norm == "neutral" inst = build_emotion_instruction(voice_id, norm) assert inst == "Speak in a neutral tone." class TestBuildEmotionInstructionSystemVoiceEmotion: CN_PREFIX = "你说话的情感是" CN_SUFFIX = "。" @pytest.mark.parametrize("voice_id", SYSTEM_EMOTION_VOICES) @pytest.mark.parametrize("enum_val", SEVEN_STANDARD_ENUMS) def test_system_emotion_voice_cn_fixed_format(self, voice_id: str, enum_val: str) -> None: inst = build_emotion_instruction(voice_id, enum_val) assert inst == f"{self.CN_PREFIX}{enum_val}{self.CN_SUFFIX}" assert inst.count("。") == 1 mid = inst[len(self.CN_PREFIX) : -len(self.CN_SUFFIX)] assert mid == enum_val assert mid.isascii(), f"系统音色 emotion 值必须是纯 ASCII 英文枚举: {inst!r}" @pytest.mark.parametrize("voice_id", SYSTEM_EMOTION_VOICES) def test_system_emotion_voice_empty_returns_empty(self, voice_id: str) -> None: assert build_emotion_instruction(voice_id, "") == "" class TestBuildEmotionInstructionNonInstructSystemVoice: @pytest.mark.parametrize("voice_id", NON_INSTRUCT_SYSTEM_VOICES) @pytest.mark.parametrize("enum_val", SEVEN_STANDARD_ENUMS) def test_non_instruct_voice_returns_empty(self, voice_id: str, enum_val: str) -> None: assert build_emotion_instruction(voice_id, enum_val) == "" def test_default_voice_longxiaochun_v3_no_instruction(self) -> None: assert build_emotion_instruction("longxiaochun_v3", "happy") == "" assert build_emotion_instruction("longxiaochun_v3", "neutral") == "" class TestBuildEmotionInstructionEdgeCases: def test_empty_voice_id_treated_as_system(self) -> None: assert build_emotion_instruction("", "happy") == "" @pytest.mark.parametrize( "voice_id,enum_val,expected", [ ("MYCLONE_VOICE", "happy", "Speak in a happy tone."), ("CloneVoice", "sad", "Speak in a sad tone."), ], ) def test_voice_id_case_handling(self, voice_id: str, enum_val: str, expected: str) -> None: assert build_emotion_instruction(voice_id, enum_val) == expected def test_loong_prefix_is_system_voice(self) -> None: assert build_emotion_instruction("loongandy_v3", "happy") == "" assert build_emotion_instruction("loongabby_v3", "angry") == ""