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 (#9902) 

Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
pull/9911/head
binbin 2021-10-14 19:31:03 +08:00 committed by GitHub
parent 17ed9b6ce7
commit 1df7459346
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1085,7 +1085,7 @@ class TestUtilityBase(TestcaseBase):
collection_w_1, vectors_1, _, insert_ids_1 = self.init_collection_general(prefix_1, True, nb) collection_w_1, vectors_1, _, insert_ids_1 = self.init_collection_general(prefix_1, True, nb)
vectors_l = vectors[0].loc[:, default_field_name] vectors_l = vectors[0].loc[:, default_field_name]
vectors_r = vectors_1[0].loc[:, default_field_name] vectors_r = vectors_1[0].loc[:, default_field_name]
log.info("Extracting entities form collections for distance calculating") log.info("Extracting entities from collections for distance calculating")
op_l = {"ids": insert_ids, "collection": collection_w.name, op_l = {"ids": insert_ids, "collection": collection_w.name,
"field": default_field_name} "field": default_field_name}
op_r = {"ids": insert_ids_1, "collection": collection_w_1.name, op_r = {"ids": insert_ids_1, "collection": collection_w_1.name,
@ -1117,7 +1117,7 @@ class TestUtilityBase(TestcaseBase):
for i in range(middle): for i in range(middle):
vectors_r.append(vectors[middle + i]) vectors_r.append(vectors[middle + i])
op_l = {"float_vectors": vectors_l} op_l = {"float_vectors": vectors_l}
log.info("Extracting entities form collections for distance calculating") log.info("Extracting entities from collections for distance calculating")
op_r = {"ids": insert_ids[middle:], "collection": collection_w.name, op_r = {"ids": insert_ids[middle:], "collection": collection_w.name,
"field": default_field_name} "field": default_field_name}
params = {metric_field: metric, "sqrt": sqrt} params = {metric_field: metric, "sqrt": sqrt}
@ -1144,7 +1144,7 @@ class TestUtilityBase(TestcaseBase):
vectors = vectors[0].loc[:, default_field_name] vectors = vectors[0].loc[:, default_field_name]
vectors_l = vectors[:middle] vectors_l = vectors[:middle]
vectors_r = cf.gen_vectors(nb, default_dim) vectors_r = cf.gen_vectors(nb, default_dim)
log.info("Extracting entities form collections for distance calculating") log.info("Extracting entities from collections for distance calculating")
op_l = {"ids": insert_ids[:middle], "collection": collection_w.name, op_l = {"ids": insert_ids[:middle], "collection": collection_w.name,
"field": default_field_name} "field": default_field_name}
op_r = {"float_vectors": vectors_r} op_r = {"float_vectors": vectors_r}
@ -1164,6 +1164,7 @@ class TestUtilityBase(TestcaseBase):
method: both left and right vectors are from partition method: both left and right vectors are from partition
expected: distance calculated successfully expected: distance calculated successfully
""" """
log.info("Creating 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)
@ -1171,12 +1172,14 @@ class TestUtilityBase(TestcaseBase):
middle = len(insert_ids) // 2 middle = len(insert_ids) // 2
params = {metric_field: metric, "sqrt": sqrt} params = {metric_field: metric, "sqrt": sqrt}
for i in range(len(partitions)): for i in range(len(partitions)):
log.info("Extracting entities from partitions for distance calculating")
vectors_l = vectors[i].loc[:, default_field_name] vectors_l = vectors[i].loc[:, default_field_name]
vectors_r = vectors[i].loc[:, default_field_name] vectors_r = vectors[i].loc[:, default_field_name]
op_l = {"ids": insert_ids[:middle], "collection": collection_w.name, op_l = {"ids": insert_ids[:middle], "collection": collection_w.name,
"partition": partitions[i].name, "field": default_field_name} "partition": partitions[i].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[i].name, "field": default_field_name} "partition": partitions[i].name, "field": default_field_name}
log.info("Calculating distance between entities from one partition")
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,