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

add load_collection case with partition (#4507) 

* add load_collection with partition

Signed-off-by: zw <zw@milvus.io>

* add filebeat in CI

Signed-off-by: zw <zw@milvus.io>

* remove CI on ubuntu

Signed-off-by: zw <zw@milvus.io>

Co-authored-by: zw <zw@milvus.io>
pull/4553/head
del-zhenwu 2020-12-25 10:21:03 +08:00 committed by GitHub
parent 89bbe1ee4d
commit 216d8d54be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 45 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ci/jenkins/Jenkinsfile vendored
View File

@ -40,7 +40,7 @@ pipeline {
axes {
axis {
name 'OS_NAME'
values 'ubuntu18.04', 'centos7'
values 'centos7'
}
axis {

4
ci/jenkins/step/shardsDevNightlyTest.groovy
View File

@ -10,9 +10,9 @@ timeout(time: 180, unit: 'MINUTES') {
try {
dir ('charts/milvus') {
if ("${BINARY_VERSION}" == "CPU") {
sh "helm install --wait --timeout 300s --set cluster.enabled=true --set persistence.enabled=true --set image.repository=registry.zilliz.com/milvus/engine --set mishards.image.tag=test --set mishards.image.pullPolicy=Always --set image.tag=${DOCKER_VERSION} --set image.pullPolicy=Always --set service.type=ClusterIP --set image.resources.requests.memory=8Gi --set image.resources.requests.cpu=2.0 --set image.resources.limits.memory=12Gi --set image.resources.limits.cpu=4.0 -f ci/db_backend/mysql_${BINARY_VERSION}_values.yaml --namespace milvus ${env.SHARDS_HELM_RELEASE_NAME} ."
sh "helm install --wait --timeout 300s --set cluster.enabled=true --set persistence.enabled=true --set image.repository=registry.zilliz.com/milvus/engine --set mishards.image.tag=test --set mishards.image.pullPolicy=Always --set image.tag=${DOCKER_VERSION} --set image.pullPolicy=Always --set service.type=ClusterIP --set image.resources.requests.memory=8Gi --set image.resources.requests.cpu=2.0 --set image.resources.limits.memory=12Gi --set image.resources.limits.cpu=4.0 -f ci/db_backend/mysql_${BINARY_VERSION}_values.yaml -f ci/filebeat/values.yaml --namespace milvus ${env.SHARDS_HELM_RELEASE_NAME} ."
} else {
sh "helm install --wait --timeout 300s --set cluster.enabled=true --set persistence.enabled=true --set image.repository=registry.zilliz.com/milvus/engine --set mishards.image.tag=test --set mishards.image.pullPolicy=Always --set gpu.enabled=true --set readonly.gpu.enabled=true --set image.tag=${DOCKER_VERSION} --set image.pullPolicy=Always --set service.type=ClusterIP -f ci/db_backend/mysql_${BINARY_VERSION}_values.yaml --namespace milvus ${env.SHARDS_HELM_RELEASE_NAME} ."
sh "helm install --wait --timeout 300s --set cluster.enabled=true --set persistence.enabled=true --set image.repository=registry.zilliz.com/milvus/engine --set mishards.image.tag=test --set mishards.image.pullPolicy=Always --set gpu.enabled=true --set readonly.gpu.enabled=true --set image.tag=${DOCKER_VERSION} --set image.pullPolicy=Always --set service.type=ClusterIP -f ci/db_backend/mysql_${BINARY_VERSION}_values.yaml -f ci/filebeat/values.yaml --namespace milvus ${env.SHARDS_HELM_RELEASE_NAME} ."
}
}
} catch (exc) {

42
tests/milvus_python_test/collection/test_collection.py
View File

@ -864,6 +864,48 @@ class TestCollection:
status = connect.load_collection(collection_name)
assert not status.OK()
@pytest.mark.level(1)
def test_load_collection_partition(self, connect, collection):
partition_name = gen_unique_str()
status, ids = connect.insert(collection, vectors)
assert status.OK()
status = connect.create_partition(collection, partition_name)
status = connect.load_collection(collection, partition_tags=[partition_name])
assert status.OK()
@pytest.mark.level(1)
def test_load_collection_partitions(self, connect, collection):
partition_names = []
for i in range(2):
name = gen_unique_str()
partition_names.append(name)
status = connect.create_partition(collection, name)
assert status.OK()
status, ids = connect.insert(collection, vectors)
status = connect.load_collection(collection, partition_tags=partition_names)
assert status.OK()
@pytest.mark.level(1)
def test_load_collection_partition_not_existed(self, connect, collection):
partition_name = gen_unique_str()
status, ids = connect.insert(collection, vectors)
assert status.OK()
status = connect.load_collection(collection, partition_tags=[partition_name])
assert not status.OK()
@pytest.mark.level(1)
def test_load_collection_partition_invalid_string(self, connect, collection):
partition_name = "invalid string"
status, ids = connect.insert(collection, vectors)
assert status.OK()
status = connect.load_collection(collection, partition_tags=[partition_name])
assert not status.OK()
@pytest.mark.level(1)
def test_load_collection_partition_None(self, connect, collection):
status = connect.load_collection(collection, partition_tags=None)
assert status.OK()
@pytest.mark.level(2)
def test_load_collection_not_existed_ip(self, connect, ip_collection, get_simple_index):
index_param = get_simple_index["index_param"]