|
| 1 | +"""Unit tests for GLM-ASR vLLM sampling-parameter handling. |
| 2 | +
|
| 3 | +These tests exercise ``GLMASRVLLMEngine.generate`` without a GPU or a real |
| 4 | +vLLM installation: the vLLM entry points are stubbed in ``sys.modules`` and the |
| 5 | +audio/encoder/engine collaborators are mocked, so only the sampling-parameter |
| 6 | +wiring is under test. |
| 7 | +""" |
| 8 | + |
| 9 | +import re |
| 10 | +import sys |
| 11 | +import types |
| 12 | +import unittest |
| 13 | +from unittest import mock |
| 14 | + |
| 15 | + |
| 16 | +def _install_vllm_stub(): |
| 17 | + """Install a minimal ``vllm`` stub whose SamplingParams records kwargs.""" |
| 18 | + |
| 19 | + captured = {} |
| 20 | + |
| 21 | + class _RecordingSamplingParams: |
| 22 | + def __init__(self, **kwargs): |
| 23 | + captured.clear() |
| 24 | + captured.update(kwargs) |
| 25 | + |
| 26 | + class _EmbedsPrompt: |
| 27 | + def __init__(self, **kwargs): |
| 28 | + self.kwargs = kwargs |
| 29 | + |
| 30 | + vllm_mod = types.ModuleType("vllm") |
| 31 | + vllm_mod.SamplingParams = _RecordingSamplingParams |
| 32 | + vllm_mod.LLM = object |
| 33 | + inputs_mod = types.ModuleType("vllm.inputs") |
| 34 | + inputs_mod.EmbedsPrompt = _EmbedsPrompt |
| 35 | + data_mod = types.ModuleType("vllm.inputs.data") |
| 36 | + data_mod.EmbedsPrompt = _EmbedsPrompt |
| 37 | + |
| 38 | + sys.modules["vllm"] = vllm_mod |
| 39 | + sys.modules["vllm.inputs"] = inputs_mod |
| 40 | + sys.modules["vllm.inputs.data"] = data_mod |
| 41 | + return captured |
| 42 | + |
| 43 | + |
| 44 | +class GLMASRSamplingParamsTest(unittest.TestCase): |
| 45 | + def setUp(self): |
| 46 | + self.captured = _install_vllm_stub() |
| 47 | + from funasr.models.glm_asr.inference_vllm import GLMASRVLLMEngine |
| 48 | + |
| 49 | + # Build an engine without running __init__ (no model load / GPU needed). |
| 50 | + engine = GLMASRVLLMEngine.__new__(GLMASRVLLMEngine) |
| 51 | + engine.device = "cpu" |
| 52 | + engine._encode_audio = mock.Mock(return_value="audio_embeds") |
| 53 | + engine._build_prompt_embeds = mock.Mock( |
| 54 | + return_value=mock.Mock(float=lambda: "embeds") |
| 55 | + ) |
| 56 | + |
| 57 | + token_out = types.SimpleNamespace(token_ids=[1, 2, 3]) |
| 58 | + vllm_output = types.SimpleNamespace(outputs=[token_out]) |
| 59 | + engine.vllm_engine = mock.Mock() |
| 60 | + engine.vllm_engine.generate = mock.Mock(return_value=[vllm_output]) |
| 61 | + engine.tokenizer = mock.Mock() |
| 62 | + engine.tokenizer.decode = mock.Mock(return_value="hello world") |
| 63 | + self.engine = engine |
| 64 | + |
| 65 | + def test_defaults_preserve_greedy_behavior(self): |
| 66 | + results = self.engine.generate("a.wav") |
| 67 | + self.assertEqual(results, [{"key": "a", "text": "hello world"}]) |
| 68 | + self.assertEqual(self.captured["temperature"], 0.0) |
| 69 | + self.assertEqual(self.captured["top_p"], 1.0) |
| 70 | + self.assertEqual(self.captured["top_k"], -1) |
| 71 | + self.assertEqual(self.captured["repetition_penalty"], 1.0) |
| 72 | + self.assertEqual(self.captured["max_tokens"], 500) |
| 73 | + |
| 74 | + def test_caller_sampling_params_are_forwarded(self): |
| 75 | + self.engine.generate( |
| 76 | + "a.wav", max_new_tokens=128, temperature=0.7, top_p=0.9, top_k=20 |
| 77 | + ) |
| 78 | + self.assertEqual(self.captured["max_tokens"], 128) |
| 79 | + self.assertEqual(self.captured["temperature"], 0.7) |
| 80 | + self.assertEqual(self.captured["top_p"], 0.9) |
| 81 | + self.assertEqual(self.captured["top_k"], 20) |
| 82 | + |
| 83 | + def test_non_positive_top_k_is_normalized_to_disabled(self): |
| 84 | + self.engine.generate("a.wav", top_k=0) |
| 85 | + self.assertEqual(self.captured["top_k"], -1) |
| 86 | + |
| 87 | + def test_repetition_penalty_is_forced_neutral_in_prompt_embeds_mode(self): |
| 88 | + # A non-neutral repetition_penalty would crash vLLM prompt-embeds mode |
| 89 | + # (issue #2948), so it must be coerced back to 1.0 rather than forwarded. |
| 90 | + self.engine.generate("a.wav", repetition_penalty=1.3) |
| 91 | + self.assertEqual(self.captured["repetition_penalty"], 1.0) |
| 92 | + |
| 93 | + def test_neutral_repetition_penalty_passes_through(self): |
| 94 | + self.engine.generate("a.wav", repetition_penalty=1.0) |
| 95 | + self.assertEqual(self.captured["repetition_penalty"], 1.0) |
| 96 | + |
| 97 | + |
| 98 | +class SafeRepetitionPenaltyTest(unittest.TestCase): |
| 99 | + def setUp(self): |
| 100 | + _install_vllm_stub() |
| 101 | + import funasr.models.glm_asr.inference_vllm as mod |
| 102 | + |
| 103 | + self.mod = mod |
| 104 | + # Reset the process-wide warn-once flag between tests. |
| 105 | + mod._warned_rep_penalty = False |
| 106 | + |
| 107 | + def test_neutral_and_none_map_to_one(self): |
| 108 | + self.assertEqual(self.mod._safe_repetition_penalty(1.0), 1.0) |
| 109 | + self.assertEqual(self.mod._safe_repetition_penalty(None), 1.0) |
| 110 | + |
| 111 | + def test_non_neutral_is_coerced_and_warns_once(self): |
| 112 | + with self.assertLogs(self.mod.logger, level="WARNING") as cm: |
| 113 | + self.assertEqual(self.mod._safe_repetition_penalty(1.5), 1.0) |
| 114 | + self.assertTrue(any("2948" in line for line in cm.output)) |
| 115 | + self.assertTrue(self.mod._warned_rep_penalty) |
| 116 | + |
| 117 | + |
| 118 | +if __name__ == "__main__": |
| 119 | + unittest.main() |
0 commit comments