Sync from v0.13
This commit is contained in:
21
tests/test_outputs.py
Normal file
21
tests/test_outputs.py
Normal file
@@ -0,0 +1,21 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
||||
|
||||
import pytest
|
||||
|
||||
from vllm.outputs import RequestOutput
|
||||
|
||||
pytestmark = pytest.mark.cpu_test
|
||||
|
||||
|
||||
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
|
||||
Reference in New Issue
Block a user