re-structure

tests/web/test_document_category.py

@@ -10,7 +10,7 @@ from datetime import datetime
 from unittest.mock import MagicMock
 from uuid import UUID, uuid4
 
-from inference.data.admin_models import AdminDocument
+from backend.data.admin_models import AdminDocument
 
 
 # Test constants
@@ -76,7 +76,7 @@ class TestDocumentCategoryInReadModel:
 
     def test_admin_document_read_has_category(self):
         """Test AdminDocumentRead includes category field."""
-        from inference.data.admin_models import AdminDocumentRead
+        from backend.data.admin_models import AdminDocumentRead
 
         # Check the model has category field in its schema
         assert "category" in AdminDocumentRead.model_fields
@@ -94,7 +94,7 @@ class TestDocumentCategoryAPI:
 
     def test_upload_document_with_category(self, mock_admin_db):
         """Test uploading document with category parameter."""
-        from inference.web.schemas.admin import DocumentUploadResponse
+        from backend.web.schemas.admin import DocumentUploadResponse
 
         # Verify response schema supports category
         response = DocumentUploadResponse(
@@ -110,7 +110,7 @@ class TestDocumentCategoryAPI:
 
     def test_list_documents_returns_category(self, mock_admin_db):
         """Test list documents endpoint returns category."""
-        from inference.web.schemas.admin import DocumentItem
+        from backend.web.schemas.admin import DocumentItem
 
         item = DocumentItem(
             document_id=TEST_DOC_UUID,
@@ -127,7 +127,7 @@ class TestDocumentCategoryAPI:
 
     def test_document_detail_includes_category(self, mock_admin_db):
         """Test document detail response includes category."""
-        from inference.web.schemas.admin import DocumentDetailResponse
+        from backend.web.schemas.admin import DocumentDetailResponse
 
         # Check schema has category
         assert "category" in DocumentDetailResponse.model_fields
@@ -167,14 +167,14 @@ class TestDocumentCategoryUpdate:
 
     def test_update_document_category_schema(self):
         """Test update document request supports category."""
-        from inference.web.schemas.admin import DocumentUpdateRequest
+        from backend.web.schemas.admin import DocumentUpdateRequest
 
         request = DocumentUpdateRequest(category="letter")
         assert request.category == "letter"
 
     def test_update_document_category_optional(self):
         """Test category is optional in update request."""
-        from inference.web.schemas.admin import DocumentUpdateRequest
+        from backend.web.schemas.admin import DocumentUpdateRequest
 
         # Should not raise - category is optional
         request = DocumentUpdateRequest()
@@ -186,7 +186,7 @@ class TestDatasetWithCategory:
 
     def test_dataset_create_with_category_filter(self):
         """Test creating dataset can filter by document category."""
-        from inference.web.schemas.admin import DatasetCreateRequest
+        from backend.web.schemas.admin import DatasetCreateRequest
 
         request = DatasetCreateRequest(
             name="Invoice Training Set",
@@ -197,7 +197,7 @@ class TestDatasetWithCategory:
 
     def test_dataset_create_category_is_optional(self):
         """Test category filter is optional when creating dataset."""
-        from inference.web.schemas.admin import DatasetCreateRequest
+        from backend.web.schemas.admin import DatasetCreateRequest
 
         request = DatasetCreateRequest(
             name="Mixed Training Set",