add test for knn retriever (#7769)

# What
- This is to add test for knn retriever.
---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
shibuiwilliam 2023-07-19 01:52:11 +09:00 committed by GitHub
parent 75d3f1e5e6
commit f29a5d4bcc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -0,0 +1,11 @@
from langchain.embeddings import FakeEmbeddings
from langchain.retrievers.knn import KNNRetriever
class TestKNNRetriever:
def test_from_texts(self) -> None:
input_texts = ["I have a pen.", "Do you have a pen?", "I have a bag."]
knn_retriever = KNNRetriever.from_texts(
texts=input_texts, embeddings=FakeEmbeddings(size=100)
)
assert len(knn_retriever.texts) == 3