mirror of
https://github.com/primedigitaltech/market-assistant.git
synced 2026-07-22 08:01:34 +08:00
614d631261
补充策略决策键、fixture 与相关单测；分析/营销包本地存储与任务状态；竞品简报 schema 与 LLM 章节生成、促销与导出调整；并纳入 Cursor 工程规则 user-dev-style。 Made-with: Cursor
123 lines
4.5 KiB
Python
123 lines
4.5 KiB
Python
"""brief 按矩阵分组收窄。"""
|
|
from __future__ import annotations
|
|
|
|
from django.test import SimpleTestCase
|
|
|
|
from pipeline.reporting.brief_strategy_scope import (
|
|
filter_brief_for_strategy_matrix_group,
|
|
list_matrix_groups_for_api,
|
|
resolve_strategy_matrix_group_index,
|
|
)
|
|
|
|
|
|
class BriefStrategyScopeTests(SimpleTestCase):
|
|
def _sample_brief(self) -> dict:
|
|
return {
|
|
"schema_version": 1,
|
|
"keyword": "低GI",
|
|
"matrix_by_group": [
|
|
{
|
|
"group": "饮料",
|
|
"sku_count": 1,
|
|
"skus": [
|
|
{
|
|
"brand": "A",
|
|
"shop": "S1",
|
|
"category": "水饮 > 茶",
|
|
"list_price_show": "10",
|
|
}
|
|
],
|
|
},
|
|
{
|
|
"group": "饼干",
|
|
"sku_count": 2,
|
|
"skus": [
|
|
{
|
|
"brand": "B",
|
|
"shop": "S2",
|
|
"category": "休闲食品 > 饼干 > 粗粮饼干",
|
|
"标价": "100",
|
|
"券后到手价": "80",
|
|
},
|
|
{
|
|
"brand": "B",
|
|
"shop": "S2",
|
|
"category": "休闲食品 > 饼干 > 苏打饼干",
|
|
"标价": "50",
|
|
"券后到手价": "50",
|
|
},
|
|
],
|
|
},
|
|
],
|
|
"consumer_feedback_by_matrix_group": [
|
|
{
|
|
"group": "饮料",
|
|
"comment_rows": 5,
|
|
"effective_comment_text_units": 5,
|
|
},
|
|
{
|
|
"group": "饼干",
|
|
"comment_rows": 8,
|
|
"effective_comment_text_units": 8,
|
|
},
|
|
],
|
|
"usage_scenarios_by_matrix_group": [
|
|
{"group": "饮料", "scenarios": []},
|
|
{"group": "饼干", "scenarios": [{"scenario": "早餐", "count": 4}]},
|
|
],
|
|
"notes": ["原注"],
|
|
}
|
|
|
|
def test_list_matrix_groups(self) -> None:
|
|
b = self._sample_brief()
|
|
m = list_matrix_groups_for_api(b)
|
|
self.assertEqual(len(m), 2)
|
|
self.assertEqual(m[1]["group"], "饼干")
|
|
self.assertEqual(m[1]["index"], 1)
|
|
|
|
def test_resolve_by_label(self) -> None:
|
|
b = self._sample_brief()
|
|
idx, err = resolve_strategy_matrix_group_index(
|
|
b, matrix_group_label="饼干"
|
|
)
|
|
self.assertIsNone(err)
|
|
self.assertEqual(idx, 1)
|
|
|
|
def test_resolve_by_index(self) -> None:
|
|
b = self._sample_brief()
|
|
idx, err = resolve_strategy_matrix_group_index(
|
|
b, matrix_group_index=1
|
|
)
|
|
self.assertIsNone(err)
|
|
self.assertEqual(idx, 1)
|
|
|
|
def test_resolve_mismatch(self) -> None:
|
|
b = self._sample_brief()
|
|
idx, err = resolve_strategy_matrix_group_index(
|
|
b, matrix_group_index=0, matrix_group_label="饼干"
|
|
)
|
|
self.assertIsNotNone(err)
|
|
self.assertIsNone(idx)
|
|
|
|
def test_filter_keeps_only_group(self) -> None:
|
|
b = self._sample_brief()
|
|
out = filter_brief_for_strategy_matrix_group(b, matrix_group_index=1)
|
|
self.assertEqual(len(out["matrix_by_group"]), 1)
|
|
self.assertEqual(out["matrix_by_group"][0]["group"], "饼干")
|
|
self.assertEqual(len(out["matrix_by_group"][0]["skus"]), 2)
|
|
self.assertEqual(len(out["consumer_feedback_by_matrix_group"]), 1)
|
|
self.assertEqual(out["comment_focus_keywords"], [])
|
|
self.assertEqual(out["price_stats_source"], "strategy_scope_matrix_group_skus")
|
|
self.assertIn("strategy_scope_applied", out)
|
|
self.assertEqual(out["strategy_scope_applied"]["group"], "饼干")
|
|
|
|
def test_filter_recomputes_price_promotion_signals(self) -> None:
|
|
b = self._sample_brief()
|
|
out = filter_brief_for_strategy_matrix_group(b, matrix_group_index=1)
|
|
pps = out.get("price_promotion_signals")
|
|
self.assertIsInstance(pps, dict)
|
|
assert isinstance(pps, dict)
|
|
self.assertEqual(pps.get("row_count"), 2)
|
|
self.assertEqual(pps.get("rows_with_both_list_and_coupon"), 2)
|
|
self.assertEqual(pps.get("rows_coupon_below_list_price"), 1)
|