chore: Update shared API tests

PiperOrigin-RevId: 930107901

Author: jaycee-li

google/genai/tests/conftest.py

@@ -61,6 +61,12 @@ def replays_prefix():
   return 'test'
 
 
+# Overridden at the module level for each test file.
+@pytest.fixture
+def location():
+  return None
+
+
 def _get_replay_id(use_vertex: bool, replays_prefix: str) -> str:
   test_name_ending = os.environ.get('PYTEST_CURRENT_TEST').split('::')[-1]
   test_name = (
@@ -72,7 +78,7 @@ def _get_replay_id(use_vertex: bool, replays_prefix: str) -> str:
 
 
 @pytest.fixture
-def client(use_vertex, replays_prefix, http_options, request):
+def client(use_vertex, replays_prefix, http_options, location, request):
   mode = request.config.getoption('--mode')
   if mode not in ['auto', 'record', 'replay', 'api', 'tap']:
     raise ValueError('Invalid mode: ' + mode)
@@ -121,6 +127,7 @@ def client(use_vertex, replays_prefix, http_options, request):
       vertexai=use_vertex,
       http_options=http_options,
       private=private,
+      location=location,
   )
 
   with mock.patch.object(

google/genai/tests/pytest_helper.py

@@ -143,6 +143,7 @@ def setup(
     test_method: Optional[str] = None,
     test_table: Optional[list[TestTableItem]] = None,
     http_options: Optional[HttpOptions] = None,
+    location: Optional[str] = None,
 ):
   """Generates parameterization for tests, run for both Vertex and MLDev."""
   replays_directory = (
@@ -192,10 +193,10 @@ def setup(
 
   # Add fixture for requested client option.
   return pytest.mark.parametrize(
-      'use_vertex, replays_prefix, http_options',
+      'use_vertex, replays_prefix, http_options, location',
       [
-          (True, replays_directory, http_options),
-          (False, replays_directory, http_options),
+          (True, replays_directory, http_options, location),
+          (False, replays_directory, http_options, location),
       ],
   )
 

google/genai/tests/shared/__init__.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 #
 
-GEMINI_MODEL = 'gemini-3.1-pro-preview'  # Gemini only
+GEMINI_MODEL = 'gemini-3.5-flash'  # Gemini only

google/genai/tests/shared/models/test_embed.py

@@ -34,7 +34,7 @@
     pytest_helper.TestTableItem(
         name='test_embed_gemini_embedding_2',
         parameters=genai_types.EmbedContentParameters(
-            model='gemini-embedding-2-preview',
+            model='gemini-embedding-2',
             contents='Hello world!',
             config={
                 'output_dimensionality': 10,

google/genai/tests/shared/tunings/__init__.py

@@ -11,5 +11,4 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# GEMINI_TUNABLE_MODEL = 'gemini-3.1-flash-lite'
-GEMINI_TUNABLE_MODEL = 'gemini-2.5-pro'
+GEMINI_TUNABLE_MODEL = 'gemini-3.5-flash'

google/genai/tests/shared/tunings/test_create.py

@@ -43,6 +43,8 @@
     globals_for_file=globals(),
     test_method='tunings.tune',
     test_table=test_table,
+    # Ignore the env var and always set the location to us-central1
+    location='us-central1',
 )
 
 pytest_plugins = ('pytest_asyncio',)

google/genai/tests/shared/tunings/test_create_get_cancel.py

@@ -54,6 +54,8 @@ def create_get_cancel(client, parameters):
     globals_for_file=globals(),
     test_method="create_get_cancel",
     test_table=test_table,
+    # Ignore the env var and always set the location to us-central1
+    location="us-central1",
 )
 
 pytest_plugins = ("pytest_asyncio",)