mirror of
https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git
synced 2026-09-20 17:08:18 +00:00
8b34725872
- test_prompt_manager.py: 验证系统提示词包含5步流程、所有工具列表、无Python函数注记 - test_web_api.py: 验证关键API端点存在、静态文件路径正确、提示词可加载
89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
"""Tests for PromptManager - system prompt loading"""
|
||
|
||
from core.prompts import PromptManager
|
||
|
||
|
||
def test_get_system_prompt_not_empty():
|
||
"""系统提示词不应为空"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert prompt is not None
|
||
assert len(prompt) > 100
|
||
|
||
|
||
def test_system_prompt_contains_identity():
|
||
"""系统提示词应包含核心身份"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert "TrulyMEM" in prompt
|
||
|
||
|
||
def test_system_prompt_has_execution_order():
|
||
"""系统提示词应包含强制执行顺序"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert "强制执行顺序" in prompt
|
||
|
||
|
||
def test_system_prompt_has_memory_commit_step():
|
||
"""系统提示词应包含 memory_commit 写入步骤"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert "memory_commit (写入关键信息)" in prompt
|
||
|
||
|
||
def test_system_prompt_has_five_steps():
|
||
"""系统提示词应有5个执行步骤"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
# 检查步骤个数
|
||
steps = [l for l in prompt.split('\n') if l.strip().startswith('步骤')]
|
||
assert len(steps) == 5, f"期望5个步骤，实际{len(steps)}个"
|
||
|
||
|
||
def test_system_prompt_has_triplet_rules():
|
||
"""系统提示词应包含三元组规范"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert "三元组规范" in prompt or "短关键字" in prompt
|
||
|
||
|
||
def test_system_prompt_has_memory_tools():
|
||
"""系统提示词应列出记忆工具"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert "memory_recall" in prompt
|
||
assert "memory_commit" in prompt
|
||
assert "memory_purge" in prompt
|
||
|
||
|
||
def test_system_prompt_has_persona_tools():
|
||
"""系统提示词应列出人设工具"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert "persona_update" in prompt
|
||
assert "persona_clear" in prompt
|
||
|
||
|
||
def test_system_prompt_has_task_tools():
|
||
"""系统提示词应列出任务工具"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert "task_create" in prompt
|
||
assert "task_set_state" in prompt
|
||
assert "task_delete" in prompt
|
||
|
||
|
||
def test_system_prompt_no_python_function_notes():
|
||
"""系统提示词不应包含Python函数实现注记"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert "Python函数" not in prompt
|
||
|
||
|
||
def test_system_prompt_has_memory_principles():
|
||
"""系统提示词应包含记忆原则"""
|
||
pm = PromptManager()
|
||
prompt = pm.get_system_prompt()
|
||
assert "明确内容必须写入" in prompt
|