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

Add 0331 tag 

Signed-off-by: 紫晴 <ting.wang@zilliz.com>
pull/4973/head^2
紫晴 2021-03-13 11:38:05 +08:00 committed by yefu.chen
parent 3bf617385a
commit 9d71afbf9c
2 changed files with 108 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

123
tests/python_test/collection/test_load_collection.py
View File

@ -38,6 +38,7 @@ class TestLoadCollection:
# @pytest.mark.tags("0331")
# TODO ci failed
@pytest.mark.tags("fail")
def test_load_collection_after_index(self, connect, collection, get_simple_index):
'''
target: test load collection, after index created
@ -52,6 +53,8 @@ class TestLoadCollection:
@pytest.mark.tags("0331")
@pytest.mark.level(2)
# todo can't load repeat
@pytest.mark.tags("fail")
def test_load_collection_after_index_binary(self, connect, binary_collection, get_binary_index):
'''
target: test load binary_collection, after index created
@ -110,16 +113,27 @@ class TestLoadCollection:
@pytest.mark.tags("0331")
def test_load_collection_not_existed(self, connect, collection):
collection_name = gen_unique_str(uid)
with pytest.raises(Exception) as e:
try:
connect.load_collection(collection_name)
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "describe collection failed: can't find collection: %s" % collection_name
@pytest.mark.level(2)
@pytest.mark.tags("0331")
def test_release_collection_not_existed(self, connect, collection):
collection_name = gen_unique_str(uid)
with pytest.raises(Exception) as e:
try:
connect.release_collection(collection_name)
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "describe collection failed: can't find collection: %s" % collection_name
@pytest.mark.tags("0331")
def test_release_collection_not_load(self, connect, collection):
"""
target: test release collection without load
@ -140,6 +154,7 @@ class TestLoadCollection:
connect.load_collection(collection)
@pytest.mark.level(2)
@pytest.mark.tags("0331")
def test_load_release_collection(self, connect, collection):
collection_name = gen_unique_str(uid)
connect.create_collection(collection_name, default_fields)
@ -148,11 +163,23 @@ class TestLoadCollection:
connect.load_collection(collection_name)
connect.release_collection(collection_name)
connect.drop_collection(collection_name)
with pytest.raises(Exception) as e:
try:
connect.load_collection(collection_name)
with pytest.raises(Exception) as e:
connect.release_collection(collection_name)
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "describe collection failed: can't find collection: %s" % collection_name
try:
connect.release_collection(collection_name)
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "describe collection failed: can't find collection: %s" % collection_name
@pytest.mark.tags("0331")
def test_release_collection_after_drop(self, connect, collection):
"""
target: test release collection after drop
@ -164,10 +191,16 @@ class TestLoadCollection:
connect.flush([collection])
connect.load_collection(collection)
connect.drop_collection(collection)
with pytest.raises(Exception) as e:
try:
connect.release_collection(collection)
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "describe collection failed: can't find collection: %s" % collection
# 1132
# TODO
@pytest.mark.tags("fail")
def test_load_collection_without_flush(self, connect, collection):
"""
target: test load collection without flush
@ -180,6 +213,7 @@ class TestLoadCollection:
connect.load_collection(collection)
# TODO
@pytest.mark.tags("fail")
def _test_load_collection_larger_than_memory(self):
"""
target: test load collection when memory less than collection size
@ -187,6 +221,7 @@ class TestLoadCollection:
expected: raise exception
"""
@pytest.mark.tags("fail")
def test_load_collection_release_part_partitions(self, connect, collection):
"""
target: test release part partitions after load collection
@ -206,6 +241,7 @@ class TestLoadCollection:
res = connect.search(collection, default_single_query, partition_tags=[default_partition_name])
assert len(res[0]) == default_top_k
@pytest.mark.tags("fail")
def test_load_collection_release_all_partitions(self, connect, collection):
"""
target: test release all partitions after load collection
@ -223,6 +259,7 @@ class TestLoadCollection:
res = connect.search(collection, default_single_query)
assert len(res[0]) == 0
@pytest.mark.tags("0331")
def test_load_partitions_release_collection(self, connect, collection):
"""
target: test release collection after load partitions
@ -242,6 +279,7 @@ class TestLoadCollection:
class TestReleaseAdvanced:
@pytest.mark.tags("fail")
def test_release_collection_during_searching(self, connect, collection):
"""
target: test release collection during searching
@ -260,6 +298,7 @@ class TestReleaseAdvanced:
connect.search(collection, default_single_query)
# assert len(res[0]) == 0
@pytest.mark.tags("fail")
def test_release_partition_during_searching(self, connect, collection):
"""
target: test release partition during searching
@ -278,6 +317,7 @@ class TestReleaseAdvanced:
res = connect.search(collection, default_single_query)
assert len(res[0]) == 0
@pytest.mark.tags("fail")
def test_release_collection_during_searching_A(self, connect, collection):
"""
target: test release collection during searching
@ -297,6 +337,7 @@ class TestReleaseAdvanced:
connect.search(collection, default_single_query)
# assert len(res[0]) == 0
@pytest.mark.tags("fail")
def _test_release_collection_during_loading(self, connect, collection):
"""
target: test release collection during loading
@ -315,6 +356,7 @@ class TestReleaseAdvanced:
res = connect.search(collection, default_single_query)
assert len(res[0]) == 0
@pytest.mark.tags("fail")
def test_release_partition_during_loading(self, connect, collection):
"""
target: test release partition during loading
@ -334,6 +376,7 @@ class TestReleaseAdvanced:
res = connect.search(collection, default_single_query)
assert len(res[0]) == 0
@pytest.mark.tags("fail")
def test_release_collection_during_inserting(self, connect, collection):
"""
target: test release collection during inserting
@ -354,6 +397,7 @@ class TestReleaseAdvanced:
res = connect.search(collection, default_single_query)
# assert len(res[0]) == 0
@pytest.mark.tags("fail")
def _test_release_collection_during_indexing(self, connect, collection):
"""
target: test release collection during building index
@ -362,6 +406,7 @@ class TestReleaseAdvanced:
"""
pass
@pytest.mark.tags("fail")
def _test_release_collection_during_droping_index(self, connect, collection):
"""
target: test release collection during droping index
@ -426,6 +471,7 @@ class TestLoadPartition:
else:
pytest.skip("Skip index Temporary")
@pytest.mark.tags("fail")
def test_load_partition_after_index(self, connect, collection, get_simple_index):
'''
target: test load collection, after index created
@ -444,6 +490,7 @@ class TestLoadPartition:
assert len(res[0]) == default_top_k
@pytest.mark.level(2)
@pytest.mark.tags("fail")
def test_load_partition_after_index_binary(self, connect, binary_collection, get_binary_index):
'''
target: test load binary_collection, after index created
@ -505,16 +552,27 @@ class TestLoadPartition:
@pytest.mark.tags("0331")
def test_load_partition_not_existed(self, connect, collection):
partition_name = gen_unique_str(uid)
with pytest.raises(Exception) as e:
try:
connect.load_partitions(collection, [partition_name])
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "partitionID of partitionName:%s can not be find" % partition_name
@pytest.mark.level(2)
@pytest.mark.tags("0331")
def test_release_partition_not_existed(self, connect, collection):
partition_name = gen_unique_str(uid)
with pytest.raises(Exception) as e:
try:
connect.release_partitions(collection, [partition_name])
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "partitionID of partitionName:%s can not be find" % partition_name
@pytest.mark.tags("fail")
def test_release_partition_not_load(self, connect, collection):
"""
target: test release collection without load
@ -528,18 +586,32 @@ class TestLoadPartition:
connect.release_partitions(collection, [default_tag])
@pytest.mark.level(2)
@pytest.mark.tags("0331")
def test_load_release_after_drop(self, connect, collection):
connect.create_partition(collection, default_tag)
ids = connect.insert(collection, default_entities, partition_tag=default_tag)
assert len(ids) == default_nb
connect.flush([collection])
connect.load_partitions(collection, [default_tag])
connect.release_partitions(collection, [default_tag])
connect.drop_partition(collection, default_tag)
with pytest.raises(Exception) as e:
try:
connect.load_partitions(collection, [default_tag])
with pytest.raises(Exception) as e:
connect.release_partitions(collection, [default_tag])
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "partitionID of partitionName:%s can not be find" % default_tag
try:
connect.release_partitions(collection, [default_tag])
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "partitionID of partitionName:%s can not be find" % default_tag
@pytest.mark.tags("0331")
def test_release_partition_after_drop(self, connect, collection):
"""
target: test release collection after drop
@ -548,12 +620,19 @@ class TestLoadPartition:
"""
connect.create_partition(collection, default_tag)
ids = connect.insert(collection, default_entities, partition_tag=default_tag)
assert len(ids) == default_nb
connect.flush([collection])
connect.load_partitions(collection, [default_tag])
connect.drop_partition(collection, default_tag)
with pytest.raises(Exception) as e:
connect.release_partitions(collection, [default_tag])
try:
connect.load_partitions(collection, [default_tag])
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "partitionID of partitionName:%s can not be find" % default_tag
@pytest.mark.tags("0331")
def test_load_release_after_collection_drop(self, connect, collection):
"""
target: test release collection after drop
@ -562,14 +641,26 @@ class TestLoadPartition:
"""
connect.create_partition(collection, default_tag)
ids = connect.insert(collection, default_entities, partition_tag=default_tag)
assert len(ids) == default_nb
connect.flush([collection])
connect.load_partitions(collection, [default_tag])
connect.release_partitions(collection, [default_tag])
connect.drop_collection(collection)
with pytest.raises(Exception) as e:
try:
connect.load_partitions(collection, [default_tag])
with pytest.raises(Exception) as e:
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "describe collection failed: can't find collection: %s" % collection
try:
connect.release_partitions(collection, [default_tag])
except Exception as e:
code = getattr(e, 'code', "The exception does not contain the field of code.")
assert code == 1
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "describe collection failed: can't find collection: %s" % collection
class TestLoadPartitionInvalid(object):

1
tests/python_test/test_flush.py
View File

@ -173,6 +173,7 @@ class TestFlushBase:
res = connect.get_collection_stats(collection_new)
assert res["row_count"] == nb_new
# TODO ci failed
@pytest.mark.tags("fail")
def test_add_flush_multiable_times(self, connect, collection):
'''