Skip to content

Commit

Permalink
Update VDMS test tp use new VectorStoreIntegrationTests
Browse files Browse the repository at this point in the history
  • Loading branch information
cwlacewe committed Dec 10, 2024
1 parent 147ab06 commit 76ea1b3
Showing 1 changed file with 2 additions and 20 deletions.
22 changes: 2 additions & 20 deletions libs/community/tests/integration_tests/vectorstores/test_vdms.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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
Expand All @@ -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,
)

0 comments on commit 76ea1b3

Please sign in to comment.