diff --git a/microservices/model-download/docs/user-guide/running-tests.md b/microservices/model-download/docs/user-guide/running-tests.md index 6b9cac1f0..a2179f816 100644 --- a/microservices/model-download/docs/user-guide/running-tests.md +++ b/microservices/model-download/docs/user-guide/running-tests.md @@ -11,9 +11,10 @@ You can use either traditional pip/venv or the modern uv tool for dependency man #### Using uv ```bash +# Navigate to model-download service directory +cd microservices/model-download + # Install uv if not already installed -curl -LsSf https://astral.sh/uv/install.sh | sh -# or pip install uv # Create and activate virtual environment diff --git a/microservices/model-download/tests/unit/test_ultralytics_plugin.py b/microservices/model-download/tests/unit/test_ultralytics_plugin.py index 5e9e1ac51..83351cf46 100644 --- a/microservices/model-download/tests/unit/test_ultralytics_plugin.py +++ b/microservices/model-download/tests/unit/test_ultralytics_plugin.py @@ -213,7 +213,7 @@ def test_download_success(self, mock_getenv, mock_call_script, ultralytics_plugi assert result["model_name"] == "yolov8n.pt" assert result["source"] == "ultralytics" - assert result["return_code"] == 0 + assert result["success"] == True assert "ultralytics" in result["download_path"] @patch.object(UltralyticsDownloader, '_call_bash_script') @@ -246,7 +246,7 @@ def test_download_without_quantization(self, mock_call_script, ultralytics_plugi assert result["model_name"] == "yolov8n.pt" assert result["source"] == "ultralytics" - assert result["return_code"] == 0 + assert result["success"] == True @patch.object(UltralyticsDownloader, '_call_bash_script') @patch('os.getenv') @@ -327,7 +327,7 @@ def test_download_different_parameters(self, mock_call_script, ultralytics_plugi assert mock_call_script.call_count == expected_calls assert result["model_name"] == model_name assert result["source"] == "ultralytics" - assert result["return_code"] == 0 + assert result["success"] == True @patch('os.path.exists') @patch('subprocess.Popen') @@ -444,7 +444,7 @@ def test_end_to_end_download_workflow(self, mock_call_script, ultralytics_plugin # Verify results assert result["model_name"] == "yolov8n.pt" assert result["source"] == "ultralytics" - assert result["return_code"] == 0 + assert result["success"] == True assert "ultralytics" in result["download_path"] # Test post-processing