146 lines
5.0 KiB
Python
146 lines
5.0 KiB
Python
"""主模块测试套件。
|
|
|
|
本文件包含三个测试类，分别覆盖：
|
|
1. 配置加载功能（从 config.json 读取项目设置）
|
|
2. 语言规则生成（根据配置或自动检测确定输出语言）
|
|
3. 分析提示构建（组装完整的 AI 分析提示词）
|
|
"""
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# 将 src 目录加入 Python 路径，使测试可以导入项目模块
|
|
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
|
|
|
|
from config import get_language, load_config
|
|
from main import build_analysis_prompt, language_rule
|
|
|
|
|
|
class TestConfig:
|
|
"""测试配置加载模块。
|
|
|
|
验证 config.json 能否被正确读取，以及语言设置是否能正常获取。
|
|
"""
|
|
|
|
def test_load_config_returns_dict(self) -> None:
|
|
"""测试 load_config 返回字典且包含 language 键。
|
|
|
|
验证配置文件的解析结果是一个字典结构，
|
|
并且其中包含必需的 language 字段。
|
|
"""
|
|
config = load_config()
|
|
assert isinstance(config, dict)
|
|
assert "language" in config
|
|
|
|
def test_get_language_from_config(self) -> None:
|
|
"""测试 get_language 能正常返回语言设置。
|
|
|
|
当 config.json 中 language 不为 "auto" 时，
|
|
get_language 应返回非 None 值。
|
|
"""
|
|
language = get_language()
|
|
assert language is not None
|
|
|
|
|
|
class TestLanguageRule:
|
|
"""测试语言规则生成函数。
|
|
|
|
language_rule 负责根据配置或源文档内容，
|
|
生成指导 AI 输出语言的提示文本。
|
|
"""
|
|
|
|
def test_explicit_language(self) -> None:
|
|
"""测试显式指定语言时，提示中包含该语言名称。
|
|
|
|
当 configured_language 为 "Chinese" 时，
|
|
返回的提示应包含 "Chinese" 字样。
|
|
"""
|
|
result = language_rule("test content", "Chinese")
|
|
assert "Chinese" in result
|
|
|
|
def test_auto_with_chinese_content(self) -> None:
|
|
"""测试中文内容自动检测。
|
|
|
|
当源文档包含中文字符且未指定语言时，
|
|
应自动选择简体中文作为输出语言。
|
|
"""
|
|
result = language_rule("这是一段中文内容", None)
|
|
assert "简体中文" in result
|
|
|
|
def test_auto_with_english_content(self) -> None:
|
|
"""测试英文内容自动检测。
|
|
|
|
当源文档为纯英文且未指定语言时，
|
|
应返回跟随源文档语言的提示。
|
|
"""
|
|
result = language_rule("This is English content", None)
|
|
assert "源文档" in result
|
|
|
|
def test_auto_detection_string(self) -> None:
|
|
"""测试传入 "auto" 字符串时触发自动检测。
|
|
|
|
"auto" 和 None 等效，都表示启用自动语言检测。
|
|
"""
|
|
result = language_rule("test", "auto")
|
|
assert "源文档" in result
|
|
|
|
|
|
class TestBuildAnalysisPrompt:
|
|
"""测试分析提示构建函数。
|
|
|
|
build_analysis_prompt 负责将 wiki 目的、索引、
|
|
源文档内容和语言设置组装成完整的 AI 提示词。
|
|
"""
|
|
|
|
def test_basic_prompt(self) -> None:
|
|
"""测试基础提示词包含核心结构。
|
|
|
|
即使目的和索引都为空，提示词仍应包含
|
|
角色定义（专业研究分析师）和分析框架（关键实体）。
|
|
"""
|
|
prompt = build_analysis_prompt(purpose="", index="")
|
|
assert "专业研究分析师" in prompt
|
|
assert "关键实体" in prompt
|
|
|
|
def test_with_purpose(self) -> None:
|
|
"""测试传入目的时，提示词包含 Wiki 目的部分。
|
|
|
|
当 purpose 参数非空时，生成的提示词应包含
|
|
"Wiki 目的" 标题和传入的目的内容。
|
|
"""
|
|
prompt = build_analysis_prompt(purpose="测试目的", index="")
|
|
assert "Wiki 目的" in prompt
|
|
assert "测试目的" in prompt
|
|
|
|
def test_with_index(self) -> None:
|
|
"""测试传入索引时，提示词包含 Wiki 索引部分。
|
|
|
|
当 index 参数非空时，生成的提示词应包含
|
|
"当前 Wiki 索引" 标题和传入的索引内容。
|
|
"""
|
|
prompt = build_analysis_prompt(purpose="", index="测试索引")
|
|
assert "当前 Wiki 索引" in prompt
|
|
assert "测试索引" in prompt
|
|
|
|
def test_empty_content(self) -> None:
|
|
"""测试空内容时，可选部分不出现在提示词中。
|
|
|
|
当 purpose 和 index 都为空时，
|
|
提示词不应包含 "Wiki 目的" 和 "当前 Wiki 索引" 部分，
|
|
以保持提示词简洁。
|
|
"""
|
|
prompt = build_analysis_prompt(purpose="", index="")
|
|
assert "Wiki 目的" not in prompt
|
|
assert "当前 Wiki 索引" not in prompt
|
|
|
|
def test_language_from_config(self) -> None:
|
|
"""测试语言设置从配置文件读取。
|
|
|
|
当 configured_language 为 None 时，
|
|
build_analysis_prompt 会从 config.json 读取语言设置，
|
|
生成的提示词应包含语言规则。
|
|
"""
|
|
prompt = build_analysis_prompt(purpose="", index="", configured_language=None)
|
|
assert "使用" in prompt
|
|
assert "语言输出分析报告" in prompt
|