Spaces:
Sleeping
Sleeping
| from __future__ import annotations | |
| from pathlib import Path | |
| from typing import cast | |
| import gradio as gr | |
| from src.ui import runtime | |
| def test_embedding_warmup_is_disabled_by_default(monkeypatch) -> None: | |
| monkeypatch.setattr(runtime.settings, "embedding_warmup_enabled", False) | |
| monkeypatch.setattr( | |
| runtime, | |
| "warm_up_embeddings", | |
| lambda: (_ for _ in ()).throw(AssertionError("unexpected warmup")), | |
| ) | |
| assert runtime.warm_up_if_enabled() is False | |
| def test_embedding_warmup_runs_when_enabled(monkeypatch) -> None: | |
| monkeypatch.setattr(runtime.settings, "embedding_warmup_enabled", True) | |
| monkeypatch.setattr(runtime, "warm_up_embeddings", lambda: 12.5) | |
| assert runtime.warm_up_if_enabled() is True | |
| def test_embedding_warmup_failure_does_not_block_startup(monkeypatch) -> None: | |
| monkeypatch.setattr(runtime.settings, "embedding_warmup_enabled", True) | |
| monkeypatch.setattr( | |
| runtime, | |
| "warm_up_embeddings", | |
| lambda: (_ for _ in ()).throw(RuntimeError("model unavailable")), | |
| ) | |
| assert runtime.warm_up_if_enabled() is False | |
| def test_launch_preserves_the_app_with_custom_routes(monkeypatch, tmp_path: Path) -> None: | |
| class DemoStub: | |
| def __init__(self) -> None: | |
| self.app = object() | |
| self.queued_app = object() | |
| self.launch_kwargs: dict[str, object] = {} | |
| def queue(self, **_: object) -> DemoStub: | |
| self.app = self.queued_app | |
| return self | |
| def launch(self, **kwargs: object) -> None: | |
| self.launch_kwargs = kwargs | |
| demo = DemoStub() | |
| routed_apps: list[object] = [] | |
| monkeypatch.setattr(runtime.settings, "export_dir", tmp_path) | |
| monkeypatch.setattr(runtime, "warm_up_if_enabled", lambda: False) | |
| monkeypatch.setattr(runtime, "add_monitoring_route", routed_apps.append) | |
| runtime.launch_demo(cast(gr.Blocks, demo)) | |
| assert routed_apps == [demo.queued_app] | |
| assert demo.launch_kwargs["_app"] is demo.queued_app | |