[Misc] Cleanup: use consistent pytest-mock in unit tests (#2698)
Signed-off-by: yuanheng <jonathan.zhaoyh@gmail.com>
This commit is contained in:
1 parent
2c67c30550
commit
e0cdbe9a5d
26 files changed
+1598
-1369
No files matched your search
@@ -6,7 +6,7 @@ explicitly patch values that differ from vLLM.
|
||||
|
||||
import argparse
|
||||
import inspect
|
||||
from unittest.mock import Mock
|
||||
from types import SimpleNamespace
|
||||
|
||||
import pytest
|
||||
from pydantic import ValidationError
|
||||
@@ -102,7 +102,7 @@ def test_qwen3_tts_codec_frame_rate_patching():
|
||||
vllm_config = EngineArgs().create_model_config()
|
||||
|
||||
# Create a mock talking config with a dummy value for position_id_per_seconds
|
||||
mock_talker_config = Mock()
|
||||
mock_talker_config = SimpleNamespace()
|
||||
mock_talker_config.position_id_per_seconds = 12.3
|
||||
vllm_config.hf_config.talker_config = mock_talker_config
|
||||
|
||||
@@ -146,13 +146,12 @@ def test_stage_specific_text_config_override():
|
||||
# Switch the created hf text config with a mock whose
|
||||
# values we want to pull through the text config helper
|
||||
stage_text_config = vllm_config.hf_text_config
|
||||
vllm_config.hf_text_config = Mock()
|
||||
vllm_config.hf_text_config = SimpleNamespace()
|
||||
stage_text_config.sliding_window = 4096
|
||||
stage_text_config.attention_chunk_size = 2048
|
||||
|
||||
# Move the stage config's text config getter & thinker config
|
||||
mock_stage_config = Mock()
|
||||
mock_stage_config.get_text_config.return_value = stage_text_config
|
||||
mock_stage_config = SimpleNamespace(get_text_config=lambda: stage_text_config)
|
||||
vllm_config.hf_config.thinker_config = mock_stage_config
|
||||
|
||||
# Ensure that create from a vLLM config correctly pulls the
|
||||
|
||||
Reference in new issue
Block a user