Remove GCS test

Author: dishaprakash

tests/unit_tests/v2/test_async_pg_vectorstore.py

@@ -1,7 +1,6 @@
 import os
 import uuid
 from typing import AsyncIterator, Sequence
-from unittest.mock import MagicMock, patch
 
 import pytest
 import pytest_asyncio
@@ -124,14 +123,13 @@ async def image_uris(self) -> AsyncIterator[list[str]]:
         red_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_red.jpg"
         green_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_green.jpg"
         blue_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_blue.jpg"
-        gcs_uri = "gs://github-repo/img/vision/google-cloud-next.jpeg"
         image = Image.new("RGB", (100, 100), color="red")
         image.save(red_uri)
         image = Image.new("RGB", (100, 100), color="green")
         image.save(green_uri)
         image = Image.new("RGB", (100, 100), color="blue")
         image.save(blue_uri)
-        image_uris = [red_uri, green_uri, blue_uri, gcs_uri]
+        image_uris = [red_uri, green_uri, blue_uri]
         yield image_uris
         for uri in image_uris:
             try:
@@ -217,25 +215,12 @@ async def test_adelete(self, engine: PGEngine, vs: AsyncPGVectorStore) -> None:
         assert result == False
         await aexecute(engine, f'TRUNCATE TABLE "{DEFAULT_TABLE}"')
 
-    @patch("langchain_postgres.v2.async_vectorstore.storage.Client")
     async def test_aadd_images(
         self,
-        MockStorageClient: MagicMock,
         engine: PGEngine,
         image_vs: AsyncPGVectorStore,
         image_uris: list[str],
     ) -> None:
-        mock_blob_instance = MagicMock()
-        fake_image_bytes = b"fake_gcs_image_data"  # Differentiated fake data
-        mock_blob_instance.download_as_bytes.return_value = fake_image_bytes
-
-        mock_bucket_instance = MagicMock()
-        mock_bucket_instance.blob.return_value = mock_blob_instance
-
-        mock_storage_client_instance = MagicMock()
-        mock_storage_client_instance.bucket.return_value = mock_bucket_instance
-
-        MockStorageClient.return_value = mock_storage_client_instance
         ids = [str(uuid.uuid4()) for i in range(len(image_uris))]
         metadatas = [
             {"image_id": str(i), "source": "postgres"} for i in range(len(image_uris))

tests/unit_tests/v2/test_async_pg_vectorstore_search.py

@@ -165,14 +165,13 @@ async def image_uris(self) -> AsyncIterator[list[str]]:
         red_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_red.jpg"
         green_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_green.jpg"
         blue_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_blue.jpg"
-        gcs_uri = "gs://github-repo/img/vision/google-cloud-next.jpeg"
         image = Image.new("RGB", (100, 100), color="red")
         image.save(red_uri)
         image = Image.new("RGB", (100, 100), color="green")
         image.save(green_uri)
         image = Image.new("RGB", (100, 100), color="blue")
         image.save(blue_uri)
-        image_uris = [red_uri, green_uri, blue_uri, gcs_uri]
+        image_uris = [red_uri, green_uri, blue_uri]
         yield image_uris
         for uri in image_uris:
             try:
@@ -191,7 +190,7 @@ async def image_vs(
             table_name=IMAGE_TABLE,
             distance_strategy=DistanceStrategy.COSINE_DISTANCE,
         )
-        await vs.aadd_images(image_uris, ids=ids)
+        await vs.aadd_images(image_uris, ids=ids[:3])
         yield vs
 
     async def test_asimilarity_search_score(self, vs: AsyncPGVectorStore) -> None:

tests/unit_tests/v2/test_pg_vectorstore.py

@@ -144,14 +144,13 @@ def image_uris(self) -> Iterator[list[str]]:
         red_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_red.jpg"
         green_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_green.jpg"
         blue_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_blue.jpg"
-        gcs_uri = "gs://github-repo/img/vision/google-cloud-next.jpeg"
         image = Image.new("RGB", (100, 100), color="red")
         image.save(red_uri)
         image = Image.new("RGB", (100, 100), color="green")
         image.save(green_uri)
         image = Image.new("RGB", (100, 100), color="blue")
         image.save(blue_uri)
-        image_uris = [red_uri, green_uri, blue_uri, gcs_uri]
+        image_uris = [red_uri, green_uri, blue_uri]
         yield image_uris
         for uri in image_uris:
             try:

tests/unit_tests/v2/test_pg_vectorstore_search.py

@@ -175,14 +175,13 @@ async def image_uris(self) -> AsyncIterator[list[str]]:
         red_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_red.jpg"
         green_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_green.jpg"
         blue_uri = str(uuid.uuid4()).replace("-", "_") + "test_image_blue.jpg"
-        gcs_uri = "gs://github-repo/img/vision/google-cloud-next.jpeg"
         image = Image.new("RGB", (100, 100), color="red")
         image.save(red_uri)
         image = Image.new("RGB", (100, 100), color="green")
         image.save(green_uri)
         image = Image.new("RGB", (100, 100), color="blue")
         image.save(blue_uri)
-        image_uris = [red_uri, green_uri, blue_uri, gcs_uri]
+        image_uris = [red_uri, green_uri, blue_uri]
         yield image_uris
         for uri in image_uris:
             try: