Skip to content

Commit 3500de5

Browse files
committed
move test to top
1 parent a420738 commit 3500de5

1 file changed

Lines changed: 34 additions & 36 deletions

File tree

python/openai/tests/test_lora.py

Lines changed: 34 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,40 @@
3939
from .utils import OpenAIServer
4040

4141

42+
@pytest.mark.parametrize(
43+
"model_repository,model_name,expect_error",
44+
[
45+
("openai_model_repository", "", True), # Empty string as model name.
46+
("openai_model_repository", " ", True), # Whitespace-only model name.
47+
("openai_model_repository", "invalid/path", True),
48+
("openai_model_repository", "invalid\\path", True),
49+
("openai_model_repository", "../outside/repo", True),
50+
("openai_model_repository", "../test_models/identity_py", True),
51+
("test_models", "../test_models/identity_py", True),
52+
("test_models", "identity_py", False),
53+
("test_models", "mock_llm", False),
54+
],
55+
)
56+
def test_get_vllm_lora_name(model_repository: str, model_name: str, expect_error: bool):
57+
try:
58+
get_vllm_lora_names(model_repository, model_name, 1)
59+
except ValueError as e:
60+
if expect_error:
61+
assert (
62+
f"Invalid model name: '{model_name}'. Model names must be valid file-system-path segment names."
63+
== str(e)
64+
)
65+
else:
66+
raise pytest.fail(
67+
f"(model_repository='{model_repository}', model_name='{model_name}') raised ValueError unexpectedly: {e}"
68+
)
69+
else:
70+
if expect_error:
71+
raise pytest.fail(
72+
f"(model_repository='{model_repository}', model_name='{model_name}') did not raise ValueError as expected."
73+
)
74+
75+
4276
def is_vllm_installed():
4377
try:
4478
import vllm as _
@@ -365,41 +399,5 @@ def test_lora_separator_set_for_non_vllm_formatted_models(self):
365399
self._test_chat_completion(client, "doll")
366400

367401

368-
@pytest.mark.parametrize(
369-
"model_repository,model_name,expect_error",
370-
[
371-
("openai_model_repository", "", True), # Empty string as model name.
372-
("openai_model_repository", " ", True), # Whitespace-only model name.
373-
("openai_model_repository", "invalid/path", True),
374-
("openai_model_repository", "invalid\\path", True),
375-
("openai_model_repository", "../outside/repo", True),
376-
("openai_model_repository", "../test_models/identity_py", True),
377-
("test_models", "../test_models/identity_py", True),
378-
("test_models", "identity_py", False),
379-
("test_models", "mock_llm", False),
380-
],
381-
)
382-
def test_get_vllm_lora_name(
383-
model_repository: str, model_name: str, expect_error: bool
384-
):
385-
try:
386-
get_vllm_lora_names(model_repository, model_name, 1)
387-
except ValueError as e:
388-
if expect_error:
389-
assert (
390-
f"Invalid model name: '{model_name}'. Model names must be valid file-system-path segment names."
391-
== str(e)
392-
)
393-
else:
394-
raise pytest.fail(
395-
f"(model_repository='{model_repository}', model_name='{model_name}') raised ValueError unexpectedly: {e}"
396-
)
397-
else:
398-
if expect_error:
399-
raise pytest.fail(
400-
f"(model_repository='{model_repository}', model_name='{model_name}') did not raise ValueError as expected."
401-
)
402-
403-
404402
if __name__ == "__main__":
405403
unittest.main()

0 commit comments

Comments
 (0)