docs: 为全部代码补充中文注释并加入 AGENTS 注释规范

This commit is contained in:
cat-shark
2026-08-13 22:09:55 +08:00
parent 812d276998
commit c1a8354282
5 changed files with 55 additions and 1 deletions
+21
View File
@@ -1,3 +1,9 @@
"""faster-whisper 转写节点测试。
通过注入假 faster_whisper 模块覆盖时间戳格式化、参数传递、SRT 生成、
异常处理和入口点启动等真实代码路径。
"""
import runpy
import sys
import types
@@ -8,6 +14,8 @@ from wov_sdk.models import InvokeRequest
class FakeSegment:
"""模拟 faster-whisper 的分段对象,只提供转写测试需要的字段。"""
def __init__(self, start, end, text):
self.start = start
self.end = end
@@ -15,14 +23,18 @@ class FakeSegment:
class FakeWhisperModel:
"""记录构造参数并返回固定分段的假 WhisperModel。"""
instances: list[tuple[tuple, dict]] = []
def __init__(self, *args, **kwargs):
# 记录每次构造参数,测试据此断言 device/compute_type 传递。
FakeWhisperModel.instances.append((args, kwargs))
self.args = args
self.kwargs = kwargs
def transcribe(self, path, **kwargs):
# 返回固定两个分段:一个普通时长,一个跨小时验证时间戳格式。
self.transcribe_args = (path, kwargs)
return (
[
@@ -34,11 +46,13 @@ class FakeWhisperModel:
def _install_fake_whisper(monkeypatch, model_class=FakeWhisperModel) -> None:
"""把假 faster_whisper 模块注入 sys.modules,替代真实依赖。"""
fake_module = types.SimpleNamespace(WhisperModel=model_class)
monkeypatch.setitem(sys.modules, "faster_whisper", fake_module)
def _request(tmp_path, **overrides) -> InvokeRequest:
"""构造默认音频输入与日语参数的调用请求。"""
payload = {
"run_id": "run_1",
"node_instance_id": "ni_1",
@@ -51,12 +65,14 @@ def _request(tmp_path, **overrides) -> InvokeRequest:
def test_format_timestamp() -> None:
"""验证秒数到 SRT 时间戳的格式化结果。"""
assert format_timestamp(0) == "00:00:00,000"
assert format_timestamp(3600.5) == "01:00:00,500"
assert format_timestamp(61.25) == "00:01:01,250"
def test_success(tmp_path, monkeypatch) -> None:
"""验证成功转写会生成 SRT 并默认使用 auto 设备/计算类型。"""
FakeWhisperModel.instances.clear()
_install_fake_whisper(monkeypatch)
(tmp_path / "audio.wav").write_bytes(b"fake")
@@ -71,6 +87,7 @@ def test_success(tmp_path, monkeypatch) -> None:
def test_compute_type_override(tmp_path, monkeypatch) -> None:
"""验证请求参数可以覆盖默认计算类型。"""
FakeWhisperModel.instances.clear()
_install_fake_whisper(monkeypatch)
(tmp_path / "audio.wav").write_bytes(b"fake")
@@ -81,6 +98,7 @@ def test_compute_type_override(tmp_path, monkeypatch) -> None:
def test_model_raises(tmp_path, monkeypatch) -> None:
"""验证模型加载失败时返回 failed 与错误信息。"""
class BrokenModel:
def __init__(self, *args, **kwargs):
raise RuntimeError("model load failed")
@@ -93,17 +111,20 @@ def test_model_raises(tmp_path, monkeypatch) -> None:
def test_missing_input(tmp_path) -> None:
"""验证缺少 audio_uri 时返回失败。"""
response = invoke(_request(tmp_path, inputs={}))
assert response.status == "failed"
def test_missing_file(tmp_path) -> None:
"""验证音频文件不存在时返回失败。"""
response = invoke(_request(tmp_path))
assert response.status == "failed"
assert "audio file not found" in response.error
def test_entrypoint(monkeypatch) -> None:
"""验证 python -m wov_node_whisper 会加载 faster-whisper manifest。"""
module_path = Path(__file__).resolve().parent.parent / "wov_node_whisper" / "__main__.py"
captured = {}