test_outputs.py 641 Bytes
Newer Older
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3

4
5
import pytest

6
7
from vllm.outputs import RequestOutput

8
9
pytestmark = pytest.mark.cpu_test

10
11
12
13
14
15
16
17
18
19

def test_request_output_forward_compatible():
    output = RequestOutput(request_id="test_request_id",
                           prompt="test prompt",
                           prompt_token_ids=[1, 2, 3],
                           prompt_logprobs=None,
                           outputs=[],
                           finished=False,
                           example_arg_added_in_new_version="some_value")
    assert output is not None