scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test: fix tests

Browse source
This commit is contained in:
Krrish Dholakia 2025-11-22 16:50:50 -08:00
parent 93c2103097
commit cc5ecfd479
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/llms/vertex_ai/test_vertex_image_generation.py
View file

@ -28,7 +28,7 @@ class TestVertexImageGeneration:
def test_transform_optional_params_empty_dict(self):
"""Test transform_optional_params with empty dict"""
result = self.vertex_image_gen.transform_optional_params({})
expected = {}
expected = {"sampleCount": 1}
assert result == expected
def test_transform_optional_params_no_underscores(self):
@ -94,6 +94,7 @@ class TestVertexImageGeneration:
"veryLongParameterName": "test",
"multiWordConfigSetting": 42,
"anotherTestParam": True,
"sampleCount": 1,
}
assert result == expected
@ -101,7 +102,7 @@ class TestVertexImageGeneration:
"""Test transform_optional_params with single underscore params"""
input_params = {"test_param": "value", "a_b": "short"}
result = self.vertex_image_gen.transform_optional_params(input_params)
expected = {"testParam": "value", "aB": "short"}
expected = {"testParam": "value", "aB": "short", "sampleCount": 1}
assert result == expected
def test_transform_optional_params_preserves_values(self):
@ -124,6 +125,7 @@ class TestVertexImageGeneration:
"listParam": [1, 2, 3],
"dictParam": {"nested": "value"},
"noneParam": None,
"sampleCount": 1,
}
assert result == expected