From a45bcf0aa9d92a4190523429048f2b715306ca8a Mon Sep 17 00:00:00 2001 From: CI Bot Date: Sun, 4 Oct 2026 09:43:40 +0000 Subject: [PATCH] style: auto-format with black + isort + ruff + prettier [skip ci-format-check] --- tests/unit/test_viral_video.py | 2 +- tests/unit/test_viral_video_wiring.py | 50 +++++++++++++++++---------- 2 files changed, 33 insertions(+), 19 deletions(-) diff --git a/tests/unit/test_viral_video.py b/tests/unit/test_viral_video.py index 5e915d568..7d5ba2845 100755 --- a/tests/unit/test_viral_video.py +++ b/tests/unit/test_viral_video.py @@ -453,7 +453,7 @@ class TestViralVideoPipeline: def test_review_pass_v16(self, mock_review, mock_job): """v1.6 _step_review 接收 copy_result dict。""" from apps.worker.worker_app.tasks.viral_video import _step_review - from packages.application.viral_video.reviewer import ReviewResult, ReviewIssue + from packages.application.viral_video.reviewer import ReviewIssue, ReviewResult mock_review.return_value = ReviewResult(passed=True, score=90, issues=[], rewrite_suggestions=[]) cr = {"voiceover_script": "大家好", "shots": []} diff --git a/tests/unit/test_viral_video_wiring.py b/tests/unit/test_viral_video_wiring.py index e72a8822a..fe5ada00e 100644 --- a/tests/unit/test_viral_video_wiring.py +++ b/tests/unit/test_viral_video_wiring.py @@ -87,6 +87,7 @@ STORYBOARD_XML = """ @pytest.fixture(autouse=True) def invalidate_loader_cache(): from packages.application.viral_video import prompt_loader as pl + pl.invalidate() yield pl.invalidate() @@ -120,8 +121,7 @@ class TestIntentParsingWiring: def test_uses_loader_and_parses_xml(self, job): from apps.worker.worker_app.tasks import viral_video as vv - img_result = {"products": [{"name": "lipstick", "brand": "品牌X", - "key_features": ["显白", "持久"]}]} + img_result = {"products": [{"name": "lipstick", "brand": "品牌X", "key_features": ["显白", "持久"]}]} with patch("packages.shared.ai_service.call_llm", return_value=INTENT_XML) as mock_llm: result = vv._step_intent_parsing(job, img_result) @@ -154,8 +154,7 @@ class TestScriptGenerationWiring: result = vv._step_script_generation(job, intent, {}) # fusion_level 对应的指令文本被注入到 system prompt 中 - assert FUSION_INSTRUCTIONS[level] in captured_system["final"], \ - f"fusion_level {level} 指令未注入 system_prompt" + assert FUSION_INSTRUCTIONS[level] in captured_system["final"], f"fusion_level {level} 指令未注入 system_prompt" # 输出保持 Seedance 兼容结构 assert "overview" in result assert "shots" in result @@ -201,7 +200,7 @@ class TestReviewWiring: """审核不通过时触发自动重写，并更新 job.copy_result""" from apps.worker.worker_app.tasks import viral_video as vv from packages.application.viral_video.reviewer import Reviewer, ReviewResult - from packages.application.viral_video.schemas import FusionResult, ScriptSegment, ReviewIssue + from packages.application.viral_video.schemas import FusionResult, ReviewIssue, ScriptSegment copy_result = { "title": "原标题", @@ -212,19 +211,25 @@ class TestReviewWiring: job.intent_result = {"key_messages": ["显白"], "intent": "推广"} fail_result = ReviewResult( - passed=False, score=50, + passed=False, + score=50, issues=[ReviewIssue(dimension="违规词", severity="high", location="开头", text="绝了")], rewrite_suggestions=["去掉夸大词"], ) rewritten = FusionResult( - title="新标题", hook="修改后钩子", - script_segments=[ScriptSegment(text="修改后口播正文")], cta="行动号召", - word_count=10, estimated_duration=10, + title="新标题", + hook="修改后钩子", + script_segments=[ScriptSegment(text="修改后口播正文")], + cta="行动号召", + word_count=10, + estimated_duration=10, ) pass_after = ReviewResult(passed=True, score=88, issues=[], rewrite_suggestions=[]) - with patch.object(Reviewer, "review", side_effect=[fail_result, pass_after]), \ - patch.object(Reviewer, "rewrite", return_value=rewritten): + with ( + patch.object(Reviewer, "review", side_effect=[fail_result, pass_after]), + patch.object(Reviewer, "rewrite", return_value=rewritten), + ): out = vv._step_review(job, copy_result) assert out["passed"] is True @@ -247,9 +252,11 @@ class TestEndToEndLoaderUsed: called_types.append(prompt_type) return real_get(prompt_type, **kwargs) - with patch.object(pl, "get_template", side_effect=spy_get), \ - patch("packages.shared.ai_service.call_vision", return_value=IMAGE_XML), \ - patch("packages.shared.ai_service.call_llm", return_value=INTENT_XML): + with ( + patch.object(pl, "get_template", side_effect=spy_get), + patch("packages.shared.ai_service.call_vision", return_value=IMAGE_XML), + patch("packages.shared.ai_service.call_llm", return_value=INTENT_XML), + ): # 1) image img_res = vv._analyze_single_image(0, "https://img/1.jpg", "vlm", 15) # 2) intent @@ -261,26 +268,33 @@ class TestEndToEndLoaderUsed: # script 和 review 单独验证（需要不同的 LLM 返回） called_types_2 = [] + def spy_get_2(prompt_type, **kwargs): called_types_2.append(prompt_type) return real_get(prompt_type, **kwargs) - with patch.object(pl, "get_template", side_effect=spy_get_2), \ - patch("packages.shared.ai_service.call_llm", return_value=STORYBOARD_XML): + with ( + patch.object(pl, "get_template", side_effect=spy_get_2), + patch("packages.shared.ai_service.call_llm", return_value=STORYBOARD_XML), + ): copy_res = vv._step_script_generation(job, intent_res, {"products": [img_res]}) assert "storyboard" in called_types_2 called_types_3 = [] + def spy_get_3(prompt_type, **kwargs): called_types_3.append(prompt_type) return real_get(prompt_type, **kwargs) from packages.application.viral_video.reviewer import Reviewer, ReviewResult + pass_result = ReviewResult(passed=True, score=90, issues=[], rewrite_suggestions=[]) job.intent_result = intent_res job.copy_result = copy_res - with patch.object(pl, "get_template", side_effect=spy_get_3), \ - patch.object(Reviewer, "review", return_value=pass_result) as mock_review: + with ( + patch.object(pl, "get_template", side_effect=spy_get_3), + patch.object(Reviewer, "review", return_value=pass_result) as mock_review, + ): review_res = vv._step_review(job, copy_res) # review 步骤内部直接调用 Reviewer.review，该方法被 mock，因此 get_template 不会被调用； # 此处验证 Reviewer.review 被调用即可说明 review 步骤走通了。