[Refactor][CI] Reorganize test tree, and move slow e2e to weekly (#4951)
Signed-off-by: wangyu <410167048@qq.com>
This commit is contained in:
1 parent
925347332f
commit
7c17111ff5
158 files changed
+3479
-4130
No files matched your search
@@ -0,0 +1,51 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
||||
"""Tests for vllm_omni.engine.arg_utils — invariants that must
|
||||
hold for the orchestrator/engine/server CLI flag partition."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import fields
|
||||
|
||||
import pytest
|
||||
|
||||
from vllm_omni.engine.arg_utils import (
|
||||
SHARED_FIELDS,
|
||||
OmniEngineArgs,
|
||||
internal_blacklist_keys,
|
||||
orchestrator_field_names,
|
||||
)
|
||||
|
||||
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
|
||||
|
||||
|
||||
def test_no_ambiguous_overlap_with_real_engine():
|
||||
"""OrchestratorArgs ∩ OmniEngineArgs must be ⊆ SHARED_FIELDS ∪ orchestrator-captured.
|
||||
|
||||
Fields that the orchestrator intentionally captures to prevent uniform
|
||||
propagation (async_chunk, tokenizer) are excluded — they exist in both
|
||||
classes by design but flow through DeployConfig, not kwargs forwarding.
|
||||
"""
|
||||
# Fields on both classes by design: orchestrator captures them to prevent
|
||||
# uniform per-stage propagation; redistributed via DeployConfig.
|
||||
orchestrator_captured = {"async_chunk", "tokenizer"}
|
||||
|
||||
orch = orchestrator_field_names()
|
||||
engine = {f.name for f in fields(OmniEngineArgs)}
|
||||
overlap = orch & engine
|
||||
unexpected = overlap - SHARED_FIELDS - orchestrator_captured
|
||||
assert not unexpected, (
|
||||
f"OmniEngineArgs has ambiguous overlap with OrchestratorArgs: "
|
||||
f"{sorted(unexpected)}. Update SHARED_FIELDS or remove duplication."
|
||||
)
|
||||
|
||||
|
||||
def test_internal_blacklist_keys_derived_from_orchestrator():
|
||||
"""Blacklist is exactly OrchestratorArgs fields minus SHARED_FIELDS.
|
||||
|
||||
This function replaces the old hardcoded INTERNAL_STAGE_OVERRIDE_KEYS
|
||||
frozenset. Asserts the contract so future changes to OrchestratorArgs
|
||||
automatically propagate to the blacklist.
|
||||
"""
|
||||
blacklist = internal_blacklist_keys()
|
||||
assert blacklist == orchestrator_field_names() - SHARED_FIELDS
|
||||
Reference in new issue
Block a user