mirror of
https://github.com/primedigitaltech/market-assistant.git
synced 2026-07-21 23:41:39 +08:00
6143ad6180
新增 strategy_matrix_group / strategy_matrix_group_index 请求参数；收窄后价盘与集中度按分组内 SKU 重算；摘要 GET 与策略 POST 响应附带 matrix_groups。 Made-with: Cursor
127 lines
4.6 KiB
Python
127 lines
4.6 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": "休闲食品 > 饼干 > 粗粮饼干",
|
|
"list_price_show": "20",
|
|
},
|
|
{
|
|
"brand": "B",
|
|
"shop": "S2",
|
|
"category": "休闲食品 > 饼干 > 苏打饼干",
|
|
"list_price_show": "22",
|
|
},
|
|
],
|
|
},
|
|
],
|
|
"consumer_feedback_by_matrix_group": [
|
|
{
|
|
"group": "饮料",
|
|
"comment_rows": 5,
|
|
"effective_comment_text_units": 5,
|
|
"focus_keyword_hits": [{"word": "甜", "count": 2}],
|
|
"scenarios_top": [
|
|
{
|
|
"scenario": "解渴",
|
|
"count": 2,
|
|
"share_of_text_units": 0.4,
|
|
}
|
|
],
|
|
},
|
|
{
|
|
"group": "饼干",
|
|
"comment_rows": 8,
|
|
"effective_comment_text_units": 8,
|
|
"focus_keyword_hits": [{"word": "脆", "count": 3}],
|
|
"scenarios_top": [
|
|
{
|
|
"scenario": "早餐",
|
|
"count": 4,
|
|
"share_of_text_units": 0.5,
|
|
}
|
|
],
|
|
},
|
|
],
|
|
"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"][0]["word"], "脆")
|
|
self.assertEqual(out["price_stats_source"], "strategy_scope_matrix_group_skus")
|
|
self.assertIn("strategy_scope_applied", out)
|
|
self.assertEqual(out["strategy_scope_applied"]["group"], "饼干")
|