re-structure
This commit is contained in:
Yaojia Wang
@@ -1,33 +1,33 @@
|
||||
"""
|
||||
Tests to verify admin schemas split maintains backward compatibility.
|
||||
|
||||
All existing imports from inference.web.schemas.admin must continue to work.
|
||||
All existing imports from backend.web.schemas.admin must continue to work.
|
||||
"""
|
||||
|
||||
import pytest
|
||||
|
||||
|
||||
class TestEnumImports:
|
||||
"""All enums importable from inference.web.schemas.admin."""
|
||||
"""All enums importable from backend.web.schemas.admin."""
|
||||
|
||||
def test_document_status(self):
|
||||
from inference.web.schemas.admin import DocumentStatus
|
||||
from backend.web.schemas.admin import DocumentStatus
|
||||
assert DocumentStatus.PENDING == "pending"
|
||||
|
||||
def test_auto_label_status(self):
|
||||
from inference.web.schemas.admin import AutoLabelStatus
|
||||
from backend.web.schemas.admin import AutoLabelStatus
|
||||
assert AutoLabelStatus.RUNNING == "running"
|
||||
|
||||
def test_training_status(self):
|
||||
from inference.web.schemas.admin import TrainingStatus
|
||||
from backend.web.schemas.admin import TrainingStatus
|
||||
assert TrainingStatus.PENDING == "pending"
|
||||
|
||||
def test_training_type(self):
|
||||
from inference.web.schemas.admin import TrainingType
|
||||
from backend.web.schemas.admin import TrainingType
|
||||
assert TrainingType.TRAIN == "train"
|
||||
|
||||
def test_annotation_source(self):
|
||||
from inference.web.schemas.admin import AnnotationSource
|
||||
from backend.web.schemas.admin import AnnotationSource
|
||||
assert AnnotationSource.MANUAL == "manual"
|
||||
|
||||
|
||||
@@ -35,12 +35,12 @@ class TestAuthImports:
|
||||
"""Auth schemas importable."""
|
||||
|
||||
def test_admin_token_create(self):
|
||||
from inference.web.schemas.admin import AdminTokenCreate
|
||||
from backend.web.schemas.admin import AdminTokenCreate
|
||||
token = AdminTokenCreate(name="test")
|
||||
assert token.name == "test"
|
||||
|
||||
def test_admin_token_response(self):
|
||||
from inference.web.schemas.admin import AdminTokenResponse
|
||||
from backend.web.schemas.admin import AdminTokenResponse
|
||||
assert AdminTokenResponse is not None
|
||||
|
||||
|
||||
@@ -48,23 +48,23 @@ class TestDocumentImports:
|
||||
"""Document schemas importable."""
|
||||
|
||||
def test_document_upload_response(self):
|
||||
from inference.web.schemas.admin import DocumentUploadResponse
|
||||
from backend.web.schemas.admin import DocumentUploadResponse
|
||||
assert DocumentUploadResponse is not None
|
||||
|
||||
def test_document_item(self):
|
||||
from inference.web.schemas.admin import DocumentItem
|
||||
from backend.web.schemas.admin import DocumentItem
|
||||
assert DocumentItem is not None
|
||||
|
||||
def test_document_list_response(self):
|
||||
from inference.web.schemas.admin import DocumentListResponse
|
||||
from backend.web.schemas.admin import DocumentListResponse
|
||||
assert DocumentListResponse is not None
|
||||
|
||||
def test_document_detail_response(self):
|
||||
from inference.web.schemas.admin import DocumentDetailResponse
|
||||
from backend.web.schemas.admin import DocumentDetailResponse
|
||||
assert DocumentDetailResponse is not None
|
||||
|
||||
def test_document_stats_response(self):
|
||||
from inference.web.schemas.admin import DocumentStatsResponse
|
||||
from backend.web.schemas.admin import DocumentStatsResponse
|
||||
assert DocumentStatsResponse is not None
|
||||
|
||||
|
||||
@@ -72,60 +72,60 @@ class TestAnnotationImports:
|
||||
"""Annotation schemas importable."""
|
||||
|
||||
def test_bounding_box(self):
|
||||
from inference.web.schemas.admin import BoundingBox
|
||||
from backend.web.schemas.admin import BoundingBox
|
||||
bb = BoundingBox(x=0, y=0, width=100, height=50)
|
||||
assert bb.width == 100
|
||||
|
||||
def test_annotation_create(self):
|
||||
from inference.web.schemas.admin import AnnotationCreate
|
||||
from backend.web.schemas.admin import AnnotationCreate
|
||||
assert AnnotationCreate is not None
|
||||
|
||||
def test_annotation_update(self):
|
||||
from inference.web.schemas.admin import AnnotationUpdate
|
||||
from backend.web.schemas.admin import AnnotationUpdate
|
||||
assert AnnotationUpdate is not None
|
||||
|
||||
def test_annotation_item(self):
|
||||
from inference.web.schemas.admin import AnnotationItem
|
||||
from backend.web.schemas.admin import AnnotationItem
|
||||
assert AnnotationItem is not None
|
||||
|
||||
def test_annotation_response(self):
|
||||
from inference.web.schemas.admin import AnnotationResponse
|
||||
from backend.web.schemas.admin import AnnotationResponse
|
||||
assert AnnotationResponse is not None
|
||||
|
||||
def test_annotation_list_response(self):
|
||||
from inference.web.schemas.admin import AnnotationListResponse
|
||||
from backend.web.schemas.admin import AnnotationListResponse
|
||||
assert AnnotationListResponse is not None
|
||||
|
||||
def test_annotation_lock_request(self):
|
||||
from inference.web.schemas.admin import AnnotationLockRequest
|
||||
from backend.web.schemas.admin import AnnotationLockRequest
|
||||
assert AnnotationLockRequest is not None
|
||||
|
||||
def test_annotation_lock_response(self):
|
||||
from inference.web.schemas.admin import AnnotationLockResponse
|
||||
from backend.web.schemas.admin import AnnotationLockResponse
|
||||
assert AnnotationLockResponse is not None
|
||||
|
||||
def test_auto_label_request(self):
|
||||
from inference.web.schemas.admin import AutoLabelRequest
|
||||
from backend.web.schemas.admin import AutoLabelRequest
|
||||
assert AutoLabelRequest is not None
|
||||
|
||||
def test_auto_label_response(self):
|
||||
from inference.web.schemas.admin import AutoLabelResponse
|
||||
from backend.web.schemas.admin import AutoLabelResponse
|
||||
assert AutoLabelResponse is not None
|
||||
|
||||
def test_annotation_verify_request(self):
|
||||
from inference.web.schemas.admin import AnnotationVerifyRequest
|
||||
from backend.web.schemas.admin import AnnotationVerifyRequest
|
||||
assert AnnotationVerifyRequest is not None
|
||||
|
||||
def test_annotation_verify_response(self):
|
||||
from inference.web.schemas.admin import AnnotationVerifyResponse
|
||||
from backend.web.schemas.admin import AnnotationVerifyResponse
|
||||
assert AnnotationVerifyResponse is not None
|
||||
|
||||
def test_annotation_override_request(self):
|
||||
from inference.web.schemas.admin import AnnotationOverrideRequest
|
||||
from backend.web.schemas.admin import AnnotationOverrideRequest
|
||||
assert AnnotationOverrideRequest is not None
|
||||
|
||||
def test_annotation_override_response(self):
|
||||
from inference.web.schemas.admin import AnnotationOverrideResponse
|
||||
from backend.web.schemas.admin import AnnotationOverrideResponse
|
||||
assert AnnotationOverrideResponse is not None
|
||||
|
||||
|
||||
@@ -133,68 +133,68 @@ class TestTrainingImports:
|
||||
"""Training schemas importable."""
|
||||
|
||||
def test_training_config(self):
|
||||
from inference.web.schemas.admin import TrainingConfig
|
||||
from backend.web.schemas.admin import TrainingConfig
|
||||
config = TrainingConfig()
|
||||
assert config.epochs == 100
|
||||
|
||||
def test_training_task_create(self):
|
||||
from inference.web.schemas.admin import TrainingTaskCreate
|
||||
from backend.web.schemas.admin import TrainingTaskCreate
|
||||
assert TrainingTaskCreate is not None
|
||||
|
||||
def test_training_task_item(self):
|
||||
from inference.web.schemas.admin import TrainingTaskItem
|
||||
from backend.web.schemas.admin import TrainingTaskItem
|
||||
assert TrainingTaskItem is not None
|
||||
|
||||
def test_training_task_list_response(self):
|
||||
from inference.web.schemas.admin import TrainingTaskListResponse
|
||||
from backend.web.schemas.admin import TrainingTaskListResponse
|
||||
assert TrainingTaskListResponse is not None
|
||||
|
||||
def test_training_task_detail_response(self):
|
||||
from inference.web.schemas.admin import TrainingTaskDetailResponse
|
||||
from backend.web.schemas.admin import TrainingTaskDetailResponse
|
||||
assert TrainingTaskDetailResponse is not None
|
||||
|
||||
def test_training_task_response(self):
|
||||
from inference.web.schemas.admin import TrainingTaskResponse
|
||||
from backend.web.schemas.admin import TrainingTaskResponse
|
||||
assert TrainingTaskResponse is not None
|
||||
|
||||
def test_training_log_item(self):
|
||||
from inference.web.schemas.admin import TrainingLogItem
|
||||
from backend.web.schemas.admin import TrainingLogItem
|
||||
assert TrainingLogItem is not None
|
||||
|
||||
def test_training_logs_response(self):
|
||||
from inference.web.schemas.admin import TrainingLogsResponse
|
||||
from backend.web.schemas.admin import TrainingLogsResponse
|
||||
assert TrainingLogsResponse is not None
|
||||
|
||||
def test_export_request(self):
|
||||
from inference.web.schemas.admin import ExportRequest
|
||||
from backend.web.schemas.admin import ExportRequest
|
||||
assert ExportRequest is not None
|
||||
|
||||
def test_export_response(self):
|
||||
from inference.web.schemas.admin import ExportResponse
|
||||
from backend.web.schemas.admin import ExportResponse
|
||||
assert ExportResponse is not None
|
||||
|
||||
def test_training_document_item(self):
|
||||
from inference.web.schemas.admin import TrainingDocumentItem
|
||||
from backend.web.schemas.admin import TrainingDocumentItem
|
||||
assert TrainingDocumentItem is not None
|
||||
|
||||
def test_training_documents_response(self):
|
||||
from inference.web.schemas.admin import TrainingDocumentsResponse
|
||||
from backend.web.schemas.admin import TrainingDocumentsResponse
|
||||
assert TrainingDocumentsResponse is not None
|
||||
|
||||
def test_model_metrics(self):
|
||||
from inference.web.schemas.admin import ModelMetrics
|
||||
from backend.web.schemas.admin import ModelMetrics
|
||||
assert ModelMetrics is not None
|
||||
|
||||
def test_training_model_item(self):
|
||||
from inference.web.schemas.admin import TrainingModelItem
|
||||
from backend.web.schemas.admin import TrainingModelItem
|
||||
assert TrainingModelItem is not None
|
||||
|
||||
def test_training_models_response(self):
|
||||
from inference.web.schemas.admin import TrainingModelsResponse
|
||||
from backend.web.schemas.admin import TrainingModelsResponse
|
||||
assert TrainingModelsResponse is not None
|
||||
|
||||
def test_training_history_item(self):
|
||||
from inference.web.schemas.admin import TrainingHistoryItem
|
||||
from backend.web.schemas.admin import TrainingHistoryItem
|
||||
assert TrainingHistoryItem is not None
|
||||
|
||||
|
||||
@@ -202,31 +202,31 @@ class TestDatasetImports:
|
||||
"""Dataset schemas importable."""
|
||||
|
||||
def test_dataset_create_request(self):
|
||||
from inference.web.schemas.admin import DatasetCreateRequest
|
||||
from backend.web.schemas.admin import DatasetCreateRequest
|
||||
assert DatasetCreateRequest is not None
|
||||
|
||||
def test_dataset_document_item(self):
|
||||
from inference.web.schemas.admin import DatasetDocumentItem
|
||||
from backend.web.schemas.admin import DatasetDocumentItem
|
||||
assert DatasetDocumentItem is not None
|
||||
|
||||
def test_dataset_response(self):
|
||||
from inference.web.schemas.admin import DatasetResponse
|
||||
from backend.web.schemas.admin import DatasetResponse
|
||||
assert DatasetResponse is not None
|
||||
|
||||
def test_dataset_detail_response(self):
|
||||
from inference.web.schemas.admin import DatasetDetailResponse
|
||||
from backend.web.schemas.admin import DatasetDetailResponse
|
||||
assert DatasetDetailResponse is not None
|
||||
|
||||
def test_dataset_list_item(self):
|
||||
from inference.web.schemas.admin import DatasetListItem
|
||||
from backend.web.schemas.admin import DatasetListItem
|
||||
assert DatasetListItem is not None
|
||||
|
||||
def test_dataset_list_response(self):
|
||||
from inference.web.schemas.admin import DatasetListResponse
|
||||
from backend.web.schemas.admin import DatasetListResponse
|
||||
assert DatasetListResponse is not None
|
||||
|
||||
def test_dataset_train_request(self):
|
||||
from inference.web.schemas.admin import DatasetTrainRequest
|
||||
from backend.web.schemas.admin import DatasetTrainRequest
|
||||
assert DatasetTrainRequest is not None
|
||||
|
||||
|
||||
@@ -234,12 +234,12 @@ class TestForwardReferences:
|
||||
"""Forward references resolve correctly."""
|
||||
|
||||
def test_document_detail_has_annotation_items(self):
|
||||
from inference.web.schemas.admin import DocumentDetailResponse
|
||||
from backend.web.schemas.admin import DocumentDetailResponse
|
||||
fields = DocumentDetailResponse.model_fields
|
||||
assert "annotations" in fields
|
||||
assert "training_history" in fields
|
||||
|
||||
def test_dataset_train_request_has_config(self):
|
||||
from inference.web.schemas.admin import DatasetTrainRequest, TrainingConfig
|
||||
from backend.web.schemas.admin import DatasetTrainRequest, TrainingConfig
|
||||
req = DatasetTrainRequest(name="test", config=TrainingConfig())
|
||||
assert req.config.epochs == 100
|
||||
|
||||
Reference in New Issue
Block a user