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 new test case to benchmark (#8162) 

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
pull/8119/head
wt 2021-09-17 18:49:53 +08:00 committed by GitHub
parent f5da780c89
commit 374376fe08
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tests/benchmark/milvus_benchmark/client.py
View File

@ -457,6 +457,30 @@ class MilvusClient(object):
collection_name = self._collection_name
return self._milvus.release_partitions(collection_name, tag_names, timeout=timeout)
@time_wrapper
def scene_test(self, collection_name=None, vectors=None, ids=None):
logger.debug("[scene_test] Start scene test : %s" % collection_name)
self.create_collection(dimension=128, collection_name=collection_name)
time.sleep(1)
collection_info = self.get_info(collection_name)
entities = utils.generate_entities(collection_info, vectors, ids)
logger.debug("[scene_test] Start insert : %s" % collection_name)
self.insert(entities)
logger.debug("[scene_test] Start flush : %s" % collection_name)
self.flush()
logger.debug("[scene_test] Start create index : %s" % collection_name)
self.create_index(field_name='float_vector', index_type="ivf_sq8", metric_type='l2',
collection_name=collection_name, index_param={'nlist': 2048})
# time.sleep(59)
logger.debug("[scene_test] Start drop : %s" % collection_name)
self.drop(collection_name=collection_name)
logger.debug("[scene_test]Scene test close : %s" % collection_name)
# time.sleep(1)
# TODO: remove
# def get_server_version(self):
# return self._milvus.server_version()