diff --git a/tests/test_async_vectorstore_index.py b/tests/test_async_vectorstore_index.py index a3ff8c12..68bc4e72 100644 --- a/tests/test_async_vectorstore_index.py +++ b/tests/test_async_vectorstore_index.py @@ -109,6 +109,7 @@ async def test_aapply_vector_index(self, vs): index = HNSWIndex() await vs.aapply_vector_index(index) assert await vs.is_valid_index(DEFAULT_INDEX_NAME) + await vs.adrop_vector_index() async def test_areindex(self, vs): if not await vs.is_valid_index(DEFAULT_INDEX_NAME): @@ -117,6 +118,7 @@ async def test_areindex(self, vs): await vs.areindex() await vs.areindex(DEFAULT_INDEX_NAME) assert await vs.is_valid_index(DEFAULT_INDEX_NAME) + await vs.adrop_vector_index() async def test_dropindex(self, vs): await vs.adrop_vector_index() @@ -134,6 +136,7 @@ async def test_aapply_vector_index_ivfflat(self, vs): await vs.aapply_vector_index(index) assert await vs.is_valid_index("secondindex") await vs.adrop_vector_index("secondindex") + await vs.adrop_vector_index() async def test_is_valid_index(self, vs): is_valid = await vs.is_valid_index("invalid_index") diff --git a/tests/test_vectorstore_index.py b/tests/test_vectorstore_index.py index a6b5af92..cb797219 100644 --- a/tests/test_vectorstore_index.py +++ b/tests/test_vectorstore_index.py @@ -114,6 +114,7 @@ async def test_apply_vector_index(self, vs): index = HNSWIndex() vs.apply_vector_index(index) assert vs.is_valid_index(DEFAULT_INDEX_NAME) + vs.drop_vector_index() async def test_areindex(self, vs): if not vs.is_valid_index(DEFAULT_INDEX_NAME): @@ -122,6 +123,7 @@ async def test_areindex(self, vs): vs.reindex() vs.reindex(DEFAULT_INDEX_NAME) assert vs.is_valid_index(DEFAULT_INDEX_NAME) + vs.drop_vector_index(DEFAULT_INDEX_NAME) async def test_dropindex(self, vs): vs.drop_vector_index() @@ -139,6 +141,7 @@ async def test_apply_vector_index_ivfflat(self, vs): vs.apply_vector_index(index) assert vs.is_valid_index("secondindex") vs.drop_vector_index("secondindex") + vs.drop_vector_index() async def test_is_valid_index(self, vs): is_valid = vs.is_valid_index("invalid_index") @@ -192,6 +195,7 @@ async def test_aapply_vector_index(self, vs): index = HNSWIndex() await vs.aapply_vector_index(index) assert await vs.ais_valid_index(DEFAULT_INDEX_NAME) + await vs.adrop_vector_index() async def test_areindex(self, vs): if not await vs.ais_valid_index(DEFAULT_INDEX_NAME): @@ -200,6 +204,7 @@ async def test_areindex(self, vs): await vs.areindex() await vs.areindex(DEFAULT_INDEX_NAME) assert await vs.ais_valid_index(DEFAULT_INDEX_NAME) + await vs.adrop_vector_index(DEFAULT_INDEX_NAME) async def test_dropindex(self, vs): await vs.adrop_vector_index() @@ -217,6 +222,7 @@ async def test_aapply_vector_index_ivfflat(self, vs): await vs.aapply_vector_index(index) assert await vs.ais_valid_index("secondindex") await vs.adrop_vector_index("secondindex") + await vs.adrop_vector_index() async def test_is_valid_index(self, vs): is_valid = await vs.ais_valid_index("invalid_index")