scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 124

remove unnecessary test

Browse source
This commit is contained in:
Jared Moore 2026-03-03 15:31:24 -08:00
parent f0020f2f37
commit 77912e6590
1 changed files with 0 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
tests/test_litellm/batches/test_vertex_batch_mapping.py
View file

@ -89,39 +89,3 @@ async def test_should_map_vertex_output_by_input_order(monkeypatch):
custom_llm_provider="vertex_ai",
)
@pytest.mark.asyncio
async def test_should_raise_on_vertex_output_count_mismatch(monkeypatch):
"""Guard against mismatched input/output line counts for Vertex."""
input_lines = [
{"custom_id": "id-1", "body": {"model": "m", "messages": []}},
{"custom_id": "id-2", "body": {"model": "m", "messages": []}},
]
output_lines = [
{"response": {"status_code": 200}},
]
content_map = {
"gs://bucket/input.jsonl": "\n".join(
json.dumps(line) for line in input_lines
).encode("utf-8"),
"gs://bucket/output.jsonl": "\n".join(
json.dumps(line) for line in output_lines
).encode("utf-8"),
}
async def fake_afile_content(*, file_id, **_kwargs):
return SimpleNamespace(content=content_map[file_id])
monkeypatch.setattr("litellm.files.main.afile_content", fake_afile_content)
batch = DummyBatch(
input_file_id="gs://bucket/input.jsonl",
output_file_id="gs://bucket/output.jsonl",
)
with pytest.raises(ValueError, match="output line count does not match input"):
await _get_batch_output_file_content_as_dictionary(
batch=batch,
custom_llm_provider="vertex_ai",
)