admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Tags load release

Browse Source 
Signed-off-by: ThreadDao <yufen.zong@zilliz.com>
This commit is contained in:
ThreadDao 2021-03-16 09:44:00 +08:00 committed by yefu.chen
parent b92ad3edd6
commit dd9345aeb4
1 changed files with 19 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -36,9 +36,7 @@ class TestLoadCollection:
def get_binary_index(self, request, connect):
return request.param
# @pytest.mark.tags("0331")
# TODO ci failed
@pytest.mark.tags("fail")
@pytest.mark.tags("0331")
def test_load_collection_after_index(self, connect, collection, get_simple_index):
'''
target: test load collection, after index created
@ -53,8 +51,6 @@ 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
@ -145,6 +141,15 @@ class TestLoadCollection:
connect.flush([collection])
connect.release_collection(collection)
@pytest.mark.tags("0331")
def test_load_collection_after_load_release(self, connect, collection):
ids = connect.insert(collection, default_entities)
assert len(ids) == default_nb
connect.flush([collection])
connect.load_collection(collection)
connect.release_collection(collection)
connect.load_collection(collection)
@pytest.mark.tags("0331")
def test_load_collection_repeatedly(self, connect, collection):
ids = connect.insert(collection, default_entities)
@ -199,8 +204,7 @@ class TestLoadCollection:
message = getattr(e, 'message', "The exception does not contain the field of message.")
assert message == "describe collection failed: can't find collection: %s" % collection
# TODO
@pytest.mark.tags("fail")
@pytest.mark.tags("0331")
def test_load_collection_without_flush(self, connect, collection):
"""
target: test load collection without flush
@ -209,11 +213,9 @@ class TestLoadCollection:
"""
ids = connect.insert(collection, default_entities)
assert len(ids) == default_nb
# with pytest.raises(Exception) as e:
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
@ -236,8 +238,9 @@ class TestLoadCollection:
connect.flush([collection])
connect.load_collection(collection)
connect.release_partitions(collection, [default_tag])
res = connect.search(collection, default_single_query, partition_tags=[default_tag])
assert len(res[0]) == 0
logging.getLogger().info(connect.get_collection_stats(collection))
with pytest.raises(Exception) as e:
connect.search(collection, default_single_query, partition_tags=[default_tag])
res = connect.search(collection, default_single_query, partition_tags=[default_partition_name])
assert len(res[0]) == default_top_k
@ -256,8 +259,8 @@ class TestLoadCollection:
connect.flush([collection])
connect.load_collection(collection)
connect.release_partitions(collection, [default_partition_name, default_tag])
res = connect.search(collection, default_single_query)
assert len(res[0]) == 0
with pytest.raises(Exception) as e:
connect.search(collection, default_single_query)
@pytest.mark.tags("0331")
def test_load_partitions_release_collection(self, connect, collection):
@ -337,7 +340,6 @@ 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
@ -397,7 +399,6 @@ 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
@ -406,7 +407,6 @@ class TestReleaseAdvanced:
"""
pass
@pytest.mark.tags("fail")
def _test_release_collection_during_droping_index(self, connect, collection):
"""
target: test release collection during droping index
@ -471,7 +471,7 @@ class TestLoadPartition:
else:
pytest.skip("Skip index Temporary")
@pytest.mark.tags("fail")
@pytest.mark.tags("0331")
def test_load_partition_after_index(self, connect, collection, get_simple_index):
'''
target: test load collection, after index created
@ -490,7 +490,7 @@ class TestLoadPartition:
assert len(res[0]) == default_top_k
@pytest.mark.level(2)
@pytest.mark.tags("fail")
@pytest.mark.tags("0331")
def test_load_partition_after_index_binary(self, connect, binary_collection, get_binary_index):
'''
target: test load binary_collection, after index created
@ -572,7 +572,7 @@ class TestLoadPartition:
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")
@pytest.mark.tags("0331")
def test_release_partition_not_load(self, connect, collection):
"""
target: test release collection without load