From cc527fcfc6566069e0c9b6ad496f7f3dcec551de Mon Sep 17 00:00:00 2001 From: Mubashir Osmani Date: Sat, 18 Jul 2026 23:03:48 +0000 Subject: [PATCH] test(pruna): cover invalid size parsing branch Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../image_generation/test_pruna_image_gen_transformation.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/llms/pruna/image_generation/test_pruna_image_gen_transformation.py b/tests/test_litellm/llms/pruna/image_generation/test_pruna_image_gen_transformation.py index 8e491caabbe..48e59cfc0a5 100644 --- a/tests/test_litellm/llms/pruna/image_generation/test_pruna_image_gen_transformation.py +++ b/tests/test_litellm/llms/pruna/image_generation/test_pruna_image_gen_transformation.py @@ -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,