From 44e313382792e88b05be141c9c2b0bd4325bfaa8 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Wed, 26 Nov 2025 12:26:49 -0800
Subject: [PATCH] test_imagen_get_complete_url

---
 ...rtex_ai_image_generation_transformation.py | 30 +++++++++----------
 1 file changed, 14 insertions(+), 16 deletions(-)

diff --git a/tests/test_litellm/llms/vertex_ai/image_generation/test_vertex_ai_image_generation_transformation.py b/tests/test_litellm/llms/vertex_ai/image_generation/test_vertex_ai_image_generation_transformation.py
index 9f33400594b..c387c466bc8 100644
--- a/tests/test_litellm/llms/vertex_ai/image_generation/test_vertex_ai_image_generation_transformation.py
+++ b/tests/test_litellm/llms/vertex_ai/image_generation/test_vertex_ai_image_generation_transformation.py
@@ -438,20 +438,18 @@ class TestVertexAIImageGenerationIntegration:
     def test_imagen_get_complete_url(self):
         """Test Imagen config URL generation"""
         config = VertexAIImagenImageGenerationConfig()
-        with patch.object(
-            config, "_resolve_vertex_project", return_value="test-project"
-        ), patch.object(
-            config, "_resolve_vertex_location", return_value="us-central1"
-        ):
-            url = config.get_complete_url(
-                api_base=None,
-                api_key=None,
-                model="imagegeneration@006",
-                optional_params={},
-                litellm_params={},
-            )
-            assert "test-project" in url
-            assert "us-central1" in url
-            assert "imagegeneration@006" in url
-            assert "predict" in url
+        url = config.get_complete_url(
+            api_base=None,
+            api_key=None,
+            model="imagegeneration@006",
+            optional_params={},
+            litellm_params={
+                "vertex_project": "test-project",
+                "vertex_location": "us-central1",
+            },
+        )
+        assert "test-project" in url
+        assert "us-central1" in url
+        assert "imagegeneration@006" in url
+        assert "predict" in url