mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 05:25:00 +08:00
53 lines
981 B
Python
53 lines
981 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
import ast
|
|
|
|
import pytest
|
|
|
|
from tools.validate_config import validate_ast
|
|
|
|
_TestConfig1 = """
|
|
@config
|
|
class _TestConfig1:
|
|
pass
|
|
"""
|
|
|
|
_TestConfig2 = '''
|
|
@config
|
|
@dataclass
|
|
class _TestConfig2:
|
|
a: int
|
|
"""docstring"""
|
|
'''
|
|
|
|
_TestConfig3 = """
|
|
@config
|
|
@dataclass
|
|
class _TestConfig3:
|
|
a: int = 1
|
|
"""
|
|
|
|
_TestConfig4 = '''
|
|
@config
|
|
@dataclass
|
|
class _TestConfig4:
|
|
a: Union[Literal[1], Literal[2]] = 1
|
|
"""docstring"""
|
|
'''
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("test_config", "expected_error"),
|
|
[
|
|
(_TestConfig1, "must be a dataclass"),
|
|
(_TestConfig2, "must have a default"),
|
|
(_TestConfig3, "must have a docstring"),
|
|
(_TestConfig4, "must use a single Literal"),
|
|
],
|
|
)
|
|
def test_config(test_config, expected_error):
|
|
tree = ast.parse(test_config)
|
|
with pytest.raises(Exception, match=expected_error):
|
|
validate_ast(tree)
|