scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

test(pruna): cover invalid size parsing branch

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Mubashir Osmani 2026-07-18 23:03:48 +00:00
parent 2e20a427c9
commit cc527fcfc6
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_litellm/llms/pruna/image_generation/test_pruna_image_gen_transformation.py
View file

@ -67,9 +67,10 @@ class TestPrunaImageGenerationTransformation:
)
assert result == {"aspect_ratio": "16:9", "seed": 7}
def test_map_openai_params_ignores_invalid_size(self):
@pytest.mark.parametrize("size", ["auto", "wide", "bigxsmall", "1024x"])
def test_map_openai_params_ignores_invalid_size(self, size):
result = self.config.map_openai_params(
non_default_params={"size": "auto"},
non_default_params={"size": size},
optional_params={},
model=self.model,
drop_params=False,