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,