Remove collection after case run finished in test cases

Signed-off-by: del-zhenwu <zhenxiang.li@zilliz.com>
pull/4973/head^2
del-zhenwu 2021-04-01 07:44:40 -05:00 committed by yefu.chen
parent d7f5c6b1ef
commit 84f950bc4f
4 changed files with 13 additions and 0 deletions

View File

@ -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])

View File

@ -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):

View File

@ -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)

View File

@ -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):