kai/invoice-master-poc-v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

re-structure

Browse Source
This commit is contained in:
Yaojia Wang
2026-02-01 22:55:31 +01:00
parent 400b12a967
commit b602d0a340
176 changed files with 856 additions and 853 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/web/test_model_versions.py
View File

@@ -9,9 +9,9 @@ from uuid import UUID
import pytest
from inference.data.admin_models import ModelVersion
from inference.web.api.v1.admin.training import create_training_router
from inference.web.schemas.admin import (
from backend.data.admin_models import ModelVersion
from backend.web.api.v1.admin.training import create_training_router
from backend.web.schemas.admin import (
ModelVersionCreateRequest,
ModelVersionUpdateRequest,
)
@@ -173,13 +173,16 @@ class TestGetActiveModelRoute:
def test_get_active_model_when_exists(self, mock_models_repo):
fn = _find_endpoint("get_active_model")
mock_models_repo.get_active.return_value = _make_model_version(status="active", is_active=True)
mock_model = _make_model_version(status="active", is_active=True)
mock_models_repo.get_active.return_value = mock_model
result = asyncio.run(fn(admin_token=TEST_TOKEN, models=mock_models_repo))
assert result.has_active_model is True
assert result.model is not None
# model is now a ModelVersionItem, not ModelVersion
assert result.model.is_active is True
assert result.model.version == "1.0.0"
def test_get_active_model_when_none(self, mock_models_repo):
fn = _find_endpoint("get_active_model")