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

Add get_query_segment_info utility wrapper (#11447) 

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
pull/11461/head
zhuwenxing 2021-11-08 20:46:26 +08:00 committed by GitHub
parent 9f1b8dc54f
commit fa6b9f89da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/python_client/base/utility_wrapper.py
View File

@ -14,6 +14,14 @@ class ApiUtilityWrapper:
ut = utility
def get_query_segment_info(self, collection_name, timeout=None, using="default", check_task=None, check_items=None):
timeout = TIMEOUT if timeout is None else timeout
func_name = sys._getframe().f_code.co_name
res, is_succ = api_request([self.ut.get_query_segment_info, collection_name, timeout, using])
check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ,
collection_name=collection_name, timeout=timeout, using=using).run()
return res, check_result
def loading_progress(self, collection_name, partition_names=None,
using="default", check_task=None, check_items=None):
func_name = sys._getframe().f_code.co_name