re-structure

tests/web/test_inference_api.py

@@ -11,8 +11,8 @@ from fastapi.testclient import TestClient
 from PIL import Image
 import io
 
-from inference.web.app import create_app
-from inference.web.config import ModelConfig, StorageConfig, AppConfig
+from backend.web.app import create_app
+from backend.web.config import ModelConfig, StorageConfig, AppConfig
 
 
 @pytest.fixture
@@ -87,8 +87,8 @@ class TestHealthEndpoint:
 class TestInferEndpoint:
     """Test /api/v1/infer endpoint."""
 
-    @patch('inference.pipeline.pipeline.InferencePipeline')
-    @patch('inference.pipeline.yolo_detector.YOLODetector')
+    @patch('backend.pipeline.pipeline.InferencePipeline')
+    @patch('backend.pipeline.yolo_detector.YOLODetector')
     def test_infer_accepts_png_file(
         self,
         mock_yolo_detector,
@@ -150,8 +150,8 @@ class TestInferEndpoint:
 
         assert response.status_code == 422  # Unprocessable Entity
 
-    @patch('inference.pipeline.pipeline.InferencePipeline')
-    @patch('inference.pipeline.yolo_detector.YOLODetector')
+    @patch('backend.pipeline.pipeline.InferencePipeline')
+    @patch('backend.pipeline.yolo_detector.YOLODetector')
     def test_infer_returns_cross_validation_if_available(
         self,
         mock_yolo_detector,
@@ -210,8 +210,8 @@ class TestInferEndpoint:
         # This test documents that it should be added
 
 
-    @patch('inference.pipeline.pipeline.InferencePipeline')
-    @patch('inference.pipeline.yolo_detector.YOLODetector')
+    @patch('backend.pipeline.pipeline.InferencePipeline')
+    @patch('backend.pipeline.yolo_detector.YOLODetector')
     def test_infer_handles_processing_errors_gracefully(
         self,
         mock_yolo_detector,
@@ -263,7 +263,7 @@ class TestResultsEndpoint:
 
         # Mock the storage helper to return our test file path
         with patch(
-            "inference.web.api.v1.public.inference.get_storage_helper"
+            "backend.web.api.v1.public.inference.get_storage_helper"
         ) as mock_storage:
             mock_helper = Mock()
             mock_helper.get_result_local_path.return_value = result_file
@@ -285,15 +285,15 @@ class TestInferenceServiceImports:
 
         This test will fail if there are ImportError issues like:
         - from ..inference.pipeline (wrong relative import)
-        - from inference.web.inference (non-existent module)
+        - from backend.web.inference (non-existent module)
 
         It ensures the imports are correct before runtime.
         """
-        from inference.web.services.inference import InferenceService
+        from backend.web.services.inference import InferenceService
 
         # Import the modules that InferenceService tries to import
-        from inference.pipeline.pipeline import InferencePipeline
-        from inference.pipeline.yolo_detector import YOLODetector
+        from backend.pipeline.pipeline import InferencePipeline
+        from backend.pipeline.yolo_detector import YOLODetector
         from shared.pdf.renderer import render_pdf_to_images
 
         # If we got here, all imports work correctly