MCPcopy Create free account
hub / github.com/SkyworkAI/DeepResearchAgent / test_search_functionality

Method test_search_functionality

tests/test_faiss.py:120–173  ·  view source on GitHub ↗

Test search functionality.

(self)

Source from the content-addressed store, hash-verified

118 f"Expected >=4 documents, got {index_info.total_documents}")
119
120 async def test_search_functionality(self):
121 """Test search functionality."""
122 print("\n=== Testing Search Functionality ===")
123
124 # Test basic search
125 query = "machine learning"
126 request = FaissSearchRequest(
127 query=query,
128 k=3,
129 fetch_k=10
130 )
131 result = await self.faiss_service.search_similar(request)
132
133 self.assert_test(len(result.documents) > 0,
134 "Basic search",
135 f"Expected documents, got docs={len(result.documents)}")
136
137 # Test search with filter
138 request = FaissSearchRequest(
139 query=query,
140 k=2,
141 filter={"category": "AI"},
142 fetch_k=10
143 )
144 result = await self.faiss_service.search_similar(request)
145
146 self.assert_test(True,
147 "Search with filter",
148 "Search completed")
149
150 # Test search with score threshold
151 request = FaissSearchRequest(
152 query=query,
153 k=5,
154 score_threshold=0.1,
155 fetch_k=10
156 )
157 result = await self.faiss_service.search_similar(request)
158
159 self.assert_test(True,
160 "Search with score threshold",
161 "Search completed")
162
163 # Test search with different query
164 request = FaissSearchRequest(
165 query="programming language",
166 k=2,
167 fetch_k=10
168 )
169 result = await self.faiss_service.search_similar(request)
170
171 self.assert_test(True,
172 "Search with different query",
173 "Search completed")
174
175 async def test_document_deletion(self):
176 """Test document deletion."""

Callers 1

run_all_testsMethod · 0.95

Calls 4

assert_testMethod · 0.95
FaissSearchRequestClass · 0.90
printFunction · 0.85
search_similarMethod · 0.45

Tested by

no test coverage detected