mirror of
https://github.com/primedigitaltech/market-assistant.git
synced 2026-07-22 08:01:34 +08:00
932c3d6aeb
报告配置 chapter8_text_mining_probe 为 True（默认）时，规则底稿第五节改为说明口径并去掉子串命中列表；LLM 请求从 structured_brief 剔除相关字段并附 omission_note。关闭探针时保留旧版第五节。任务 API 传入 report_config。 Made-with: Cursor
103 lines
3.8 KiB
Python
103 lines
3.8 KiB
Python
"""市场策略草稿 Markdown（规则，无 LLM）。"""
|
||
from __future__ import annotations
|
||
|
||
from django.test import SimpleTestCase
|
||
|
||
from pipeline.reporting.strategy_draft import build_strategy_draft_markdown
|
||
|
||
|
||
class StrategyDraftTests(SimpleTestCase):
|
||
def test_build_contains_sections_and_notes(self) -> None:
|
||
brief = {
|
||
"schema_version": 1,
|
||
"keyword": "测试K",
|
||
"batch_label": "b1",
|
||
"scope": {"merged_sku_count": 2, "comment_flat_rows": 5},
|
||
"strategy_hints": ["假设A"],
|
||
}
|
||
md = build_strategy_draft_markdown(
|
||
job_id=99,
|
||
keyword="测试K",
|
||
brief=brief,
|
||
business_notes="重点：华东",
|
||
generated_at_iso="2026-04-09T12:00:00",
|
||
)
|
||
self.assertIn("测试K", md)
|
||
self.assertIn("任务 ID**：99", md)
|
||
self.assertIn("假设A", md)
|
||
self.assertIn("重点：华东", md)
|
||
self.assertIn("战略背景与目标", md)
|
||
self.assertIn("市场策略制定草稿", md)
|
||
|
||
def test_strategy_decisions_merge(self) -> None:
|
||
brief = {"schema_version": 1, "keyword": "K", "batch_label": "b"}
|
||
decisions = {
|
||
"product_role": "追赶型",
|
||
"positioning_choice": "mid",
|
||
"competitive_stance": "flank",
|
||
"pillar_product": "做低糖配方",
|
||
"ack_risk_keywords": True,
|
||
"ack_risk_price": False,
|
||
"ack_risk_concentration": True,
|
||
}
|
||
md = build_strategy_draft_markdown(
|
||
job_id=1,
|
||
keyword="K",
|
||
brief=brief,
|
||
strategy_decisions=decisions,
|
||
report_config={"chapter8_text_mining_probe": False},
|
||
)
|
||
self.assertIn("**本品角色**：追赶型", md)
|
||
self.assertIn("- [x] **卡腰**", md)
|
||
self.assertIn("- [ ] **贴顶**", md)
|
||
self.assertIn("侧翼切入", md)
|
||
self.assertIn("| 产品 | 做低糖配方 |", md)
|
||
self.assertIn("- [x] 关注词/场景是否**以偏概全**", md)
|
||
self.assertIn("- [ ] 价格带是否含大促", md)
|
||
self.assertIn("- [x] 列表集中度与深入样本品牌是否**矛盾**", md)
|
||
|
||
def test_chapter8_probe_omits_focus_scenario_count_bullets(self) -> None:
|
||
brief = {
|
||
"schema_version": 1,
|
||
"keyword": "低GI",
|
||
"comment_focus_keywords": [{"word": "口感", "count": 501}],
|
||
"usage_scenarios": [
|
||
{
|
||
"scenario": "控糖/血糖相关",
|
||
"count": 305,
|
||
"share_of_text_units": 0.272,
|
||
}
|
||
],
|
||
}
|
||
md = build_strategy_draft_markdown(
|
||
job_id=1,
|
||
keyword="低GI",
|
||
brief=brief,
|
||
report_config={"chapter8_text_mining_probe": True},
|
||
)
|
||
self.assertIn("文本挖掘", md)
|
||
self.assertNotIn("假设**：用户决策中「口感」", md)
|
||
self.assertNotIn("场景命题**：「控糖", md)
|
||
|
||
def test_legacy_report_shows_focus_scenario_bullets(self) -> None:
|
||
brief = {
|
||
"schema_version": 1,
|
||
"keyword": "低GI",
|
||
"comment_focus_keywords": [{"word": "口感", "count": 501}],
|
||
"usage_scenarios": [
|
||
{
|
||
"scenario": "控糖/血糖相关",
|
||
"count": 305,
|
||
"share_of_text_units": 0.272,
|
||
}
|
||
],
|
||
}
|
||
md = build_strategy_draft_markdown(
|
||
job_id=1,
|
||
keyword="低GI",
|
||
brief=brief,
|
||
report_config={"chapter8_text_mining_probe": False},
|
||
)
|
||
self.assertIn("假设**：用户决策中「口感」", md)
|
||
self.assertIn("场景命题**：「控糖/血糖相关」", md)
|