315 lines
14 KiB
Python
315 lines
14 KiB
Python
import os
|
||
import sys
|
||
import unittest
|
||
from types import SimpleNamespace
|
||
|
||
sys.path.insert(0, os.path.dirname(__file__))
|
||
|
||
from app import product_suite, prompts
|
||
|
||
|
||
class ProductSuiteTests(unittest.TestCase):
|
||
def test_defaults_and_per_image_total(self):
|
||
settings = product_suite.default_suite_settings()
|
||
|
||
self.assertEqual(
|
||
["白底图", "场景图", "模特场景图", "细节说明图", "卖点图"],
|
||
product_suite.category_order(settings),
|
||
)
|
||
self.assertEqual(5, product_suite.suite_total_count(settings, 3))
|
||
|
||
settings["per_image_primary"] = True
|
||
|
||
self.assertEqual(13, product_suite.suite_total_count(settings, 3))
|
||
|
||
def test_custom_category_validation_and_normalization(self):
|
||
self.assertEqual("分类名称不能为空", product_suite.suite_name_error(""))
|
||
self.assertEqual("分类名称不能包含空格", product_suite.suite_name_error("使用 场景"))
|
||
self.assertEqual("分类名称不能超过10个字", product_suite.suite_name_error("一二三四五六七八九十甲"))
|
||
self.assertEqual("分类名称已存在", product_suite.suite_name_error("场景图", {"场景图"}))
|
||
|
||
settings = product_suite.normalize_suite_settings(
|
||
{
|
||
"ratio": "3:4",
|
||
"categories": {"白底图": 2, "场景图": 0, "卖点图": 1, "尺寸图": 2},
|
||
"custom_category_order": ["尺寸图"],
|
||
}
|
||
)
|
||
|
||
self.assertEqual("3:4", settings["ratio"])
|
||
self.assertEqual(
|
||
["白底图", "场景图", "模特场景图", "细节说明图", "卖点图", "尺寸图"],
|
||
product_suite.category_order(settings),
|
||
)
|
||
self.assertEqual(5, product_suite.suite_total_count(settings, 1))
|
||
|
||
def test_recent_settings_only_keep_allowed_dropdown_values(self):
|
||
self.assertEqual(
|
||
{
|
||
"platform": "Shopee",
|
||
"country": "中国台湾",
|
||
"language": "繁体中文",
|
||
"ratio": "1:1",
|
||
},
|
||
product_suite.last_suite_settings(
|
||
{
|
||
"platform": "未知平台",
|
||
"country": "未知站点",
|
||
"language": "未知语言",
|
||
"ratio": "2:3",
|
||
"per_image_primary": True,
|
||
"categories": {"白底图": 9},
|
||
}
|
||
),
|
||
)
|
||
selected = product_suite.last_suite_settings(
|
||
{
|
||
"platform": "Amazon",
|
||
"country": "新加坡",
|
||
"language": "英文",
|
||
"ratio": "4:3",
|
||
}
|
||
)
|
||
self.assertEqual("Amazon", selected["platform"])
|
||
self.assertEqual("新加坡", selected["country"])
|
||
self.assertEqual("英文", selected["language"])
|
||
self.assertEqual("4:3", selected["ratio"])
|
||
|
||
def test_job_specs_include_selected_context_and_source_assignment(self):
|
||
settings = product_suite.default_suite_settings()
|
||
settings.update(
|
||
{
|
||
"platform": "Shopee",
|
||
"country": "中国台湾",
|
||
"language": "繁体中文",
|
||
"ratio": "4:3",
|
||
"per_image_primary": True,
|
||
"categories": {"白底图": 1, "场景图": 1, "卖点图": 0},
|
||
}
|
||
)
|
||
assets = [SimpleNamespace(id=11), SimpleNamespace(id=12)]
|
||
|
||
specs = product_suite.build_job_specs(
|
||
assets,
|
||
"40小时续航，适合通勤",
|
||
settings,
|
||
"51100639510",
|
||
template_text=prompts.load_default_product_suite_prompt(),
|
||
)
|
||
|
||
self.assertEqual(3, len(specs))
|
||
self.assertEqual([11, 11, 12], [spec["source_asset_id"] for spec in specs])
|
||
self.assertEqual([[], [], []], [spec["reference_asset_ids"] for spec in specs])
|
||
self.assertEqual(["白底图", "场景图", "场景图"], [spec["job_type"] for spec in specs])
|
||
for spec in specs:
|
||
self.assertIn("平台：Shopee", spec["prompt"])
|
||
self.assertIn("国家地区：中国台湾", spec["prompt"])
|
||
self.assertIn("输出语言：繁体中文", spec["prompt"])
|
||
self.assertIn("本次生成比例：4:3", spec["prompt"])
|
||
self.assertNotIn("商品ID：51100639510", spec["prompt"])
|
||
self.assertIn("40小时续航", spec["prompt"])
|
||
self.assertIn("禁止海报长图、详情页长图和多宫格拼接版面", spec["prompt"])
|
||
self.assertIn("禁止在画面中出现任何国旗", spec["prompt"])
|
||
self.assertIn("参考图规则:当前上传图片是本任务唯一主参考图", spec["prompt"])
|
||
self.assertNotIn("参考图规则：", spec["prompt"])
|
||
self.assertNotIn("序号%d" % spec["source_index"], spec["prompt"])
|
||
self.assertTrue(
|
||
spec["prompt"].splitlines()[-1].startswith("本次生成比例：4:3")
|
||
)
|
||
|
||
self.assertIn(
|
||
"生成目标：白底图,生成 Shopee 台灣商品白底主圖，純白背景，商品清晰置中，不添加多餘文字。",
|
||
specs[0]["prompt"],
|
||
)
|
||
self.assertIn(
|
||
"生成目标：场景图,生成生活化使用場景圖，展示商品在真實情境中的用途與氛圍，畫面自然可信。",
|
||
specs[1]["prompt"],
|
||
)
|
||
white_prompt = specs[0]["prompt"]
|
||
reference_rule = (
|
||
"当前上传图片是本任务唯一主参考图；保持商品主体、款式、颜色和关键细节准确；"
|
||
"不编造用户与参考图均未提供的信息。"
|
||
)
|
||
self.assertLess(white_prompt.index("重要尺寸要求"), white_prompt.index("重要禁用内容"))
|
||
self.assertLess(white_prompt.index("重要禁用内容"), white_prompt.index("价格信息规则"))
|
||
self.assertLess(white_prompt.index("价格信息规则"), white_prompt.index("尺码信息规则"))
|
||
self.assertLess(white_prompt.index("尺码信息规则"), white_prompt.index(reference_rule))
|
||
self.assertLess(white_prompt.index(reference_rule), white_prompt.index("商品卖点与要求"))
|
||
|
||
def test_job_specs_freeze_first_image_references_when_not_per_image_primary(self):
|
||
settings = product_suite.default_suite_settings()
|
||
settings.update(
|
||
{
|
||
"per_image_primary": False,
|
||
"categories": {"白底图": 1, "场景图": 1, "卖点图": 0},
|
||
}
|
||
)
|
||
specs = product_suite.build_job_specs(
|
||
[SimpleNamespace(id=11), SimpleNamespace(id=12), SimpleNamespace(id=13)],
|
||
"卖点",
|
||
settings,
|
||
"51100639510",
|
||
template_text=prompts.load_default_product_suite_prompt(),
|
||
)
|
||
|
||
self.assertEqual(2, len(specs))
|
||
self.assertEqual([11, 11], [spec["source_asset_id"] for spec in specs])
|
||
self.assertEqual([[12, 13], [12, 13]], [spec["reference_asset_ids"] for spec in specs])
|
||
|
||
def test_job_specs_use_fixed_reference_rule_for_one_three_and_eight_images(self):
|
||
settings = product_suite.default_suite_settings()
|
||
settings.update(
|
||
{
|
||
"per_image_primary": False,
|
||
"categories": {"白底图": 1, "场景图": 0, "卖点图": 0},
|
||
}
|
||
)
|
||
fixed_rule = "参考图规则:使用第一張上傳圖作為主商品圖，其餘圖片只作為參考。"
|
||
|
||
for image_count in (1, 3, 8):
|
||
specs = product_suite.build_job_specs(
|
||
[SimpleNamespace(id=index) for index in range(1, image_count + 1)],
|
||
"卖点",
|
||
settings,
|
||
"51100639510",
|
||
template_text=prompts.load_default_product_suite_prompt(),
|
||
)
|
||
self.assertEqual(1, len(specs))
|
||
self.assertIn(fixed_rule, specs[0]["prompt"])
|
||
self.assertNotIn("第2至", specs[0]["prompt"])
|
||
|
||
def test_reference_rule_follows_per_image_primary_setting(self):
|
||
template = prompts.load_default_product_suite_prompt()
|
||
settings = product_suite.default_suite_settings()
|
||
|
||
fixed_rule = "参考图规则:使用第一張上傳圖作為主商品圖，其餘圖片只作為參考。"
|
||
for reference_count in (0, 2, 7):
|
||
not_per_image_primary = product_suite.build_suite_prompt(
|
||
"卖点",
|
||
settings,
|
||
"场景图",
|
||
"51100639510",
|
||
reference_asset_count=reference_count,
|
||
template_text=template,
|
||
)
|
||
self.assertIn(fixed_rule, not_per_image_primary)
|
||
self.assertNotIn("第2至", not_per_image_primary)
|
||
|
||
settings["per_image_primary"] = True
|
||
per_image_primary = product_suite.build_suite_prompt(
|
||
"卖点",
|
||
settings,
|
||
"场景图",
|
||
"51100639510",
|
||
source_index=2,
|
||
template_text=template,
|
||
)
|
||
self.assertIn(
|
||
"参考图规则:当前上传图片是本任务唯一主参考图；保持商品主体、款式、颜色和关键细节准确；不编造用户与参考图均未提供的信息。",
|
||
per_image_primary,
|
||
)
|
||
self.assertNotIn("第2至", per_image_primary)
|
||
self.assertNotIn("序号2", per_image_primary)
|
||
|
||
self.assertEqual(7, product_suite.reference_asset_count({}, 9))
|
||
|
||
def test_product_suite_template_validation_and_custom_category_rendering(self):
|
||
default_text = prompts.load_default_product_suite_prompt()
|
||
self.assertEqual([], product_suite.product_suite_prompt_errors(default_text))
|
||
|
||
unknown = default_text + "\n{未知变量}"
|
||
self.assertIn("未知变量", "；".join(product_suite.product_suite_prompt_errors(unknown)))
|
||
|
||
missing = default_text.replace("{图片比例}", "")
|
||
self.assertIn(
|
||
"缺少必需变量",
|
||
"；".join(product_suite.product_suite_prompt_errors(missing)),
|
||
)
|
||
|
||
inline_rule = default_text.replace(
|
||
"{价格信息规则}",
|
||
"价格：{价格信息规则}",
|
||
)
|
||
self.assertIn(
|
||
"只读规则变量必须独占一行",
|
||
"；".join(product_suite.product_suite_prompt_errors(inline_rule)),
|
||
)
|
||
|
||
literal_brace = default_text + "\n普通内容{"
|
||
self.assertIn(
|
||
"不支持的字面花括号",
|
||
"；".join(product_suite.product_suite_prompt_errors(literal_brace)),
|
||
)
|
||
|
||
settings = product_suite.default_suite_settings()
|
||
context = product_suite.product_suite_prompt_context(
|
||
"突出轻量材质",
|
||
settings,
|
||
"尺寸图",
|
||
"draft_hidden",
|
||
)
|
||
rendered = product_suite.render_product_suite_prompt(default_text, context)
|
||
self.assertIn("生成目标：尺寸图,生成自定义分类图片：尺寸图。", rendered)
|
||
self.assertNotIn("draft_", rendered)
|
||
self.assertNotIn("套图名称：", rendered)
|
||
|
||
legacy_template = default_text.replace(
|
||
"生成目标：{套图名称},{生成目标}",
|
||
"套图名称：{套图名称}{补充描述}",
|
||
)
|
||
self.assertEqual([], product_suite.product_suite_prompt_errors(legacy_template))
|
||
legacy_rendered = product_suite.render_product_suite_prompt(legacy_template, context)
|
||
self.assertIn("套图名称：尺寸图，生成自定义分类图片：尺寸图。", legacy_rendered)
|
||
|
||
def test_category_descriptions_are_fixed_and_custom_name_is_explicit(self):
|
||
expected = {
|
||
"白底图": "生成 Shopee 台灣商品白底主圖，純白背景，商品清晰置中，不添加多餘文字。",
|
||
"场景图": "生成生活化使用場景圖，展示商品在真實情境中的用途與氛圍，畫面自然可信。",
|
||
"模特场景图": "生成模特或手持使用情境圖，畫面自然可信，商品為主角。",
|
||
"细节说明图": "生成商品細節特寫說明圖，突出材質、做工、接口、紋理或關鍵細節。",
|
||
"卖点图": "生成賣點詳解圖，使用繁體中文短文案呈現核心優勢，版面乾淨。",
|
||
}
|
||
for category, description in expected.items():
|
||
self.assertEqual(description, product_suite.category_description(category))
|
||
|
||
self.assertEqual(
|
||
"生成自定义分类图片：结构展示。",
|
||
product_suite.category_description("结构展示"),
|
||
)
|
||
|
||
def test_product_suite_optional_item_and_reference_variables(self):
|
||
template = prompts.load_default_product_suite_prompt() + (
|
||
"\n商品ID：{商品ID}\n参考图序号：{主参考图序号}"
|
||
)
|
||
context = product_suite.product_suite_prompt_context(
|
||
"卖点",
|
||
product_suite.default_suite_settings(),
|
||
"卖点图",
|
||
"",
|
||
source_index=2,
|
||
)
|
||
|
||
rendered = product_suite.render_product_suite_prompt(template, context)
|
||
|
||
self.assertIn("商品ID：未绑定商品", rendered)
|
||
self.assertIn("参考图序号：2", rendered)
|
||
|
||
def test_business_prompt_braces_are_not_reparsed_as_template_variables(self):
|
||
template = prompts.load_default_product_suite_prompt()
|
||
business_prompt = "适合 {儿童} 使用，支持 {日常通勤} 场景"
|
||
|
||
rendered = product_suite.build_suite_prompt(
|
||
business_prompt,
|
||
product_suite.default_suite_settings(),
|
||
"白底图",
|
||
"51100639510",
|
||
template_text=template,
|
||
)
|
||
|
||
self.assertIn(business_prompt, rendered)
|
||
self.assertEqual([], product_suite.product_suite_prompt_errors(template))
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|