test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-04 04:49:08 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

[skip e2e]Update functions of deploy test (#14226)

Browse Source 
Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
This commit is contained in:
zhuwenxing 2021-12-25 13:44:21 +08:00 committed by GitHub
parent 684a63b698
commit 7623a648ae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 25 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/python_client/deploy/scripts/action_after_reinstall.py
View File

@ -2,7 +2,7 @@ from pymilvus import connections
from utils import *
def task_1():
def task_1(data_size):
"""
task_1:
before reinstall: create collection and insert data, load and search
@ -12,10 +12,10 @@ def 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_collections_and_insert_data(prefix, data_size)
def task_2():
def task_2(data_zise):
"""
task_2:
before reinstall: create collection, insert data and create index, load and search
@ -25,11 +25,17 @@ def 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_collections_and_insert_data(prefix, data_zise)
create_index(prefix)
load_and_search(prefix)
if __name__ == '__main__':
task_1()
task_2()
import argparse
parser = argparse.ArgumentParser(description='config for deploy test')
parser.add_argument('--data_size', type=int, default=3000, help='data size')
args = parser.parse_args()
data_size = args.data_size
print(f"data size: {data_size}")
task_1(data_size)
task_2(data_size)

18
tests/python_client/deploy/scripts/action_after_upgrade.py
View File

@ -2,7 +2,7 @@ from pymilvus import connections
from utils import *
def task_1():
def task_1(data_size):
"""
task_1:
before upgrade: create collection and insert data, load and search
@ -12,12 +12,12 @@ def 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_collections_and_insert_data(prefix, data_size)
create_index(prefix)
load_and_search(prefix)
def task_2():
def task_2(data_size):
"""
task_2:
before upgrade: create collection, insert data and create index, load and search
@ -27,11 +27,17 @@ def 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_collections_and_insert_data(prefix, data_size)
create_index(prefix)
load_and_search(prefix)
if __name__ == '__main__':
task_1()
task_2()
import argparse
parser = argparse.ArgumentParser(description='config for deploy test')
parser.add_argument('--data_size', type=int, default=3000, help='data size')
args = parser.parse_args()
data_size = args.data_size
print(f"data size: {data_size}")
task_1(data_size)
task_2(data_size)

2
tests/python_client/deploy/scripts/utils.py
View File

@ -64,7 +64,7 @@ def get_collections(prefix):
return col_list
def create_collections_and_insert_data(prefix, count):
def create_collections_and_insert_data(prefix, count=3000):
import random
dim = 128
nb = count // 10