test(13-01): add failing tests for video type detection

- 创建 conftest.py 共享 fixtures（MockAIClient, sample_segments） - 创建 test_semantic_engine.py 包含 8 个测试用例 - test_video_type_detection_from_title: 测试从标题推断类型 - test_video_type_detection_from_category: 测试从分类推断类型 - test_video_type_fallback_to_default: 测试默认类型回退 - test_video_type_in_prompt: 测试类型出现在 Prompt 中 - test_prompt_format: 测试 Prompt 格式 - test_video_segment_model: 测试 VideoSegment 模型 - test_segmentation_result_model: 测试 SegmentationResult 模型 - test_model_validation: 测试字段验证 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-22 00:23:40 +08:00
parent 47cd84b30a
commit 9862da7085
2 changed files with 191 additions and 0 deletions
@@ -0,0 +1,62 @@
+"""共享测试 fixtures 和配置"""
+
+from typing import Optional
+from unittest.mock import Mock
+
+import pytest
+
+
+class MockAIClient:
+    """模拟 AI 客户端"""
+
+    def __init__(self, response: str = "Mock AI response"):
+        self.response = response
+
+    def generate(
+        self,
+        prompt: str,
+        max_tokens: int = 16000,
+        disable_reasoning: Optional[bool] = None
+    ) -> str:
+        """生成模拟响应"""
+        return self.response
+
+
+@pytest.fixture
+def mock_ai_client():
+    """提供模拟 AI 客户端"""
+    return MockAIClient()
+
+
+@pytest.fixture
+def sample_segments():
+    """提供示例字幕分段数据"""
+    return [
+        {"index": 1, "start": "00:00:00,000", "end": "00:00:05,000", "text": "大家好，欢迎来到我的频道"},
+        {"index": 2, "start": "00:00:05,000", "end": "00:00:10,000", "text": "今天我们要讲的是机器学习的基础知识"},
+        {"index": 3, "start": "00:00:10,000", "end": "00:00:15,000", "text": "首先，什么是机器学习"},
+        {"index": 4, "start": "00:00:15,000", "end": "00:00:20,000", "text": "机器学习是人工智能的一个分支"},
+        {"index": 5, "start": "00:00:20,000", "end": "00:00:25,000", "text": "它让计算机能够从数据中学习"},
+    ]
+
+
+@pytest.fixture
+def sample_metadata():
+    """提供示例视频元数据"""
+    return {
+        "title": "机器学习入门教程",
+        "category": "教育",
+        "tags": ["编程", "AI", "教程"],
+        "description": "从零开始学习机器学习"
+    }
+
+
+@pytest.fixture
+def sample_metadata_movie():
+    """提供电影类型示例元数据"""
+    return {
+        "title": "星际穿越",
+        "category": "影视",
+        "tags": ["科幻", "电影"],
+        "description": "一部关于太空探索的科幻电影"
+    }