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

test: update test cases (#32396) 

Signed-off-by: nico <cheng.yuan@zilliz.com>
pull/32422/head
nico 2024-04-18 16:15:25 +08:00 committed by GitHub
parent 66fae53cd2
commit e83651def4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/python_client/common/common_func.py
View File

@ -430,7 +430,7 @@ def gen_binary_vectors(num, dim):
def gen_default_dataframe_data(nb=ct.default_nb, dim=ct.default_dim, start=0, with_json=True,
random_primary_key=False, multiple_dim_array=[], multiple_vector_field_name=[],
vector_data_type="FLOAT_VECTOR", auto_id = False, primary_field = ct.default_int64_field_name):
vector_data_type="FLOAT_VECTOR", auto_id=False, primary_field = ct.default_int64_field_name):
if not random_primary_key:
int_values = pd.Series(data=[i for i in range(start, start + nb)])
else:
@ -465,6 +465,7 @@ def gen_default_dataframe_data(nb=ct.default_nb, dim=ct.default_dim, start=0, wi
return df
def gen_general_default_list_data(nb=ct.default_nb, dim=ct.default_dim, start=0, with_json=True,
random_primary_key=False, multiple_dim_array=[], multiple_vector_field_name=[],
vector_data_type="FLOAT_VECTOR", auto_id = False,
@ -650,7 +651,7 @@ def gen_dataframe_multi_string_fields(string_fields, nb=ct.default_nb):
def gen_dataframe_all_data_type(nb=ct.default_nb, dim=ct.default_dim, start=0, with_json=True,
auto_id=False, random_primary_key=False, multiple_dim_array=[],
multiple_vector_field_name=[], primary_field = ct.default_int64_field_name):
multiple_vector_field_name=[], primary_field=ct.default_int64_field_name):
if not random_primary_key:
int64_values = pd.Series(data=[i for i in range(start, start + nb)])
else:
@ -694,6 +695,7 @@ def gen_dataframe_all_data_type(nb=ct.default_nb, dim=ct.default_dim, start=0, w
return df
def gen_general_list_all_data_type(nb=ct.default_nb, dim=ct.default_dim, start=0, with_json=True,
auto_id=False, random_primary_key=False, multiple_dim_array=[],
multiple_vector_field_name=[], primary_field=ct.default_int64_field_name):
@ -1734,11 +1736,11 @@ def insert_data(collection_w, nb=ct.default_nb, is_binary=False, is_all_data_typ
else:
if not enable_dynamic_field:
if vector_data_type == "FLOAT_VECTOR":
default_data = gen_dataframe_all_data_type(nb // num, dim=dim, start=start, with_json=with_json,
random_primary_key=random_primary_key,
multiple_dim_array=multiple_dim_array,
multiple_vector_field_name=vector_name_list,
auto_id=auto_id, primary_field=primary_field)
default_data = gen_general_list_all_data_type(nb // num, dim=dim, start=start, with_json=with_json,
random_primary_key=random_primary_key,
multiple_dim_array=multiple_dim_array,
multiple_vector_field_name=vector_name_list,
auto_id=auto_id, primary_field=primary_field)
elif vector_data_type == "FLOAT16_VECTOR" or "BFLOAT16_VECTOR":
default_data = gen_general_list_all_data_type(nb // num, dim=dim, start=start, with_json=with_json,
random_primary_key=random_primary_key,
@ -1929,6 +1931,7 @@ def extract_vector_field_name_list(collection_w):
return vector_name_list
def get_activate_func_from_metric_type(metric_type):
activate_function = lambda x: x
if metric_type == "COSINE":
@ -1939,6 +1942,7 @@ def get_activate_func_from_metric_type(metric_type):
activate_function = lambda x: 1.0 - 2*math.atan(x) / math.pi
return activate_function
def get_hybrid_search_base_results_rrf(search_res_dict_array, round_decimal=-1):
"""
merge the element in the dicts array

2
tests/python_client/requirements.txt
View File

@ -12,7 +12,7 @@ allure-pytest==2.7.0
pytest-print==0.2.1
pytest-level==0.1.1
pytest-xdist==2.5.0
pymilvus==2.5.0rc13
pymilvus==2.5.0rc15
pytest-rerunfailures==9.1.1
git+https://github.com/Projectplace/pytest-tags
ndg-httpsclient

1
tests/python_client/testcases/test_collection.py
View File

@ -4134,6 +4134,7 @@ class TestCollectionARRAY(TestcaseBase):
self.init_collection_wrap(schema=array_schema, check_task=CheckTasks.err_res, check_items=error)
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.skip("https://github.com/milvus-io/pymilvus/issues/2041")
def test_collection_array_field_no_capacity(self):
"""
target: Create a field without giving max_capacity

1
tests/python_client/testcases/test_compaction.py
View File

@ -216,6 +216,7 @@ class TestCompactionParams(TestcaseBase):
assert len(res[0]) == ct.default_limit
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.skip("https://github.com/milvus-io/milvus/issues/31258")
def test_compact_delete_ratio(self):
"""
target: test delete entities reaches ratio and auto-compact

1
tests/python_client/testcases/test_insert.py
View File

@ -2143,6 +2143,7 @@ class TestUpsertInvalid(TestcaseBase):
collection_w.upsert(data=data, check_task=CheckTasks.err_res, check_items=error)
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.skip("https://github.com/milvus-io/pymilvus/issues/2042")
@pytest.mark.parametrize("dim", [127, 129, 200])
def test_upsert_dim_unmatch(self, dim):
"""