| import importlib |
| import sys |
| from types import SimpleNamespace |
|
|
|
|
| def test_omnivoice_adapter_registers_zero_gpu_decorator(monkeypatch) -> None: |
| decorated = {} |
|
|
| def fake_gpu(fn=None, **kwargs): |
| def decorate(inner): |
| decorated["name"] = inner.__name__ |
| decorated["kwargs"] = kwargs |
| setattr(inner, "_zero_gpu_wrapped", True) |
| return inner |
|
|
| if fn is not None: |
| return decorate(fn) |
| return decorate |
|
|
| monkeypatch.setitem(sys.modules, "spaces", SimpleNamespace(GPU=fake_gpu)) |
| sys.modules.pop("backend.omnivoice_adapter", None) |
|
|
| module = importlib.import_module("backend.omnivoice_adapter") |
|
|
| assert decorated == {"name": "synthesize", "kwargs": {"duration": 300}} |
| assert getattr(module.OmniVoiceAdapter.synthesize, "_zero_gpu_wrapped", False) is True |
|
|
