Skip to content

Commit 03fa35d

Browse files
committed
move test to top
1 parent a420738 commit 03fa35d

1 file changed

Lines changed: 36 additions & 36 deletions

File tree

python/openai/tests/test_lora.py

Lines changed: 36 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,42 @@
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(
57+
model_repository: str, model_name: str, expect_error: bool
58+
):
59+
try:
60+
get_vllm_lora_names(model_repository, model_name, 1)
61+
except ValueError as e:
62+
if expect_error:
63+
assert (
64+
f"Invalid model name: '{model_name}'. Model names must be valid file-system-path segment names."
65+
== str(e)
66+
)
67+
else:
68+
raise pytest.fail(
69+
f"(model_repository='{model_repository}', model_name='{model_name}') raised ValueError unexpectedly: {e}"
70+
)
71+
else:
72+
if expect_error:
73+
raise pytest.fail(
74+
f"(model_repository='{model_repository}', model_name='{model_name}') did not raise ValueError as expected."
75+
)
76+
77+
4278
def is_vllm_installed():
4379
try:
4480
import vllm as _
@@ -365,41 +401,5 @@ def test_lora_separator_set_for_non_vllm_formatted_models(self):
365401
self._test_chat_completion(client, "doll")
366402

367403

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-
404404
if __name__ == "__main__":
405405
unittest.main()

0 commit comments

Comments
 (0)