diff --git a/google/genai/tests/models/test_upscale_image.py b/google/genai/tests/models/test_upscale_image.py
deleted file mode 100644
index cf9c9e4b7..000000000
--- a/google/genai/tests/models/test_upscale_image.py
+++ /dev/null
@@ -1,163 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# 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.
-#
-
-
-"""Tests for upscale_image."""
-
-import os
-
-from pydantic import ValidationError
-import pytest
-
-from ... import types
-from .. import pytest_helper
-
-IMAGEN_MODEL_LATEST = 'imagen-4.0-upscale-preview'
-
-IMAGE_FILE_PATH = os.path.abspath(
-    os.path.join(os.path.dirname(__file__), '../data/bridge1.png')
-)
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_upscale_no_config',
-        exception_if_mldev=(
-            'only supported in Gemini Enterprise Agent Platform'
-        ),
-        parameters=types.UpscaleImageParameters(
-            model=IMAGEN_MODEL_LATEST,
-            image=types.Image.from_file(location=IMAGE_FILE_PATH),
-            upscale_factor='x2',
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_upscale',
-        exception_if_mldev=(
-            'only supported in Gemini Enterprise Agent Platform'
-        ),
-        parameters=types.UpscaleImageParameters(
-            model=IMAGEN_MODEL_LATEST,
-            image=types.Image.from_file(location=IMAGE_FILE_PATH),
-            upscale_factor='x2',
-            config=types.UpscaleImageConfig(
-                include_rai_reason=True,
-                person_generation=types.PersonGeneration.ALLOW_ADULT,
-                safety_filter_level=types.SafetyFilterLevel.BLOCK_LOW_AND_ABOVE,
-                output_mime_type='image/jpeg',
-                output_compression_quality=80,
-                enhance_input_image=True,
-                image_preservation_factor=0.6,
-                labels={'imagen_label_key': 'upscale_image'}
-            ),
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_upscale_gcs',
-        exception_if_mldev=(
-            'only supported in Gemini Enterprise Agent Platform'
-        ),
-        parameters=types.UpscaleImageParameters(
-            model=IMAGEN_MODEL_LATEST,
-            image=types.Image.from_file(location=IMAGE_FILE_PATH),
-            upscale_factor='x2',
-            config=types.UpscaleImageConfig(
-                output_gcs_uri='gs://genai-sdk-tests/temp/images/',
-            ),
-        ),
-    ),
-]
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.upscale_image',
-    test_table=test_table,
-)
-
-
-def test_upscale_extra_config_parameters(client):
-  # MLDev currently does not support upscale_image, but the ValidationError
-  # occurs before the ValueError.
-  try:
-    # User is not allowed to set mode or number_of_images
-    client.models.upscale_image(
-        model=IMAGEN_MODEL_LATEST,
-        image=types.Image.from_file(location=IMAGE_FILE_PATH),
-        upscale_factor='x2',
-        config={
-            'mode': 'upscale',
-            'number_of_images': 1,
-        }
-    )
-    # Should never reach this.
-    assert False
-  except Exception as e:
-    assert isinstance(e, ValidationError)
-    assert 'Extra inputs are not permitted' in str(e)
-
-
-@pytest.mark.asyncio
-async def test_upscale_async(client):
-  with pytest_helper.exception_if_mldev(client, ValueError):
-    response = await client.aio.models.upscale_image(
-        model=IMAGEN_MODEL_LATEST,
-        image=types.Image.from_file(location=IMAGE_FILE_PATH),
-        upscale_factor='x2',
-        config=types.UpscaleImageConfig(
-            person_generation=types.PersonGeneration.ALLOW_ADULT,
-            safety_filter_level=types.SafetyFilterLevel.BLOCK_LOW_AND_ABOVE,
-            include_rai_reason=True,
-            output_mime_type='image/jpeg',
-            output_compression_quality=80,
-            enhance_input_image=True,
-            image_preservation_factor=0.6,
-        ),
-    )
-    assert response.generated_images[0].image.image_bytes
-
-
-@pytest.mark.asyncio
-async def test_upscale_gcs_async(client):
-  with pytest_helper.exception_if_mldev(client, ValueError):
-    response = await client.aio.models.upscale_image(
-        model=IMAGEN_MODEL_LATEST,
-        image=types.Image.from_file(location=IMAGE_FILE_PATH),
-        upscale_factor='x2',
-        config=types.UpscaleImageConfig(
-            output_gcs_uri='gs://genai-sdk-tests/temp/images/',
-        ),
-    )
-    assert response.generated_images[0].image.gcs_uri
-
-
-@pytest.mark.asyncio
-async def test_upscale_extra_config_parameters_async(client):
-  # MLDev currently does not support upscale_image, but the ValidationError
-  # occurs before the ValueError.
-  try:
-    # User is not allowed to set mode or number_of_images
-    await client.aio.models.upscale_image(
-        model=IMAGEN_MODEL_LATEST,
-        image=types.Image.from_file(location=IMAGE_FILE_PATH),
-        upscale_factor='x2',
-        config={
-            'mode': 'upscale',
-            'number_of_images': 1,
-        },
-    )
-    # Should never reach this.
-    assert False
-  except Exception as e:
-    assert isinstance(e, ValidationError)
-    assert 'Extra inputs are not permitted' in str(e)
diff --git a/google/genai/tests/shared/models/test_upscale_image.py b/google/genai/tests/shared/models/test_upscale_image.py
deleted file mode 100644
index 676254d4e..000000000
--- a/google/genai/tests/shared/models/test_upscale_image.py
+++ /dev/null
@@ -1,54 +0,0 @@
-# Copyright 2025 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# 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.
-#
-
-"""Tests for upscale_image."""
-
-import os
-
-from .... import types
-from ... import pytest_helper
-
-IMAGEN_MODEL_LATEST = 'imagen-4.0-upscale-preview'
-
-IMAGE_FILE_PATH = os.path.abspath(
-    os.path.join(os.path.dirname(__file__), '../../data/bridge1.png')
-)
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_upscale',
-        exception_if_mldev=(
-            'only supported in Gemini Enterprise Agent Platform'
-        ),
-        parameters=types.UpscaleImageParameters(
-            model=IMAGEN_MODEL_LATEST,
-            image=types.Image.from_file(location=IMAGE_FILE_PATH),
-            upscale_factor='x2',
-            config=types.UpscaleImageConfig(
-                include_rai_reason=True,
-                output_mime_type='image/jpeg',
-                output_compression_quality=80,
-                enhance_input_image=True,
-                image_preservation_factor=0.6,
-            ),
-        ),
-    ),
-]
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.upscale_image',
-    test_table=test_table,
-)