Spaces:
Running on Zero
Running on Zero
| import inspect | |
| import json | |
| import os | |
| import gradio as gr | |
| import pytest | |
| from config import GRADIO_VERSION, LANGUAGES, MODEL_REVISION, ROOT, SPACES_VERSION | |
| def test_gradio_api_and_native_progress_contract(monkeypatch): | |
| monkeypatch.setenv("ZONOS_UI_ONLY","1") | |
| import app | |
| assert gr.__version__==GRADIO_VERSION | |
| assert set(app.demo.get_api_info()["named_endpoints"])=={"/generate","/cancel"} | |
| event=next(x for x in app.demo.config['dependencies'] if x['api_name']=='generate') | |
| assert event['show_progress']!='hidden' | |
| assert len(event['inputs'])==13 and len(event['outputs'])==5 | |
| assert event['trigger_mode']=='once' | |
| assert 'watch' in (ROOT/'assets/player.js').read_text() | |
| assert 'js_on_load' in inspect.signature(gr.HTML).parameters | |
| def test_voice_catalog_matches_supported_languages(): | |
| c=json.loads((ROOT/'voices.json').read_text()) | |
| assert c['model_revision']==MODEL_REVISION | |
| assert len(c['voices'])==23 | |
| assert all(v['language'] in LANGUAGES for v in c['voices']) | |
| assert next(v for v in c['voices'] if v['name']=='Gaspard')['language']=='fr_fr' | |
| def test_upstream_callback_and_decoder_contract(): | |
| import engine | |
| from zonos3.reference.synth import synthesize | |
| from zonos3.tokenizer.matched_codec import registry | |
| assert {'on_plan','on_audio'} <= set(inspect.signature(synthesize).parameters) | |
| assert all(x['decoder_default']=='keep2_plain' for x in registry().values()) | |
| assert engine.ENGINE is None # imports alone must not download weights | |
| def test_all_runtime_download_helpers_receive_pin(monkeypatch): | |
| import engine | |
| from zonos3.utils import release | |
| import huggingface_hub | |
| calls=[] | |
| monkeypatch.setattr(huggingface_hub,'hf_hub_download',lambda **kwargs:(calls.append(kwargs) or '/cache/test')) | |
| monkeypatch.setattr(huggingface_hub,'snapshot_download',lambda **kwargs:(calls.append(kwargs) or '/cache/test')) | |
| release.resolve_file.cache_clear();release.resolve_dir_or_local.cache_clear() | |
| release.resolve_file('hf://Zyphra/ZONOS3-Mini/autoencoder/generator.pt') | |
| release.resolve_dir_or_local('hf://Zyphra/ZONOS3-Mini/lm/zonos3_mini') | |
| assert len(calls)==2 and all(c['revision']==MODEL_REVISION for c in calls) | |
| release.resolve_file.cache_clear();release.resolve_dir_or_local.cache_clear() | |
| def test_speaker_and_codec_import_with_resolved_dependencies(): | |
| import engine | |
| from zonos3.models.speaker_cloning import create_speaker_embedder | |
| from zonos3.models.mect_speaker import MECTFbankSpeakerEmbedding | |
| from zonos3.codec.istft_head_a_20261002.model import ISTFTCodec | |
| from zonos3.codec.istft_head_a_20261002.stream import ISTFTStream | |
| assert all(callable(x) for x in [create_speaker_embedder,MECTFbankSpeakerEmbedding,ISTFTCodec,ISTFTStream]) | |
| def test_gpu_reservation_fits_public_takes(): | |
| from engine import gpu_budget | |
| assert gpu_budget({"text":"A short take."}, None, None) == 30 | |
| assert gpu_budget({"text":"a" * 600}, None, None) == 58 | |
| assert gpu_budget({"text":"a" * 3000}, None, None) == 60 | |