admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: modify the hybrid search test case (#31702)

Browse Source 
issue: #31339

Signed-off-by: binbin lv <binbin.lv@zilliz.com>
This commit is contained in:
binbin 2024-03-29 10:47:17 +08:00 committed by GitHub
parent fe2f34d76f
commit c15beecb96
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tests/python_client/testcases/test_search.py
View File

@ -10802,21 +10802,30 @@ class TestCollectionHybridSearchValid(TestcaseBase):
collection_w.load()
# 3. prepare search params
req_list = []
id_list = []
for i in range(len(vector_name_list)):
vectors = [[random.random() for _ in range(multiple_dim_array[i])] for _ in range(1)]
search_params = {"metric_type": metric_type, "offset": 0}
search_param = {
"data": [[random.random() for _ in range(multiple_dim_array[i])] for _ in range(1)],
"data": vectors,
"anns_field": vector_name_list[i],
"param": {"metric_type": metric_type, "offset": 0},
"param": search_params,
"limit": min_dim,
"expr": "int64 > 0"}
"expr": default_search_exp}
req = AnnSearchRequest(**search_param)
req_list.append(req)
search_res = collection_w.search(vectors[:1], vector_name_list[i],
search_params, min_dim,
default_search_exp,
check_task=CheckTasks.check_search_results,
check_items={"nq": 1,
"ids": insert_ids,
"limit": min_dim})[0]
id_list.extend(search_res[0].ids)
# 4. hybrid search
collection_w.hybrid_search(req_list, WeightedRanker(0.1, 0.9), default_limit,
check_task=CheckTasks.check_search_results,
check_items={"nq": 1,
"ids": insert_ids,
"limit": min_dim*len(vector_name_list)})
hybrid_search = collection_w.hybrid_search(req_list, WeightedRanker(0.1, 0.9), default_limit)[0]
assert len(hybrid_search) == 1
assert len(hybrid_search[0].ids) == len(list(set(id_list)))
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.parametrize("primary_field", [ct.default_int64_field_name, ct.default_string_field_name])