vllm-omni/tests/engine/duplex/test_duplex_intermediate.py

# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project

import pytest

from vllm_omni.engine.duplex.intermediate import get_stream_request_key

pytestmark = [pytest.mark.core_model, pytest.mark.cpu]


def test_get_stream_request_key_requires_stable_identifier():
    with pytest.raises(ValueError, match="stable request id"):
        get_stream_request_key({"ids": {"tts": [1, 2, 3]}})


def test_get_stream_request_key_accepts_global_request_id():
    assert get_stream_request_key({"global_request_id": ["duplex-sid-stage0"]}) == "duplex-sid-stage0"