devops-pdt613/market-assistant
1
0
Fork 0
mirror of https://github.com/primedigitaltech/market-assistant.git synced 2026-07-25 01:34:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(pipeline): restore llm_keyword_suggest parse unit tests

Browse Source 
Made-with: Cursor
This commit is contained in:
hub-gif 2026-04-14 14:26:23 +08:00
parent 6f04326400
commit 6a9b6af321
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
backend/pipeline/tests/test_llm_keyword_suggest.py
View File

@ -59,6 +59,10 @@ class ParsePhrasesObjectTests(SimpleTestCase):
raw = '{"phrases": ["低糖", "口感好"]}' raw = '{"phrases": ["低糖", "口感好"]}'
self.assertEqual(_parse_phrases_object(raw), ["低糖", "口感好"]) self.assertEqual(_parse_phrases_object(raw), ["低糖", "口感好"])
def test_strips_phrase_whitespace(self) -> None:
raw = '{"phrases": ["口感", "回购 "]}'
self.assertEqual(_parse_phrases_object(raw), ["口感", "回购"])
def test_fenced_json(self) -> None: def test_fenced_json(self) -> None:
raw = '```json\n{"phrases": ["A", "B"]}\n```' raw = '```json\n{"phrases": ["A", "B"]}\n```'
self.assertEqual(_parse_phrases_object(raw), ["A", "B"]) self.assertEqual(_parse_phrases_object(raw), ["A", "B"])
@ -85,6 +89,13 @@ class ParseScenariosObjectTests(SimpleTestCase):
self.assertEqual(out[0]["label"], "A") self.assertEqual(out[0]["label"], "A")
self.assertEqual(out[0]["triggers"], ["触发甲", "触发乙"]) self.assertEqual(out[0]["triggers"], ["触发甲", "触发乙"])
def test_fenced_without_json_tag(self) -> None:
raw = '```\n{"scenarios": [{"label": "露营", "triggers": ["户外", "野餐"]}]}\n```'
out = _parse_scenarios_object(raw)
self.assertEqual(len(out), 1)
self.assertEqual(out[0]["label"], "露营")
self.assertEqual(out[0]["triggers"], ["户外", "野餐"])
class SuggestFocusKeywordsTests(SimpleTestCase): class SuggestFocusKeywordsTests(SimpleTestCase):
def test_no_comments_returns_empty(self) -> None: def test_no_comments_returns_empty(self) -> None: