wehub-resource-sync
77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
import pytest
|
|
|
|
from vllm.v1.executor.vllm_net_devices import normalize_pci
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"addr, expected",
|
|
[
|
|
("0000:3f:00.0", (0, 0x3F, 0, 0)),
|
|
("0001:00:00.0", (1, 0, 0, 0)),
|
|
("00000001:00:00.0", (1, 0, 0, 0)),
|
|
("0009:00:00.0", (9, 0, 0, 0)),
|
|
("0105:00:00.0", (0x105, 0, 0, 0)),
|
|
("0000:0a:1f.7", (0, 0x0A, 0x1F, 7)),
|
|
],
|
|
)
|
|
def test_normalize_pci_full_domain(addr, expected):
|
|
assert normalize_pci(addr) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"addr, expected",
|
|
[
|
|
("01:00.0", (0, 1, 0, 0)),
|
|
("3f:00.0", (0, 0x3F, 0, 0)),
|
|
("40:00.0", (0, 0x40, 0, 0)),
|
|
("ff:1f.7", (0, 0xFF, 0x1F, 7)),
|
|
],
|
|
)
|
|
def test_normalize_pci_short_form(addr, expected):
|
|
assert normalize_pci(addr) == expected
|
|
|
|
|
|
def test_normalize_pci_case_insensitive():
|
|
assert normalize_pci("0A:1F.7") == normalize_pci("0a:1f.7")
|
|
|
|
|
|
def test_normalize_pci_strips_whitespace():
|
|
assert normalize_pci(" 0001:00:00.0 ") == (1, 0, 0, 0)
|
|
|
|
|
|
def test_normalize_pci_strips_0x_prefix():
|
|
assert normalize_pci("0x0001:00:00.0") == (1, 0, 0, 0)
|
|
|
|
|
|
def test_normalize_pci_missing_function_raises():
|
|
with pytest.raises(ValueError, match="missing function suffix"):
|
|
normalize_pci("0001:00:00")
|
|
|
|
|
|
def test_normalize_pci_invalid_function_char_raises():
|
|
with pytest.raises(ValueError, match="invalid PCI function"):
|
|
normalize_pci("0001:00:00.z")
|
|
|
|
|
|
def test_normalize_pci_too_many_segments_raises():
|
|
with pytest.raises(ValueError, match="invalid PCI BDF"):
|
|
normalize_pci("a:b:c:d.0")
|
|
|
|
|
|
def test_normalize_pci_bus_out_of_range_raises():
|
|
with pytest.raises(ValueError, match="out of range"):
|
|
normalize_pci("0000:1ff:00.0")
|
|
|
|
|
|
def test_normalize_pci_device_out_of_range_raises():
|
|
with pytest.raises(ValueError, match="out of range"):
|
|
normalize_pci("0000:00:20.0")
|
|
|
|
|
|
def test_normalize_pci_empty_string_raises():
|
|
with pytest.raises(ValueError):
|
|
normalize_pci("")
|