mirror of
https://github.com/vllm-project/vllm.git
synced 2026-06-06 00:16:14 +00:00
[BugFix][CI] Fix added _has_module tests (#44248)
Signed-off-by: Nick Hill <nickhill123@gmail.com>
This commit is contained in:
@@ -83,42 +83,6 @@ class TestHasModule:
|
||||
):
|
||||
assert _has_module("fake_native_ext") is False
|
||||
|
||||
def test_returns_false_on_os_error_during_import(self):
|
||||
"""Some shared-library failures surface as ``OSError``."""
|
||||
fake_spec = MagicMock()
|
||||
|
||||
with (
|
||||
patch(
|
||||
"vllm.utils.import_utils.importlib.util.find_spec",
|
||||
return_value=fake_spec,
|
||||
),
|
||||
patch(
|
||||
"vllm.utils.import_utils.importlib.import_module",
|
||||
side_effect=OSError("cannot load library"),
|
||||
),
|
||||
):
|
||||
assert _has_module("fake_native_ext_os") is False
|
||||
|
||||
def test_returns_false_on_unexpected_error_during_import(self):
|
||||
"""A broken extension may raise a non-import error (e.g. ``RuntimeError``).
|
||||
|
||||
Such modules are not usable, so ``_has_module`` should still return
|
||||
``False`` rather than letting the exception propagate.
|
||||
"""
|
||||
fake_spec = MagicMock()
|
||||
|
||||
with (
|
||||
patch(
|
||||
"vllm.utils.import_utils.importlib.util.find_spec",
|
||||
return_value=fake_spec,
|
||||
),
|
||||
patch(
|
||||
"vllm.utils.import_utils.importlib.import_module",
|
||||
side_effect=RuntimeError("CUDA driver version is insufficient"),
|
||||
),
|
||||
):
|
||||
assert _has_module("fake_broken_ext") is False
|
||||
|
||||
def test_returns_false_when_find_spec_raises(self):
|
||||
"""``find_spec`` itself can raise for dotted names whose parent package
|
||||
fails to import. This should be treated as the module being unavailable.
|
||||
|
||||
Reference in New Issue
Block a user