github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-22 11:00:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Test InMemoryVectorStore with RWAPI test suite (#23603)

Browse Source 
Add standard test suite to InMemoryVectorStore implementation.
This commit is contained in:
Eugene Yurtsev
2024-06-27 16:43:43 -04:00
committed by GitHub
parent da7beb1c38
commit 68f348357e
4 changed files with 42 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
libs/community/tests/unit_tests/vectorstores/test_inmemory.py
View File

@@ -1,6 +1,11 @@
from pathlib import Path
import pytest
from langchain_core.documents import Document
from langchain_standard_tests.integration_tests.vectorstores import (
AsyncReadWriteTestSuite,
ReadWriteTestSuite,
)
from langchain_community.vectorstores.inmemory import InMemoryVectorStore
from tests.integration_tests.vectorstores.fake_embeddings import (
@@ -8,6 +13,18 @@ from tests.integration_tests.vectorstores.fake_embeddings import (
)
class TestInMemoryReadWriteTestSuite(ReadWriteTestSuite):
@pytest.fixture
def vectorstore(self) -> InMemoryVectorStore:
return InMemoryVectorStore(embedding=self.get_embeddings())
class TestAsyncInMemoryReadWriteTestSuite(AsyncReadWriteTestSuite):
@pytest.fixture
async def vectorstore(self) -> InMemoryVectorStore:
return InMemoryVectorStore(embedding=self.get_embeddings())
async def test_inmemory() -> None:
"""Test end to end construction and search."""
store = await InMemoryVectorStore.afrom_texts(