From 76ea1b37d8f14ee9072d8863ec5884881a121e18 Mon Sep 17 00:00:00 2001 From: "Lacewell, Chaunte W" Date: Tue, 10 Dec 2024 15:12:56 -0800 Subject: [PATCH] Update VDMS test tp use new VectorStoreIntegrationTests --- .../vectorstores/test_vdms.py | 22 ++----------------- 1 file changed, 2 insertions(+), 20 deletions(-) diff --git a/libs/community/tests/integration_tests/vectorstores/test_vdms.py b/libs/community/tests/integration_tests/vectorstores/test_vdms.py index e0c27d0642ccc..7d6f71a38427a 100644 --- a/libs/community/tests/integration_tests/vectorstores/test_vdms.py +++ b/libs/community/tests/integration_tests/vectorstores/test_vdms.py @@ -7,10 +7,7 @@ import uuid import pytest -from langchain_tests.integration_tests.vectorstores import ( - AsyncReadWriteTestSuite, - ReadWriteTestSuite, -) +from langchain_tests.integration_tests.vectorstores import VectorStoreIntegrationTests from langchain_community.vectorstores.vdms import VDMS, VDMS_Client @@ -22,7 +19,7 @@ # docker run -d -p $VDMS_DBPORT:55555 intellabs/vdms:latest -class TestVDMSReadWriteTestSuite(ReadWriteTestSuite): +class TestVDMSStandard(VectorStoreIntegrationTests): @pytest.fixture def vectorstore(self) -> VDMS: test_name = uuid.uuid4().hex @@ -35,18 +32,3 @@ def vectorstore(self) -> VDMS: embedding=self.get_embeddings(), collection_name=test_name, ) - - -class TestAsyncVDMSReadWriteTestSuite(AsyncReadWriteTestSuite): - @pytest.fixture - async def vectorstore(self) -> VDMS: - test_name = uuid.uuid4().hex - client = VDMS_Client( - host=os.getenv("VDMS_DBHOST", "localhost"), - port=int(os.getenv("VDMS_DBPORT", 6025)), - ) - return VDMS( - client=client, - embedding=self.get_embeddings(), - collection_name=test_name, - )