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

[skip ci] Add test logs (#10494) 

Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
pull/10511/head
binbin 2021-10-23 16:48:00 +08:00 committed by GitHub
parent 1019d0a3da
commit 879ed56d20
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/python_client/testcases/test_utility.py
View File

@ -1260,6 +1260,7 @@ class TestUtilityBase(TestcaseBase):
method: set right vectors as random vectors, left vectors are entities method: set right vectors as random vectors, left vectors are entities
expected: distance calculated successfully expected: distance calculated successfully
""" """
log.info("Create connection")
self._connect() self._connect()
nb = 10 nb = 10
collection_w, vectors, _, insert_ids, _ = self.init_collection_general(prefix, True, nb, partition_num=1) collection_w, vectors, _, insert_ids, _ = self.init_collection_general(prefix, True, nb, partition_num=1)
@ -1272,10 +1273,12 @@ class TestUtilityBase(TestcaseBase):
end = middle end = middle
for i in range(len(partitions)): for i in range(len(partitions)):
vectors_l = vectors[i].loc[:, default_field_name] vectors_l = vectors[i].loc[:, default_field_name]
log.info("Extract entities from partition %d as left vector" % i)
op_l = {"ids": insert_ids[start:end], "collection": collection_w.name, op_l = {"ids": insert_ids[start:end], "collection": collection_w.name,
"partition": partitions[i].name, "field": default_field_name} "partition": partitions[i].name, "field": default_field_name}
start += middle start += middle
end += middle end += middle
log.info("Calculate distance between vector and entities from partition %d" % i)
self.utility_wrap.calc_distance(op_l, op_r, params, self.utility_wrap.calc_distance(op_l, op_r, params,
check_task=CheckTasks.check_distance, check_task=CheckTasks.check_distance,
check_items={"vectors_l": vectors_l, check_items={"vectors_l": vectors_l,