admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-08 01:58:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

[skip e2e]Add verification for search and query result (#18006)

Browse Source 
Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
This commit is contained in:
zhuwenxing 2022-07-01 17:34:19 +08:00 committed by GitHub
parent e3807b97c1
commit 72364aa6a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/python_client/deploy/scripts/utils.py
View File

@ -157,8 +157,9 @@ def load_and_search(prefix, replicas=1):
start_time = time.time() start_time = time.time()
print(f"\nSearch...") print(f"\nSearch...")
# define output_fields of search result # define output_fields of search result
v_search = vectors[:1]
res = c.search( res = c.search(
vectors[:1], "float_vector", search_params, topK, v_search, "float_vector", search_params, topK,
"count > 500", output_fields=["count", "random_value"], timeout=120 "count > 500", output_fields=["count", "random_value"], timeout=120
) )
end_time = time.time() end_time = time.time()
@ -168,7 +169,9 @@ def load_and_search(prefix, replicas=1):
# Get value of the random value field for search result # Get value of the random value field for search result
print(hit, hit.entity.get("random_value")) print(hit, hit.entity.get("random_value"))
ids = hits.ids ids = hits.ids
assert len(ids) == topK
print(ids) print(ids)
assert len(res) == len(v_search)
print("search latency: %.4fs" % (end_time - start_time)) print("search latency: %.4fs" % (end_time - start_time))
t0 = time.time() t0 = time.time()
expr = "count in [2,4,6,8]" expr = "count in [2,4,6,8]"
@ -178,6 +181,7 @@ def load_and_search(prefix, replicas=1):
for r in sorted_res: for r in sorted_res:
print(r) print(r)
t1 = time.time() t1 = time.time()
assert len(res) == 4
print("query latency: %.4fs" % (t1 - t0)) print("query latency: %.4fs" % (t1 - t0))
# c.release() # c.release()
print("###########") print("###########")