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 logs for test case () 

Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
pull/10017/head
binbin 2021-10-16 17:20:33 +08:00 committed by GitHub
parent 76b9fe9dc3
commit a0b79bf3d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
tests/python_client/testcases

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

@ -1194,6 +1194,7 @@ class TestUtilityBase(TestcaseBase):
method: calculate distance between entities from two partitions method: calculate distance between entities from two partitions
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)
@ -1202,10 +1203,12 @@ class TestUtilityBase(TestcaseBase):
params = {metric_field: metric, "sqrt": sqrt} params = {metric_field: metric, "sqrt": sqrt}
vectors_l = vectors[0].loc[:, default_field_name] vectors_l = vectors[0].loc[:, default_field_name]
vectors_r = vectors[1].loc[:, default_field_name] vectors_r = vectors[1].loc[:, default_field_name]
log.info("Extract entities from two partitions for distance calculating")
op_l = {"ids": insert_ids[:middle], "collection": collection_w.name, op_l = {"ids": insert_ids[:middle], "collection": collection_w.name,
"partition": partitions[0].name, "field": default_field_name} "partition": partitions[0].name, "field": default_field_name}
op_r = {"ids": insert_ids[middle:], "collection": collection_w.name, op_r = {"ids": insert_ids[middle:], "collection": collection_w.name,
"partition": partitions[1].name, "field": default_field_name} "partition": partitions[1].name, "field": default_field_name}
log.info("Calculate distance between entities from two partitions")
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,