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

8
tests/web/test_augmentation_routes.py
View File

@@ -10,8 +10,8 @@ from fastapi import FastAPI
from fastapi.testclient import TestClient
import numpy as np
from inference.web.api.v1.admin.augmentation import create_augmentation_router
from inference.web.core.auth import (
from backend.web.api.v1.admin.augmentation import create_augmentation_router
from backend.web.core.auth import (
validate_admin_token,
get_document_repository,
get_dataset_repository,
@@ -175,7 +175,7 @@ class TestAugmentationPreviewEndpoint:
fake_image = np.random.randint(0, 255, (100, 100, 3), dtype=np.uint8)
with patch(
"inference.web.services.augmentation_service.AugmentationService._load_document_page"
"backend.web.services.augmentation_service.AugmentationService._load_document_page"
) as mock_load:
mock_load.return_value = fake_image
@@ -251,7 +251,7 @@ class TestAugmentationPreviewConfigEndpoint:
fake_image = np.random.randint(0, 255, (100, 100, 3), dtype=np.uint8)
with patch(
"inference.web.services.augmentation_service.AugmentationService._load_document_page"
"backend.web.services.augmentation_service.AugmentationService._load_document_page"
) as mock_load:
mock_load.return_value = fake_image