mirror of
https://github.com/primedigitaltech/market-assistant.git
synced 2026-07-21 23:41:39 +08:00
57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
"""商详包：两步 JSON LLM 与解析。"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
from unittest.mock import patch
|
||
|
||
from django.test import SimpleTestCase
|
||
|
||
from pipeline.llm.generate_marketing_detail import (
|
||
_parse_llm_json,
|
||
generate_marketing_detail_pack,
|
||
)
|
||
|
||
|
||
class MarketingDetailPackTests(SimpleTestCase):
|
||
def test_parse_llm_json_strips_wrapped(self) -> None:
|
||
raw = '前缀 {"a": 1, "b": "x"} 后缀'
|
||
d = _parse_llm_json(raw)
|
||
self.assertEqual(d, {"a": 1, "b": "x"})
|
||
|
||
def test_parse_llm_json_rejects_non_object(self) -> None:
|
||
with self.assertRaises(ValueError):
|
||
_parse_llm_json("[1,2]")
|
||
|
||
def test_generate_marketing_detail_pack_two_calls(self) -> None:
|
||
core = {
|
||
"one_liner_value": "低负担早餐选择",
|
||
"buyer_job_to_be_done": "控糖加餐",
|
||
"key_pain_or_desire": "怕升糖",
|
||
"why_this_product": "配方可核对",
|
||
"proof_or_trust_angle": "输入未体现",
|
||
"differentiation_vs_alternatives": "同价带更少添加糖",
|
||
"price_value_framing": "中端",
|
||
"compliance_taboos": "不涉疗效",
|
||
"open_points_for_business": "",
|
||
}
|
||
pack = {
|
||
"listing_titles": ["标题A", "标题B"],
|
||
"listing_subtitle": "副文案",
|
||
"detail_headline": "首屏一句",
|
||
"selling_bullets": ["卖点1"],
|
||
"spec_sidebar_lines": [],
|
||
"faq": [{"question": "是否低GI？", "answer": "以包装与检测为准。"}],
|
||
}
|
||
with patch(
|
||
"pipeline.llm.generate_marketing_detail.call_llm",
|
||
side_effect=[json.dumps(core, ensure_ascii=False), json.dumps(pack, ensure_ascii=False)],
|
||
):
|
||
out = generate_marketing_detail_pack(
|
||
keyword="低GI",
|
||
strategy_markdown="# 策略\n\n- 要点",
|
||
strategy_decisions={"product_role": "新品"},
|
||
business_notes="",
|
||
)
|
||
self.assertEqual(out["core_info_card"]["one_liner_value"], "低负担早餐选择")
|
||
self.assertEqual(out["detail_page_pack"]["listing_titles"][0], "标题A")
|