From f29a5d4bcc4953e5b4f3609fc8daa6ef07a8492a Mon Sep 17 00:00:00 2001 From: shibuiwilliam Date: Wed, 19 Jul 2023 01:52:11 +0900 Subject: [PATCH] add test for knn retriever (#7769) # What - This is to add test for knn retriever. --------- Co-authored-by: Bagatur --- tests/unit_tests/retrievers/test_knn.py | 11 +++++++++++ 1 file changed, 11 insertions(+) create mode 100644 tests/unit_tests/retrievers/test_knn.py diff --git a/tests/unit_tests/retrievers/test_knn.py b/tests/unit_tests/retrievers/test_knn.py new file mode 100644 index 00000000000..e0486d7efa7 --- /dev/null +++ b/tests/unit_tests/retrievers/test_knn.py @@ -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