milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: modify hybrid search cases (#31624) 

issue: #31339

Signed-off-by: binbin lv <binbin.lv@zilliz.com>
pull/31615/head^2
binbin 2024-03-27 12:05:09 +08:00 committed by GitHub
parent c299bdec1b
commit d995b3f0fa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -10710,21 +10710,31 @@ 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": default_limit,
"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, default_limit,
default_search_exp,
check_task=CheckTasks.check_search_results,
check_items={"nq": 1,
"ids": insert_ids,
"limit": default_limit})[0]
id_list.extend(search_res[0].ids)
# 4. hybrid search
collection_w.hybrid_search(req_list, WeightedRanker(0.1, 0.9), default_limit*len(req_list)+1,
check_task=CheckTasks.check_search_results,
check_items={"nq": 1,
"ids": insert_ids,
"limit": default_limit*len(req_list)})
hybrid_search = \
collection_w.hybrid_search(req_list, WeightedRanker(0.1, 0.9), default_limit * len(req_list) + 1)[0]
assert len(hybrid_search) == 1
assert len(hybrid_search[0].ids) == len(list(set(id_list)))
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.parametrize("primary_field", [ct.default_int64_field_name, ct.default_string_field_name])