test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-04 04:49:08 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Disable pytest timeout module (#14851)

Browse Source 
Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
This commit is contained in:
binbin 2022-01-06 09:35:30 +08:00 committed by GitHub
parent a18744346a
commit f85eb9cafd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 62 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
tests/python_client/testcases/entity/test_insert.py
View File

@ -9,7 +9,7 @@ from common.constants import default_entity, default_entities, default_binary_en
from common.common_type import CaseLabel from common.common_type import CaseLabel
from utils.util_log import test_log as log from utils.util_log import test_log as log
ADD_TIMEOUT = 60 # ADD_TIMEOUT = 60
uid = "test_insert" uid = "test_insert"
field_name = ut.default_float_vec_field_name field_name = ut.default_float_vec_field_name
binary_field_name = ut.default_binary_vec_field_name binary_field_name = ut.default_binary_vec_field_name
@ -77,7 +77,7 @@ class TestInsertBase:
with pytest.raises(Exception) as e: with pytest.raises(Exception) as e:
connect.insert(collection, entity) connect.insert(collection, entity)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def test_insert_collection_not_existed(self, connect): def test_insert_collection_not_existed(self, connect):
""" """
@ -100,7 +100,7 @@ class TestInsertBase:
dis_connect.insert(collection, default_entities) dis_connect.insert(collection, default_entities)
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
def test_insert_drop_collection(self, connect, collection): def test_insert_drop_collection(self, connect, collection):
""" """
target: test delete collection after insert entities target: test delete collection after insert entities
@ -112,7 +112,7 @@ class TestInsertBase:
connect.drop_collection(collection) connect.drop_collection(collection)
assert connect.has_collection(collection) == False assert connect.has_collection(collection) == False
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_insert_flush_drop_collection(self, connect, collection): def test_insert_flush_drop_collection(self, connect, collection):
""" """
@ -128,7 +128,7 @@ class TestInsertBase:
assert connect.has_collection(collection) == False assert connect.has_collection(collection) == False
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
def test_insert_create_index(self, connect, collection, get_simple_index): def test_insert_create_index(self, connect, collection, get_simple_index):
""" """
target: test build index insert after entities target: test build index insert after entities
@ -144,7 +144,7 @@ class TestInsertBase:
ut.create_target_index(get_simple_index, field_name) ut.create_target_index(get_simple_index, field_name)
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_insert_after_create_index(self, connect, collection, get_simple_index): def test_insert_after_create_index(self, connect, collection, get_simple_index):
""" """
@ -160,7 +160,7 @@ class TestInsertBase:
ut.create_target_index(get_simple_index, field_name) ut.create_target_index(get_simple_index, field_name)
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_insert_search(self, connect, collection): def test_insert_search(self, connect, collection):
""" """
@ -195,7 +195,7 @@ class TestInsertBase:
def insert_count(self, request): def insert_count(self, request):
yield request.param yield request.param
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_insert_ids(self, connect, id_collection, insert_count): def test_insert_ids(self, connect, id_collection, insert_count):
""" """
@ -216,7 +216,7 @@ class TestInsertBase:
assert stats[row_count] == nb assert stats[row_count] == nb
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
def test_insert_the_same_ids(self, connect, id_collection, insert_count): def test_insert_the_same_ids(self, connect, id_collection, insert_count):
""" """
target: test insert vectors in collection, use customize the same ids target: test insert vectors in collection, use customize the same ids
@ -235,7 +235,7 @@ class TestInsertBase:
stats = connect.get_collection_stats(id_collection) stats = connect.get_collection_stats(id_collection)
assert stats[row_count] == nb assert stats[row_count] == nb
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_insert_ids_fields(self, connect, get_filter_field, get_vector_field): def test_insert_ids_fields(self, connect, get_filter_field, get_vector_field):
""" """
@ -262,7 +262,7 @@ class TestInsertBase:
assert stats[row_count] == nb assert stats[row_count] == nb
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
def test_insert_ids_not_match(self, connect, id_collection, insert_count): def test_insert_ids_not_match(self, connect, id_collection, insert_count):
""" """
target: test insert entities in collection without ids target: test insert entities in collection without ids
@ -276,7 +276,7 @@ class TestInsertBase:
connect.insert(id_collection, entities) connect.insert(id_collection, entities)
# TODO # TODO
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def test_insert_twice_ids_no_ids(self, connect, id_collection): def test_insert_twice_ids_no_ids(self, connect, id_collection):
""" """
@ -293,7 +293,7 @@ class TestInsertBase:
connect.insert(id_collection, entities) connect.insert(id_collection, entities)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
def test_insert_not_ids(self, connect, id_collection): def test_insert_not_ids(self, connect, id_collection):
""" """
target: check the result of insert, with params ids and no ids target: check the result of insert, with params ids and no ids
@ -305,7 +305,7 @@ class TestInsertBase:
with pytest.raises(Exception) as e: with pytest.raises(Exception) as e:
connect.insert(id_collection, entities) connect.insert(id_collection, entities)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def test_insert_ids_length_not_match_batch(self, connect, id_collection): def test_insert_ids_length_not_match_batch(self, connect, id_collection):
""" """
@ -321,7 +321,7 @@ class TestInsertBase:
connect.insert(id_collection, entities) connect.insert(id_collection, entities)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
def test_insert_ids_length_not_match_single(self, connect, id_collection): def test_insert_ids_length_not_match_single(self, connect, id_collection):
""" """
target: test insert vectors in collection, use customize ids, len(ids) != len(vectors) target: test insert vectors in collection, use customize ids, len(ids) != len(vectors)
@ -335,7 +335,7 @@ class TestInsertBase:
with pytest.raises(Exception) as e: with pytest.raises(Exception) as e:
connect.insert(id_collection, entity) connect.insert(id_collection, entity)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_insert_partition(self, connect, collection): def test_insert_partition(self, connect, collection):
""" """
@ -351,7 +351,7 @@ class TestInsertBase:
stats = connect.get_collection_stats(collection) stats = connect.get_collection_stats(collection)
assert stats[row_count] == default_nb assert stats[row_count] == default_nb
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_insert_partition_with_ids(self, connect, id_collection): def test_insert_partition_with_ids(self, connect, id_collection):
""" """
@ -367,7 +367,7 @@ class TestInsertBase:
assert result.primary_keys == ids assert result.primary_keys == ids
log.info(connect.describe_collection(id_collection)) log.info(connect.describe_collection(id_collection))
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_insert_default_partition(self, connect, collection): def test_insert_default_partition(self, connect, collection):
""" """
@ -381,7 +381,7 @@ class TestInsertBase:
stats = connect.get_collection_stats(collection) stats = connect.get_collection_stats(collection)
assert stats[row_count] == default_nb assert stats[row_count] == default_nb
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def test_insert_partition_not_existed(self, connect, collection): def test_insert_partition_not_existed(self, connect, collection):
""" """
@ -393,7 +393,7 @@ class TestInsertBase:
with pytest.raises(Exception) as e: with pytest.raises(Exception) as e:
connect.insert(collection, default_entities, partition_name=tag) connect.insert(collection, default_entities, partition_name=tag)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_insert_partition_repeatedly(self, connect, collection): def test_insert_partition_repeatedly(self, connect, collection):
""" """
@ -535,7 +535,7 @@ class TestInsertBase:
connect.insert(collection, tmp_entity) connect.insert(collection, tmp_entity)
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
@pytest.mark.timeout(30) # @pytest.mark.timeout(30)
def test_collection_insert_rows_count_multi_threading(self, args, collection): def test_collection_insert_rows_count_multi_threading(self, args, collection):
""" """
target: test collection rows_count is correct or not with multi threading target: test collection rows_count is correct or not with multi threading
@ -646,7 +646,7 @@ class TestInsertBinary:
ut.create_target_index(get_binary_index, binary_field_name) ut.create_target_index(get_binary_index, binary_field_name)
assert index == get_binary_index assert index == get_binary_index
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def test_insert_binary_create_index(self, connect, binary_collection, get_binary_index): def test_insert_binary_create_index(self, connect, binary_collection, get_binary_index):
""" """
@ -836,7 +836,7 @@ class TestInsertMultiCollections:
for i in range(collection_num): for i in range(collection_num):
connect.drop_collection(collection_list[i]) connect.drop_collection(collection_list[i])
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_drop_collection_insert_entity_another(self, connect, collection): def test_drop_collection_insert_entity_another(self, connect, collection):
""" """
@ -851,7 +851,7 @@ class TestInsertMultiCollections:
connect.flush([collection_name]) connect.flush([collection_name])
assert len(result.primary_keys) == 1 assert len(result.primary_keys) == 1
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_create_index_insert_entity_another(self, connect, collection, get_simple_index): def test_create_index_insert_entity_another(self, connect, collection, get_simple_index):
""" """
@ -870,7 +870,7 @@ class TestInsertMultiCollections:
assert index == get_simple_index assert index == get_simple_index
connect.drop_collection(collection_name) connect.drop_collection(collection_name)
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_insert_entity_create_index_another(self, connect, collection, get_simple_index): def test_insert_entity_create_index_another(self, connect, collection, get_simple_index):
""" """
@ -890,7 +890,7 @@ class TestInsertMultiCollections:
stats = connect.get_collection_stats(collection) stats = connect.get_collection_stats(collection)
assert stats[row_count] == 1 assert stats[row_count] == 1
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_insert_entity_sleep_create_index_another(self, connect, collection, get_simple_index): def test_insert_entity_sleep_create_index_another(self, connect, collection, get_simple_index):
""" """
@ -906,7 +906,7 @@ class TestInsertMultiCollections:
stats = connect.get_collection_stats(collection) stats = connect.get_collection_stats(collection)
assert stats[row_count] == 1 assert stats[row_count] == 1
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_search_entity_insert_entity_another(self, connect, collection): def test_search_entity_insert_entity_another(self, connect, collection):
""" """
@ -924,7 +924,7 @@ class TestInsertMultiCollections:
stats = connect.get_collection_stats(collection_name) stats = connect.get_collection_stats(collection_name)
assert stats[row_count] == 1 assert stats[row_count] == 1
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
def test_insert_entity_search_entity_another(self, connect, collection): def test_insert_entity_search_entity_another(self, connect, collection):
""" """
@ -941,7 +941,7 @@ class TestInsertMultiCollections:
stats = connect.get_collection_stats(collection) stats = connect.get_collection_stats(collection)
assert stats[row_count] == 1 assert stats[row_count] == 1
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def test_insert_entity_sleep_search_entity_another(self, connect, collection): def test_insert_entity_sleep_search_entity_another(self, connect, collection):
""" """
@ -957,7 +957,7 @@ class TestInsertMultiCollections:
res = connect.search(collection_name, **default_single_query) res = connect.search(collection_name, **default_single_query)
assert len(res[0]) == 0 assert len(res[0]) == 0
@pytest.mark.timeout(ADD_TIMEOUT) # @pytest.mark.timeout(ADD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def _test_insert_entity_during_release_collection(self, connect, collection): def _test_insert_entity_during_release_collection(self, connect, collection):
""" """

8
tests/python_client/testcases/test_connection.py
View File

@ -8,7 +8,7 @@ import common.common_type as ct
import common.common_func as cf import common.common_func as cf
from common.code_mapping import ConnectionErrorMessage as cem from common.code_mapping import ConnectionErrorMessage as cem
CONNECT_TIMEOUT = 12 # CONNECT_TIMEOUT = 12
class TestConnectionParams(TestcaseBase): class TestConnectionParams(TestcaseBase):
@ -844,7 +844,7 @@ class TestConnectIPInvalid(object):
yield request.param yield request.param
@pytest.mark.tags(ct.CaseLabel.L2) @pytest.mark.tags(ct.CaseLabel.L2)
@pytest.mark.timeout(CONNECT_TIMEOUT) # @pytest.mark.timeout(CONNECT_TIMEOUT)
def test_connect_with_invalid_ip(self, args, get_invalid_ip): def test_connect_with_invalid_ip(self, args, get_invalid_ip):
ip = get_invalid_ip ip = get_invalid_ip
with pytest.raises(Exception) as e: with pytest.raises(Exception) as e:
@ -864,7 +864,7 @@ class TestConnectPortInvalid(object):
yield request.param yield request.param
@pytest.mark.tags(ct.CaseLabel.L2) @pytest.mark.tags(ct.CaseLabel.L2)
@pytest.mark.timeout(CONNECT_TIMEOUT) # @pytest.mark.timeout(CONNECT_TIMEOUT)
def test_connect_with_invalid_port(self, args, get_invalid_port): def test_connect_with_invalid_port(self, args, get_invalid_port):
""" """
target: test ip:port connect with invalid port value target: test ip:port connect with invalid port value
@ -889,7 +889,7 @@ class TestConnectURIInvalid(object):
yield request.param yield request.param
@pytest.mark.tags(ct.CaseLabel.L2) @pytest.mark.tags(ct.CaseLabel.L2)
@pytest.mark.timeout(CONNECT_TIMEOUT) # @pytest.mark.timeout(CONNECT_TIMEOUT)
def test_connect_with_invalid_uri(self, get_invalid_uri, args): def test_connect_with_invalid_uri(self, get_invalid_uri, args):
""" """
target: test uri connect with invalid uri value target: test uri connect with invalid uri value

52
tests/python_client/testcases/test_index.py
View File

@ -19,7 +19,7 @@ default_index_params = {"index_type": "IVF_SQ8", "metric_type": "L2", "params":
# copied from pymilvus # copied from pymilvus
uid = "test_index" uid = "test_index"
BUILD_TIMEOUT = 300 # BUILD_TIMEOUT = 300
field_name = default_float_vec_field_name field_name = default_float_vec_field_name
binary_field_name = default_binary_vec_field_name binary_field_name = default_binary_vec_field_name
# query = gen_search_vectors_params(field_name, default_entities, default_top_k, 1) # query = gen_search_vectors_params(field_name, default_entities, default_top_k, 1)
@ -445,7 +445,7 @@ class TestIndexBase:
connect.create_index(collection, tmp_field_name, get_simple_index) connect.create_index(collection, tmp_field_name, get_simple_index)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_no_vectors(self, connect, collection, get_simple_index): def test_create_index_no_vectors(self, connect, collection, get_simple_index):
""" """
target: test create index interface target: test create index interface
@ -459,7 +459,7 @@ class TestIndexBase:
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_partition(self, connect, collection, get_simple_index): def test_create_index_partition(self, connect, collection, get_simple_index):
""" """
target: test create index interface target: test create index interface
@ -475,7 +475,7 @@ class TestIndexBase:
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_partition_flush(self, connect, collection, get_simple_index): def test_create_index_partition_flush(self, connect, collection, get_simple_index):
""" """
target: test create index interface target: test create index interface
@ -502,7 +502,7 @@ class TestIndexBase:
dis_connect.create_index(collection, field_name, default_index) dis_connect.create_index(collection, field_name, default_index)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_search_with_query_vectors(self, connect, collection, get_simple_index, get_nq): def test_create_index_search_with_query_vectors(self, connect, collection, get_simple_index, get_nq):
""" """
target: test create index interface, search with more query vectors target: test create index interface, search with more query vectors
@ -522,7 +522,7 @@ class TestIndexBase:
res = connect.search(collection, **params) res = connect.search(collection, **params)
assert len(res) == nq assert len(res) == nq
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def test_create_index_multithread(self, connect, collection, args): def test_create_index_multithread(self, connect, collection, args):
""" """
@ -563,7 +563,7 @@ class TestIndexBase:
connect.create_index(collection_name, field_name, default_index) connect.create_index(collection_name, field_name, default_index)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_insert_flush(self, connect, collection, get_simple_index): def test_create_index_insert_flush(self, connect, collection, get_simple_index):
""" """
target: test create index target: test create index
@ -581,7 +581,7 @@ class TestIndexBase:
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_same_index_repeatedly(self, connect, collection, get_simple_index): def test_create_same_index_repeatedly(self, connect, collection, get_simple_index):
""" """
target: check if index can be created repeatedly, with the same create_index params target: check if index can be created repeatedly, with the same create_index params
@ -596,7 +596,7 @@ class TestIndexBase:
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_different_index_repeatedly(self, connect, collection): def test_create_different_index_repeatedly(self, connect, collection):
""" """
target: check if index can be created repeatedly, with the different create_index params target: check if index can be created repeatedly, with the different create_index params
@ -614,7 +614,7 @@ class TestIndexBase:
assert not index # FLAT is the last index_type, drop all indexes in server assert not index # FLAT is the last index_type, drop all indexes in server
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_different_index_repeatedly_B(self, connect, collection): def test_create_different_index_repeatedly_B(self, connect, collection):
""" """
target: check if index can be created repeatedly, with the different create_index params target: check if index can be created repeatedly, with the different create_index params
@ -634,7 +634,7 @@ class TestIndexBase:
# assert not index # FLAT is the last index_type, drop all indexes in server # assert not index # FLAT is the last index_type, drop all indexes in server
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_ip(self, connect, collection, get_simple_index): def test_create_index_ip(self, connect, collection, get_simple_index):
""" """
target: test create index interface target: test create index interface
@ -650,7 +650,7 @@ class TestIndexBase:
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_no_vectors_ip(self, connect, collection, get_simple_index): def test_create_index_no_vectors_ip(self, connect, collection, get_simple_index):
""" """
target: test create index interface target: test create index interface
@ -665,7 +665,7 @@ class TestIndexBase:
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_partition_ip(self, connect, collection, get_simple_index): def test_create_index_partition_ip(self, connect, collection, get_simple_index):
""" """
target: test create index interface target: test create index interface
@ -682,7 +682,7 @@ class TestIndexBase:
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_partition_flush_ip(self, connect, collection, get_simple_index): def test_create_index_partition_flush_ip(self, connect, collection, get_simple_index):
""" """
target: test create index interface target: test create index interface
@ -700,7 +700,7 @@ class TestIndexBase:
assert index == get_simple_index assert index == get_simple_index
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_search_with_query_vectors_ip(self, connect, collection, get_simple_index, get_nq): def test_create_index_search_with_query_vectors_ip(self, connect, collection, get_simple_index, get_nq):
""" """
target: test create index interface, search with more query vectors target: test create index interface, search with more query vectors
@ -722,7 +722,7 @@ class TestIndexBase:
res = connect.search(collection, **params) res = connect.search(collection, **params)
assert len(res) == nq assert len(res) == nq
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def test_create_index_multithread_ip(self, connect, collection, args): def test_create_index_multithread_ip(self, connect, collection, args):
""" """
@ -765,7 +765,7 @@ class TestIndexBase:
connect.create_index(collection_name, field_name, default_index) connect.create_index(collection_name, field_name, default_index)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_no_vectors_insert_ip(self, connect, collection): def test_create_index_no_vectors_insert_ip(self, connect, collection):
""" """
target: test create index interface when there is no vectors in collection, target: test create index interface when there is no vectors in collection,
@ -786,7 +786,7 @@ class TestIndexBase:
assert index == default_index assert index == default_index
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_same_index_repeatedly_ip(self, connect, collection): def test_create_same_index_repeatedly_ip(self, connect, collection):
""" """
target: check if index can be created repeatedly, with the same create_index params target: check if index can be created repeatedly, with the same create_index params
@ -802,7 +802,7 @@ class TestIndexBase:
assert index == default_index assert index == default_index
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_different_index_repeatedly_ip(self, connect, collection): def test_create_different_index_repeatedly_ip(self, connect, collection):
""" """
target: check if index can be created repeatedly, with the different create_index params target: check if index can be created repeatedly, with the different create_index params
@ -1023,7 +1023,7 @@ class TestIndexBinary:
""" """
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index(self, connect, binary_collection, get_jaccard_index): def test_create_index(self, connect, binary_collection, get_jaccard_index):
""" """
target: test create index interface target: test create index interface
@ -1037,7 +1037,7 @@ class TestIndexBinary:
assert binary_index == get_jaccard_index assert binary_index == get_jaccard_index
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_partition(self, connect, binary_collection, get_jaccard_index): def test_create_index_partition(self, connect, binary_collection, get_jaccard_index):
""" """
target: test create index interface target: test create index interface
@ -1052,7 +1052,7 @@ class TestIndexBinary:
assert binary_index == get_jaccard_index assert binary_index == get_jaccard_index
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_search_with_query_vectors(self, connect, binary_collection, get_jaccard_index, get_nq): def test_create_index_search_with_query_vectors(self, connect, binary_collection, get_jaccard_index, get_nq):
""" """
target: test create index interface, search with more query vectors target: test create index interface, search with more query vectors
@ -1071,7 +1071,7 @@ class TestIndexBinary:
res = connect.search(binary_collection, **params) res = connect.search(binary_collection, **params)
assert len(res) == nq assert len(res) == nq
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
def test_create_index_invalid_metric_type_binary(self, connect, binary_collection, get_l2_index): def test_create_index_invalid_metric_type_binary(self, connect, binary_collection, get_l2_index):
""" """
@ -1256,7 +1256,7 @@ class TestIndexAsync:
****************************************************************** ******************************************************************
""" """
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index(self, connect, collection, get_simple_index): def test_create_index(self, connect, collection, get_simple_index):
""" """
target: test create index interface target: test create index interface
@ -1272,7 +1272,7 @@ class TestIndexAsync:
log.info(res) log.info(res)
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_drop(self, connect, collection): def test_create_index_drop(self, connect, collection):
""" """
target: test create index interface target: test create index interface
@ -1293,7 +1293,7 @@ class TestIndexAsync:
res = future.result() res = future.result()
@pytest.mark.tags(CaseLabel.L0) @pytest.mark.tags(CaseLabel.L0)
@pytest.mark.timeout(BUILD_TIMEOUT) # @pytest.mark.timeout(BUILD_TIMEOUT)
def test_create_index_callback(self, connect, collection, get_simple_index): def test_create_index_callback(self, connect, collection, get_simple_index):
""" """
target: test create index interface target: test create index interface

4
tests/python_client/testcases/test_partition.py
View File

@ -7,7 +7,7 @@ from common.constants import default_entities, default_fields
from common.common_type import CaseLabel from common.common_type import CaseLabel
from utils.util_log import test_log as log from utils.util_log import test_log as log
TIMEOUT = 120 # TIMEOUT = 120
default_nb = ut.default_nb default_nb = ut.default_nb
default_tag = ut.default_tag default_tag = ut.default_tag
@ -29,7 +29,7 @@ class TestCreateBase:
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L2)
@pytest.mark.skip(reason="skip temporarily for debug") @pytest.mark.skip(reason="skip temporarily for debug")
@pytest.mark.timeout(600) # @pytest.mark.timeout(600)
def test_create_partition_limit(self, connect, collection, args): def test_create_partition_limit(self, connect, collection, args):
""" """
target: test create partitions, check status returned target: test create partitions, check status returned