45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
"""部分尺寸的尺寸标注图：仅传入已知方向，绝不补造缺失方向。"""
|
||
from __future__ import annotations
|
||
|
||
import sys
|
||
import unittest
|
||
from pathlib import Path
|
||
|
||
SERVER_DIR = Path(__file__).resolve().parents[1]
|
||
if str(SERVER_DIR) not in sys.path:
|
||
sys.path.insert(0, str(SERVER_DIR))
|
||
|
||
from api.suite import _with_dimensions
|
||
from schemas.suite import ProductDimensions
|
||
from services.prompts.common import build_context
|
||
from services.prompts.gpt import _task_size_chart
|
||
|
||
|
||
class PartialDimensionsTests(unittest.TestCase):
|
||
def test_only_known_direction_is_passed_to_prompt_context(self) -> None:
|
||
raw = _with_dimensions(
|
||
{"title": "测试商品", "params": [
|
||
{"key": "产品尺寸", "value": "12.5×8×4 cm"},
|
||
{"key": "材质", "value": "棉"},
|
||
]},
|
||
ProductDimensions(width=8, height=4, unit="cm"),
|
||
required=True,
|
||
)
|
||
|
||
self.assertEqual(raw["dimensions"], {"width": 8, "height": 4, "unit": "cm"})
|
||
self.assertEqual(raw["params"], [{"key": "材质", "value": "棉"}])
|
||
context = build_context(raw)
|
||
prompt = _task_size_chart(context, "en")
|
||
self.assertIn("width: 8.0 cm; height: 4.0 cm", prompt)
|
||
self.assertIn("render exactly 2 measurement arrows and labels, only for: width, height", prompt)
|
||
self.assertIn("Do not draw, label, or show an arrow for: length", prompt)
|
||
self.assertNotIn("12.5×8×4", prompt)
|
||
|
||
def test_size_chart_still_requires_at_least_one_dimension(self) -> None:
|
||
with self.assertRaisesRegex(Exception, "至少需要填写一项"):
|
||
_with_dimensions({"title": "测试商品"}, ProductDimensions(unit="mm"), required=True)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|