File tree Expand file tree Collapse file tree 1 file changed +11
-3
lines changed
inference_server/launcher/tests Expand file tree Collapse file tree 1 file changed +11
-3
lines changed Original file line number Diff line number Diff line change 1717Run as:
1818python -m pytest tests/test_launcher.py -v
1919"""
20-
21- from unittest .mock import patch
20+ import sys
21+ from unittest .mock import MagicMock , patch
2222
2323import pytest
2424from fastapi .testclient import TestClient
2525
26+ # Mock vllm before importing launcher
27+ sys .modules ["vllm" ] = MagicMock ()
28+ sys .modules ["vllm.utils" ] = MagicMock ()
29+ sys .modules ["vllm.utils.argparse_utils" ] = MagicMock ()
30+ sys .modules ["vllm.entrypoints.openai.api_server" ] = MagicMock ()
31+ sys .modules ["vllm.entrypoints.openai.cli_args" ] = MagicMock ()
32+ sys .modules ["vllm.entrypoints.utils" ] = MagicMock ()
33+
2634# Import the application and classes
27- from launcher import (
35+ from launcher import ( # noqa: E402
2836 VllmConfig ,
2937 VllmInstance ,
3038 VllmMultiProcessManager ,
You can’t perform that action at this time.
0 commit comments