biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Convert formatting to use ruff instead of yapf + isort (#26247)

Browse Source 
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2025-10-05 15:06:22 +01:00
committed by GitHub
parent 17edd8a807
commit d6953beb91
1508 changed files with 115244 additions and 94146 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
tests/utils_/test_gc_utils.py
View File

@@ -3,8 +3,11 @@
from dataclasses import dataclass
from typing import Any
from vllm.utils.gc_utils import (GCDebugConfig, _compute_detailed_type,
_compute_top_gc_collected_objects)
from vllm.utils.gc_utils import (
GCDebugConfig,
_compute_detailed_type,
_compute_top_gc_collected_objects,
)
@dataclass
@@ -21,38 +24,51 @@ class ListWrapper:
def test_compute_detailed_type():
assert _compute_detailed_type(
Normal(v=8)) == "<class 'tests.utils_.test_gc_utils.Normal'>"
assert (
_compute_detailed_type(Normal(v=8))
== "<class 'tests.utils_.test_gc_utils.Normal'>"
)
assert _compute_detailed_type([1, 2, 3]) == "<class 'list'>(size:3)"
assert _compute_detailed_type({4, 5}) == "<class 'set'>(size:2)"
assert _compute_detailed_type({6: 7}) == "<class 'dict'>(size:1)"
assert _compute_detailed_type(ListWrapper(
vs=[])) == "<class 'tests.utils_.test_gc_utils.ListWrapper'>(size:0)"
assert (
_compute_detailed_type(ListWrapper(vs=[]))
== "<class 'tests.utils_.test_gc_utils.ListWrapper'>(size:0)"
)
def test_compute_top_gc_collected_objects():
objects: list[Any] = [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12],
{13, 14}, {
15: 16,
17: 18
},
Normal(v=19),
Normal(v=20),
Normal(v=21)]
objects: list[Any] = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[10, 11, 12],
{13, 14},
{15: 16, 17: 18},
Normal(v=19),
Normal(v=20),
Normal(v=21),
]
assert _compute_top_gc_collected_objects(objects, top=-1) == ""
assert _compute_top_gc_collected_objects(objects, top=0) == ""
assert _compute_top_gc_collected_objects(
objects, top=1) == " 4:<class 'list'>(size:3)"
assert _compute_top_gc_collected_objects(objects, top=2) == "\n".join([
" 4:<class 'list'>(size:3)",
" 3:<class 'tests.utils_.test_gc_utils.Normal'>"
])
assert _compute_top_gc_collected_objects(objects, top=3) == "\n".join([
" 4:<class 'list'>(size:3)",
" 3:<class 'tests.utils_.test_gc_utils.Normal'>",
" 1:<class 'set'>(size:2)"
])
assert (
_compute_top_gc_collected_objects(objects, top=1)
== " 4:<class 'list'>(size:3)"
)
assert _compute_top_gc_collected_objects(objects, top=2) == "\n".join(
[
" 4:<class 'list'>(size:3)",
" 3:<class 'tests.utils_.test_gc_utils.Normal'>",
]
)
assert _compute_top_gc_collected_objects(objects, top=3) == "\n".join(
[
" 4:<class 'list'>(size:3)",
" 3:<class 'tests.utils_.test_gc_utils.Normal'>",
" 1:<class 'set'>(size:2)",
]
)
def test_gc_debug_config():
@@ -64,6 +80,6 @@ def test_gc_debug_config():
assert config.enabled
assert config.top_objects == -1
config = GCDebugConfig("{\"top_objects\":5}")
config = GCDebugConfig('{"top_objects":5}')
assert config.enabled
assert config.top_objects == 5