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

community[patch]: Add unit test to verify that init is correctly defined (#22030)

Browse Source 
Fix some __init__ files and add a unit test
This commit is contained in:
Eugene Yurtsev
2024-05-22 13:19:00 -04:00
committed by GitHub
parent ef53ccf54b
commit 58360a1e53
6 changed files with 102 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libs/community/tests/unit_tests/utilities/test_rememberizer.py
View File

@@ -23,7 +23,9 @@ class TestRememberizerAPIWrapper(unittest.TestCase):
]
},
)
wrapper = RememberizerAPIWrapper(rememberizer_api_key="dummy_key", n=10)
wrapper = RememberizerAPIWrapper(
rememberizer_api_key="dummy_key", top_k_results=10
)
result = wrapper.search("test")
self.assertEqual(
result,
@@ -44,7 +46,9 @@ class TestRememberizerAPIWrapper(unittest.TestCase):
status=400,
json={"detail": "Incorrect authentication credentials."},
)
wrapper = RememberizerAPIWrapper(rememberizer_api_key="dummy_key", n=10)
wrapper = RememberizerAPIWrapper(
rememberizer_api_key="dummy_key", top_k_results=10
)
with self.assertRaises(ValueError) as e:
wrapper.search("test")
self.assertEqual(
@@ -66,7 +70,9 @@ class TestRememberizerAPIWrapper(unittest.TestCase):
"document": {"id": "id2", "name": "name2"},
},
]
wrapper = RememberizerAPIWrapper(rememberizer_api_key="dummy_key", n=10)
wrapper = RememberizerAPIWrapper(
rememberizer_api_key="dummy_key", top_k_results=10
)
result = wrapper.load("test")
self.assertEqual(len(result), 2)
self.assertEqual(result[0].page_content, "content1")