wehub-resource-sync
36 lines
1.5 KiB
Python
36 lines
1.5 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
"""Unit tests for OmniInputPreprocessor._process_text routing."""
|
|
|
|
import pytest
|
|
from pytest_mock import MockerFixture
|
|
|
|
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
|
|
|
|
|
|
class TestProcessTextMmProcessorKwargsRouting:
|
|
"""Presence-based routing: an explicit empty ``mm_processor_kwargs`` dict
|
|
on the prompt still routes through ``_process_multimodal``. Required for
|
|
AR-based image generation (e.g. GLM-Image t2i) where the HF processor
|
|
supplies the image-generation scaffold from its own defaults and no extra
|
|
kwargs are needed from the caller.
|
|
"""
|
|
|
|
@pytest.fixture
|
|
def preprocessor(self, mocker: MockerFixture):
|
|
from vllm_omni.inputs.preprocess import OmniInputPreprocessor
|
|
|
|
instance = object.__new__(OmniInputPreprocessor)
|
|
instance._process_multimodal = mocker.MagicMock(return_value={})
|
|
instance._tokenize_prompt = mocker.MagicMock(return_value=[1, 2, 3])
|
|
return instance
|
|
|
|
def test_empty_mm_processor_kwargs_routes_to_multimodal(self, preprocessor):
|
|
preprocessor._process_text({"prompt": "hello", "mm_processor_kwargs": {}})
|
|
assert preprocessor._process_multimodal.called
|
|
assert not preprocessor._tokenize_prompt.called
|
|
|
|
def test_missing_mm_processor_kwargs_routes_to_tokenize(self, preprocessor):
|
|
preprocessor._process_text({"prompt": "hello"})
|
|
assert preprocessor._tokenize_prompt.called
|
|
assert not preprocessor._process_multimodal.called
|