Skip to content

GitLab

"benchmarks/vscode:/vscode.git/clone" did not exist on "1972f71f1a4ececfe69c9c7f1790f0de20214125"
Unverified Commit 5588d770 authored by Biswa Panda's avatar Biswa Panda Committed by GitHub
Browse files

test: add timeout for lora utils tests (#4781)

parent d66e6755
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
lib/bindings/python/tests/test_lora_utils.py
View file @ 5588d770
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0
import pytest
from dynamo.llm import lora_name_to_id from dynamo.llm import lora_name_to_id
max_int32 = 0x7FFFFFFF max_int32 = 0x7FFFFFFF
class TestLoraNameToId: class TestLoraNameToId:
@pytest.mark.timeout(5)
def test_import_function(self): def test_import_function(self):
assert callable(lora_name_to_id) assert callable(lora_name_to_id)
@pytest.mark.timeout(5)
def test_returns_positive_integer_for_different_names(self): def test_returns_positive_integer_for_different_names(self):
for i in range(100): for i in range(100):
result = lora_name_to_id(f"test_lora_{i}") result = lora_name_to_id(f"test_lora_{i}")
assert isinstance(result, int) assert isinstance(result, int)
assert 1 <= result <= max_int32 assert 1 <= result <= max_int32
@pytest.mark.timeout(5)
def test_different_names_produce_different_ids(self): def test_different_names_produce_different_ids(self):
id1 = lora_name_to_id("lora_adapter_1") id1 = lora_name_to_id("lora_adapter_1")
id2 = lora_name_to_id("lora_adapter_2") id2 = lora_name_to_id("lora_adapter_2")
assert id1 != id2 assert id1 != id2
@pytest.mark.timeout(5)
def test_consistency_across_multiple_calls(self): def test_consistency_across_multiple_calls(self):
test_names = [f"lora_{i}" for i in range(100)] test_names = [f"lora_{i}" for i in range(100)]
results_first = [lora_name_to_id(name) for name in test_names] results_first = [lora_name_to_id(name) for name in test_names]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment