fixed test_retriever_tool.py

Izukimat · Mar 29, 2024 · 0d9694b · 0d9694b
1 parent f9e9a99
commit 0d9694b
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/llama-index-core/tests/tools/test_retriever_tool.py b/llama-index-core/tests/tools/test_retriever_tool.py
@@ -41,13 +41,16 @@ def test_retriever_tool() -> None:
     retriever = MockRetriever()
     retriever_tool = RetrieverTool.from_defaults(retriever=retriever)
     response_nodes = retriever_tool("hello world")
-    assert str(response_nodes) == "mock_hello world"
-    assert response_nodes.raw_output[0].node.text == "mock_hello world"
+    assert str(response_nodes) == "mock_hello world\n\n\n\n"
+    assert response_nodes.raw_output[0].node.text == "mock_hello world\n\n"
 
     # Test node_postprocessors
-    node_postprocessors = MockPostProcessor()
+    node_postprocessors = [MockPostProcessor()]
     pr_retriever_tool = RetrieverTool.from_defaults(
         retriever=retriever, node_postprocessors=node_postprocessors
     )
     pr_response_nodes = pr_retriever_tool("hello world")
-    assert str(pr_response_nodes) == "processed_mock_hello world"
+    assert str(pr_response_nodes) == "processed_mock_hello world\n\n\n\n"
+
+
+# test_retriever_tool()