[CI] Add pytest markers to current tests and update the doc. (#577)

Signed-off-by: Alicia <115451386+congw729@users.noreply.github.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>

tests/entrypoints/openai_api/test_image_server.py

@@ -23,6 +23,8 @@ from vllm_omni.entrypoints.openai.image_api_utils import (
 )
 from vllm_omni.inputs.data import OmniDiffusionSamplingParams
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 # Unit Tests
 
 

tests/entrypoints/openai_api/test_serving_chat_sampling_params.py

@@ -11,6 +11,8 @@ from unittest.mock import MagicMock
 import pytest
 from vllm.sampling_params import SamplingParams
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 
 @pytest.fixture
 def mock_comprehension_stage():

tests/entrypoints/openai_api/test_serving_speech.py

@@ -16,6 +16,8 @@ from vllm_omni.entrypoints.openai.serving_speech import (
 )
 from vllm_omni.outputs import OmniRequestOutput
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 logger = logging.getLogger(__name__)
 
 

tests/entrypoints/test_async_omni_diffusion_config.py

@@ -1,9 +1,13 @@
 # SPDX-License-Identifier: Apache-2.0
 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
 
+import pytest
+
 from vllm_omni.entrypoints import omni as omni_module
 from vllm_omni.entrypoints.async_omni import AsyncOmni
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 
 def test_default_stage_config_includes_cache_backend(monkeypatch):
     """Ensure cache_backend/cache_config are preserved in default diffusion stage."""

tests/entrypoints/test_omni_diffusion.py

@@ -9,6 +9,8 @@ import pytest
 from vllm_omni.entrypoints.stage_utils import SHUTDOWN_TASK
 from vllm_omni.inputs.data import OmniDiffusionSamplingParams
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 # Suppress noisy DeprecationWarnings from optional Swig bindings imported by vLLM dependencies.
 warnings.filterwarnings(
     "ignore",

tests/entrypoints/test_omni_input_preprocessor.py

@@ -1,5 +1,9 @@
+import pytest
+
 from vllm_omni.inputs.preprocess import OmniInputPreprocessor
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 
 def _make_preprocessor(monkeypatch):
     preprocessor = object.__new__(OmniInputPreprocessor)

tests/entrypoints/test_omni_llm.py

@@ -9,6 +9,8 @@ from vllm import SamplingParams
 
 from vllm_omni.entrypoints.stage_utils import SHUTDOWN_TASK
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 # Suppress noisy DeprecationWarnings from optional Swig bindings imported by vLLM dependencies.
 warnings.filterwarnings(
     "ignore",

tests/entrypoints/test_omni_new_request_data.py

@@ -1,9 +1,12 @@
 from types import SimpleNamespace
 
+import pytest
 import torch
 
 from vllm_omni.core.sched.output import OmniNewRequestData
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 
 def test_omni_new_request_data_copies_payloads():
     prompt_embeds = torch.randn(2, 3)

tests/entrypoints/test_omni_stage_diffusion_config.py

@@ -1,8 +1,12 @@
 # SPDX-License-Identifier: Apache-2.0
 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
 
+import pytest
+
 from vllm_omni.entrypoints.omni_stage import _build_od_config
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 
 def test_build_od_config_includes_diffusion_fields():
     engine_args = {

tests/entrypoints/test_stage_utils.py

@@ -6,6 +6,8 @@ import pytest
 
 from vllm_omni.entrypoints.stage_utils import set_stage_devices
 
+pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
+
 
 def _make_dummy_torch(call_log):
     class _Props: