mirror of
https://ghfast.top/https://github.com/aeroxw/tick-stock-panel.git
synced 2026-09-12 21:24:16 +08:00
30 lines
888 B
Python
30 lines
888 B
Python
from __future__ import annotations
|
|
|
|
from app.strategy.ai_generator import GUIDE_PATH, AIStrategyGenerator
|
|
from app.strategy.prompt_builder import build_step1
|
|
|
|
|
|
def test_ai_strategy_generator_uses_compact_guide():
|
|
assert GUIDE_PATH.name == "strategy-guide-compact.md"
|
|
|
|
guide = AIStrategyGenerator()._get_guide()
|
|
|
|
assert "AI 策略生成精简指南" in guide
|
|
assert "策略示例" not in guide
|
|
assert len(guide) < 5000
|
|
|
|
|
|
def test_build_step1_keeps_user_prompt_compact():
|
|
prompt = build_step1(
|
|
"测试策略",
|
|
"测试描述",
|
|
"long",
|
|
"1. 收盘价站上 MA20\n2. 成交量放大\n3. RSI 不过热",
|
|
"ai_test",
|
|
)
|
|
|
|
assert "# 步骤 1：根据规则生成完整策略" not in prompt
|
|
assert "模式 A 框架" not in prompt
|
|
assert "策略ID（必须使用此ID）：ai_test" in prompt
|
|
assert len(prompt) < 1000
|