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

Update test case for interface change (#5794) 

Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
pull/5809/head^2
binbin 2021-06-16 11:29:55 +08:00 committed by GitHub
parent 443084ae1b
commit c440b9e8a0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 23 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests20/python_client/base/collection_wrapper.py
View File

@ -77,11 +77,11 @@ class ApiCollectionWrapper:
log.info("Dropping collection failed")
return res, check_result
def load(self, field_names=None, index_names=None, partition_names=None, check_task=None, check_items=None, **kwargs):
def load(self, partition_names=None, check_task=None, check_items=None, **kwargs):
log.info("loading data")
func_name = sys._getframe().f_code.co_name
res, check = api_request([self.collection.load, field_names, index_names, partition_names], **kwargs)
check_result = ResponseChecker(res, func_name, check_task, check_items, check, field_names=field_names, index_names=index_names,
res, check = api_request([self.collection.load, partition_names], **kwargs)
check_result = ResponseChecker(res, func_name, check_task, check_items, check,
partition_names=partition_names, **kwargs).run()
if check_result:
log.info("loaded data")
@ -160,27 +160,27 @@ class ApiCollectionWrapper:
# check_result = CheckFunc(res, func_name, check_task, check_items, check).run()
# return res, check_result
def index(self, index_name="", check_task=None, check_items=None):
def index(self, check_task=None, check_items=None):
func_name = sys._getframe().f_code.co_name
res, check = api_request([self.collection.index, index_name])
check_result = ResponseChecker(res, func_name, check_task, check_items, check, index_name=index_name).run()
res, check = api_request([self.collection.index])
check_result = ResponseChecker(res, func_name, check_task, check_items, check).run()
return res, check_result
def create_index(self, field_name, index_params, index_name="", check_task=None, check_items=None, **kwargs):
def create_index(self, field_name, index_params, check_task=None, check_items=None, **kwargs):
func_name = sys._getframe().f_code.co_name
res, check = api_request([self.collection.create_index, field_name, index_params, index_name], **kwargs)
check_result = ResponseChecker(res, func_name, check_task, check_items, check, field_name=field_name, index_params=index_params,
index_name=index_name, **kwargs).run()
res, check = api_request([self.collection.create_index, field_name, index_params], **kwargs)
check_result = ResponseChecker(res, func_name, check_task, check_items, check,
field_name=field_name, index_params=index_params, **kwargs).run()
return res, check_result
def has_index(self, index_name="", check_task=None, check_items=None):
def has_index(self, check_task=None, check_items=None):
func_name = sys._getframe().f_code.co_name
res, check = api_request([self.collection.has_index, index_name])
check_result = ResponseChecker(res, func_name, check_task, check_items, check, index_name=index_name).run()
res, check = api_request([self.collection.has_index])
check_result = ResponseChecker(res, func_name, check_task, check_items, check).run()
return res, check_result
def drop_index(self, index_name="", check_task=None, check_items=None, **kwargs):
def drop_index(self, check_task=None, check_items=None, **kwargs):
func_name = sys._getframe().f_code.co_name
res, check = api_request([self.collection.drop_index, index_name], **kwargs)
check_result = ResponseChecker(res, func_name, check_task, check_items, check, index_name=index_name, **kwargs).run()
res, check = api_request([self.collection.drop_index], **kwargs)
check_result = ResponseChecker(res, func_name, check_task, check_items, check, **kwargs).run()
return res, check_result

13
tests20/python_client/base/index_wrapper.py
View File

@ -10,13 +10,14 @@ from utils.api_request import api_request
class ApiIndexWrapper:
index = None
def init_index(self, collection, field_name, index_params, name="",check_task=None, check_items=None, **kwargs):
def init_index(self, collection, field_name, index_params, check_task=None, check_items=None, **kwargs):
""" In order to distinguish the same name of index """
func_name = sys._getframe().f_code.co_name
res, is_succ = api_request([Index, collection, field_name, index_params, name], **kwargs)
res, is_succ = api_request([Index, collection, field_name, index_params], **kwargs)
self.index = res if is_succ is True else None
check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ, collection=collection, field_name=field_name,
index_params=index_params, name=name, **kwargs).run()
check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ,
collection=collection, field_name=field_name,
index_params=index_params, **kwargs).run()
return res, check_result
def drop(self, check_task=None, check_items=None, **kwargs):
@ -25,10 +26,6 @@ class ApiIndexWrapper:
check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ, **kwargs).run()
return res, check_result
@property
def name(self):
return self.index.name
@property
def params(self):
return self.index.params

6
tests20/python_client/base/partition_wrapper.py
View File

@ -57,13 +57,11 @@ class ApiPartitionWrapper:
check_result = ResponseChecker(res, func_name, check_task, check_items, succ, **kwargs).run()
return res, check_result
def load(self, field_names=None, index_names=None, check_task=None, check_items=None, **kwargs):
def load(self, check_task=None, check_items=None, **kwargs):
func_name = sys._getframe().f_code.co_name
res, succ = api_request([self.partition.load, field_names, index_names], **kwargs)
res, succ = api_request([self.partition.load], **kwargs)
check_result = ResponseChecker(res, func_name, check_task,
check_items, is_succ=succ,
field_names=field_names,
index_names=index_names,
**kwargs).run()
return res, check_result