37 lines
995 B
Python
37 lines
995 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from vllm import version
|
|
|
|
|
|
def test_version_is_defined():
|
|
assert version.__version__ is not None
|
|
|
|
|
|
def test_version_tuple():
|
|
assert len(version.__version_tuple__) in (3, 4, 5)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"version_tuple, version_str, expected",
|
|
[
|
|
((0, 0, "dev"), "0.0", True),
|
|
((0, 0, "dev"), "foobar", True),
|
|
((0, 7, 4), "0.6", True),
|
|
((0, 7, 4), "0.5", False),
|
|
((0, 7, 4), "0.7", False),
|
|
((1, 2, 3), "1.1", True),
|
|
((1, 2, 3), "1.0", False),
|
|
((1, 2, 3), "1.2", False),
|
|
# This won't work as expected
|
|
((1, 0, 0), "1.-1", True),
|
|
((1, 0, 0), "0.9", False),
|
|
((1, 0, 0), "0.17", False),
|
|
])
|
|
def test_prev_minor_version_was(version_tuple, version_str, expected):
|
|
with patch("vllm.version.__version_tuple__", version_tuple):
|
|
assert version._prev_minor_version_was(version_str) == expected
|