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, - )