25 lines
821 B
Python
25 lines
821 B
Python
import pytest
|
|
|
|
|
|
@pytest.mark.parametrize("case", list(range(120)))
|
|
def test_list_models_cases(api_client, case):
|
|
resp = api_client.get("/v1/models", headers={"x-case": str(case)})
|
|
assert resp.status_code == 200
|
|
payload = resp.json()
|
|
assert payload["object"] == "list"
|
|
assert isinstance(payload["data"], list)
|
|
|
|
|
|
@pytest.mark.parametrize("model_id", [f"model-a.gguf" for _ in range(120)])
|
|
def test_get_model_ok(api_client, model_id):
|
|
resp = api_client.get(f"/v1/models/{model_id}")
|
|
assert resp.status_code == 200
|
|
payload = resp.json()
|
|
assert payload["id"] == model_id
|
|
|
|
|
|
@pytest.mark.parametrize("model_id", [f"missing-{i}" for i in range(120)])
|
|
def test_get_model_not_found(api_client, model_id):
|
|
resp = api_client.get(f"/v1/models/{model_id}")
|
|
assert resp.status_code == 404
|