Fix tests

Author: dishaprakash

tests/unit_tests/v2/test_async_pg_vectorstore_index.py

@@ -18,8 +18,9 @@
 )
 from tests.utils import VECTORSTORE_CONNECTION_STRING_ASYNCPG as CONNECTION_STRING
 
-DEFAULT_TABLE = "default" + str(uuid.uuid4()).replace("-", "_")
-DEFAULT_INDEX_NAME = DEFAULT_TABLE + "index"
+uuid_str = str(uuid.uuid4()).replace("-", "_")
+DEFAULT_TABLE = "default" + uuid_str
+DEFAULT_INDEX_NAME = uuid_str + "index"
 VECTOR_SIZE = 768
 
 embeddings_service = DeterministicFakeEmbedding(size=VECTOR_SIZE)
@@ -67,14 +68,14 @@ async def vs(self, engine: PGEngine) -> AsyncIterator[AsyncPGVectorStore]:
         )
 
         await vs.aadd_texts(texts, ids=ids)
-        await vs.adrop_vector_index()
+        await vs.adrop_vector_index(DEFAULT_INDEX_NAME)
         yield vs
 
     async def test_aapply_vector_index(self, vs: AsyncPGVectorStore) -> None:
         index = HNSWIndex()
         await vs.aapply_vector_index(index)
         assert await vs.is_valid_index(DEFAULT_INDEX_NAME)
-        await vs.adrop_vector_index()
+        await vs.adrop_vector_index(DEFAULT_INDEX_NAME)
 
     async def test_areindex(self, vs: AsyncPGVectorStore) -> None:
         if not await vs.is_valid_index(DEFAULT_INDEX_NAME):
@@ -86,7 +87,7 @@ async def test_areindex(self, vs: AsyncPGVectorStore) -> None:
         await vs.adrop_vector_index(DEFAULT_INDEX_NAME)
 
     async def test_dropindex(self, vs: AsyncPGVectorStore) -> None:
-        await vs.adrop_vector_index()
+        await vs.adrop_vector_index(DEFAULT_INDEX_NAME)
         result = await vs.is_valid_index(DEFAULT_INDEX_NAME)
         assert not result
 
@@ -101,7 +102,7 @@ async def test_aapply_vector_index_ivfflat(self, vs: AsyncPGVectorStore) -> None
         await vs.aapply_vector_index(index)
         assert await vs.is_valid_index("secondindex")
         await vs.adrop_vector_index("secondindex")
-        await vs.adrop_vector_index()
+        await vs.adrop_vector_index(DEFAULT_INDEX_NAME)
 
     async def test_is_valid_index(self, vs: AsyncPGVectorStore) -> None:
         is_valid = await vs.is_valid_index("invalid_index")

tests/unit_tests/v2/test_pg_vectorstore_index.py

@@ -17,11 +17,13 @@
 )
 from tests.utils import VECTORSTORE_CONNECTION_STRING_ASYNCPG as CONNECTION_STRING
 
-DEFAULT_TABLE = "default" + str(uuid.uuid4()).replace("-", "_")
-DEFAULT_TABLE_ASYNC = "default_sync" + str(uuid.uuid4()).replace("-", "_")
+uuid_str = str(uuid.uuid4()).replace("-", "_")
+uuid_str_sync = str(uuid.uuid4()).replace("-", "_")
+DEFAULT_TABLE = "default" + uuid_str
+DEFAULT_TABLE_ASYNC = "default_sync" + uuid_str_sync
 CUSTOM_TABLE = "custom" + str(uuid.uuid4()).replace("-", "_")
-DEFAULT_INDEX_NAME = DEFAULT_TABLE + DEFAULT_INDEX_NAME_SUFFIX
-DEFAULT_INDEX_NAME_ASYNC = DEFAULT_TABLE_ASYNC + DEFAULT_INDEX_NAME_SUFFIX
+DEFAULT_INDEX_NAME = uuid_str + "index"
+DEFAULT_INDEX_NAME_ASYNC = uuid_str_sync + "index"
 VECTOR_SIZE = 768
 
 embeddings_service = DeterministicFakeEmbedding(size=VECTOR_SIZE)