milvus/tests/python_client/deploy/scripts/action_after_upgrade.py
zhuwenxing 7feadf02e4
[skip e2e]Update deploy test scripts (#13771)
Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
2021-12-20 19:22:14 +08:00

37 lines
1.0 KiB
Python

from pymilvus import connections
from utils import *
def task_1():
"""
task_1:
before upgrade: create collection and insert data, load and search
after upgrade: get collection, load, search, create index, load, and search
"""
prefix = "task_1_"
connections.connect(host="127.0.0.1", port=19530, timeout=60)
get_collections(prefix)
load_and_search(prefix)
create_collections_and_insert_data(prefix)
create_index(prefix)
load_and_search(prefix)
def task_2():
"""
task_2:
before upgrade: create collection, insert data and create index, load and search
after upgrade: get collection, load, search, insert data, create index, load, and search
"""
prefix = "task_2_"
connections.connect(host="127.0.0.1", port=19530, timeout=60)
get_collections(prefix)
load_and_search(prefix)
create_collections_and_insert_data(prefix)
create_index(prefix)
load_and_search(prefix)
if __name__ == '__main__':
task_1()
task_2()