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

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

View File

@@ -78,6 +78,7 @@ def test_test_group_dependencies(poetry_conf: Mapping[str, Any]) -> None:
"duckdb-engine",
"freezegun",
"langchain-core",
"langchain-standard-tests",
"langchain",
"lark",
"pandas",

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(