mirror of https://github.com/milvus-io/milvus.git
Remove collection after case run finished in test cases
Signed-off-by: del-zhenwu <zhenxiang.li@zilliz.com>pull/4973/head^2
parent
d7f5c6b1ef
commit
84f950bc4f
|
@ -447,6 +447,7 @@ class TestCollectionMultiCollections:
|
|||
for i in range(collection_num):
|
||||
stats = connect.get_collection_stats(collection_list[i])
|
||||
assert stats[row_count] == insert_count
|
||||
connect.drop_collection(collection_list[i])
|
||||
|
||||
@pytest.mark.level(2)
|
||||
def test_collection_count_multi_collections_binary(self, connect, binary_collection, insert_count):
|
||||
|
@ -469,6 +470,7 @@ class TestCollectionMultiCollections:
|
|||
for i in range(collection_num):
|
||||
stats = connect.get_collection_stats(collection_list[i])
|
||||
assert stats[row_count] == insert_count
|
||||
connect.drop_collection(collection_list[i])
|
||||
|
||||
@pytest.mark.level(2)
|
||||
@pytest.mark.tags(CaseLabel.tags_smoke)
|
||||
|
@ -495,3 +497,4 @@ class TestCollectionMultiCollections:
|
|||
for i in range(collection_num):
|
||||
stats = connect.get_collection_stats(collection_list[i])
|
||||
assert stats[row_count] == default_nb
|
||||
connect.drop_collection(collection_list[i])
|
||||
|
|
|
@ -29,10 +29,14 @@ class TestListCollections:
|
|||
expected: True
|
||||
'''
|
||||
collection_num = 50
|
||||
collection_names = []
|
||||
for i in range(collection_num):
|
||||
collection_name = gen_unique_str(uid)
|
||||
collection_names.append(collection_name)
|
||||
connect.create_collection(collection_name, default_fields)
|
||||
assert collection_name in connect.list_collections()
|
||||
for i in range(collection_num):
|
||||
connect.drop_collection(collection_names[i])
|
||||
|
||||
@pytest.mark.level(2)
|
||||
def test_list_collections_without_connection(self, dis_connect):
|
||||
|
|
|
@ -792,6 +792,8 @@ class TestInsertMultiCollections:
|
|||
assert len(ids) == default_nb
|
||||
stats = connect.get_collection_stats(collection_name)
|
||||
assert stats[row_count] == default_nb
|
||||
for i in range(collection_num):
|
||||
connect.drop_collection(collection_list[i])
|
||||
|
||||
@pytest.mark.timeout(ADD_TIMEOUT)
|
||||
@pytest.mark.tags(CaseLabel.tags_smoke)
|
||||
|
|
|
@ -914,9 +914,11 @@ class TestSearchBase:
|
|||
num = 10
|
||||
top_k = 10
|
||||
nq = 20
|
||||
collection_names = []
|
||||
for i in range(num):
|
||||
collection = gen_unique_str(uid + str(i))
|
||||
connect.create_collection(collection, default_fields)
|
||||
collection_names.append(collection)
|
||||
entities, ids = init_data(connect, collection)
|
||||
assert len(ids) == default_nb
|
||||
query, vecs = gen_query_vectors(field_name, entities, top_k, nq, search_params=search_param)
|
||||
|
@ -927,6 +929,8 @@ class TestSearchBase:
|
|||
assert check_id_result(res[i], ids[i])
|
||||
assert res[i]._distances[0] < epsilon
|
||||
assert res[i]._distances[1] > epsilon
|
||||
for i in range(num):
|
||||
connect.drop_collection(collection_names[i])
|
||||
|
||||
@pytest.mark.skip("r0.3-test")
|
||||
def _test_query_entities_with_field_less_than_top_k(self, connect, id_collection):
|
||||
|
|
Loading…
Reference in New Issue