scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

style(tests): align provider tests with Ruff formatting

Browse source
This commit is contained in:
Henrik Bakke Dukefoss 2026-09-05 15:21:15 +02:00
parent 9e46e25825
commit 8c11b2a313
1 changed files with 5 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tests/test_litellm/llms/openai_like/test_standardcompute_provider.py
View file

@ -98,10 +98,7 @@ async def test_standardcompute_environment_key_and_api_base_override(monkeypatch
api_base="https://example.test/v1",
messages=[{"role": "user", "content": "Connection check"}],
)
assert (
request.calls.last.request.headers["authorization"]
== "Bearer environment-test-key"
)
assert request.calls.last.request.headers["authorization"] == "Bearer environment-test-key"
assert result.choices[0].message.content == "Connected"
@ -129,15 +126,10 @@ async def test_standardcompute_stream_consumes_chat_completion_events():
"choices": [{"index": 0, "delta": {}, "finish_reason": "stop"}],
},
]
events = (
"".join("data: " + json.dumps(chunk) + "\n\n" for chunk in chunks)
+ "data: [DONE]\n\n"
)
events = "".join("data: " + json.dumps(chunk) + "\n\n" for chunk in chunks) + "data: [DONE]\n\n"
with respx.mock as transport:
request = transport.post("https://api.stdcmpt.com/v1/chat/completions").mock(
return_value=httpx.Response(
200, text=events, headers={"content-type": "text/event-stream"}
)
return_value=httpx.Response(200, text=events, headers={"content-type": "text/event-stream"})
)
stream = await litellm.acompletion(
model="standardcompute/standardcompute",
@ -147,12 +139,5 @@ async def test_standardcompute_stream_consumes_chat_completion_events():
)
received = [chunk async for chunk in stream]
assert json.loads(request.calls.last.request.content)["stream"] is True
assert (
"".join(
chunk.choices[0].delta.content or "" for chunk in received if chunk.choices
)
== "Connected"
)
assert any(
chunk.choices and chunk.choices[0].finish_reason == "stop" for chunk in received
)
assert "".join(chunk.choices[0].delta.content or "" for chunk in received if chunk.choices) == "Connected"
assert any(chunk.choices and chunk.choices[0].finish_reason == "stop" for chunk in received)