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

[test]modify the diskann testcase (#20807) 

Signed-off-by: jingkl <jingjing.jia@zilliz.com>

Signed-off-by: jingkl <jingjing.jia@zilliz.com>
pull/20776/head
jingkl 2022-11-24 16:53:12 +08:00 committed by GitHub
parent 44d45452fa
commit 81d0056e83
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 21 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1567,7 +1567,7 @@ class TestIndexDiskann(TestcaseBase):
def call_back(self):
assert True
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_create_index_with_diskann_normal(self):
"""
target: test create index with diskann
@ -1593,7 +1593,7 @@ class TestIndexDiskann(TestcaseBase):
check_items={"nq": default_nq,
"limit": default_limit})
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_create_index_with_over_max_dim(self):
"""
target: test create index with diskann
@ -1612,7 +1612,7 @@ class TestIndexDiskann(TestcaseBase):
check_items={ct.err_code: 1,
ct.err_msg: "invalid index params"})
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_create_index_with_diskann_callback(self,_async):
"""
target: test create index with diskann
@ -1641,7 +1641,8 @@ class TestIndexDiskann(TestcaseBase):
check_task=CheckTasks.check_search_results,
check_items={"nq": default_nq,
"limit": default_limit})
@pytest.mark.tags(CaseLabel.L2)
def test_create_diskann_index_drop_with_async(self, _async):
"""
target: test create index interface
@ -1662,7 +1663,7 @@ class TestIndexDiskann(TestcaseBase):
collection_w.drop_index(index_name=ct.default_index_name)
assert len(collection_w.indexes) == 0
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_create_diskann_index_with_partition(self):
"""
target: test create index with diskann
@ -1689,7 +1690,7 @@ class TestIndexDiskann(TestcaseBase):
assert collection_w.has_index(index_name=field_name)[0] == False
assert len(collection_w.indexes) == 0
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_drop_diskann_index_with_noraml(self):
"""
target: test drop diskann index normal
@ -1709,7 +1710,7 @@ class TestIndexDiskann(TestcaseBase):
collection_w.drop_index(index_name=index_name1)
assert collection_w.has_index(index_name=index_name1)[0] == False
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_drop_diskann_index_and_create_again(self):
"""
target: test drop diskann index normal
@ -1730,7 +1731,7 @@ class TestIndexDiskann(TestcaseBase):
collection_w.create_index(default_field_name, ct.default_diskann_index)
assert len(collection_w.indexes) == 1
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_create_more_than_three_index(self):
"""
target: test create diskann index
@ -1751,7 +1752,7 @@ class TestIndexDiskann(TestcaseBase):
collection_w.create_index("float", default_params, index_name="c")
assert collection_w.has_index(index_name="c")[0] == True
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_drop_diskann_index_with_partition(self):
"""
target: test drop diskann index normal
@ -1773,7 +1774,7 @@ class TestIndexDiskann(TestcaseBase):
collection_w.drop_index()
assert len(collection_w.indexes) == 0
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_create_diskann_index_with_binary(self):
"""
target: test create diskann index with binary
@ -1790,7 +1791,7 @@ class TestIndexDiskann(TestcaseBase):
check_items={ct.err_code: 1,
ct.err_msg: "field data type BinaryVector don't support the index build type DISKANN"})
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_create_diskann_index_multithread(self):
"""
target: test create index interface with multiprocess

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

@ -1672,7 +1672,7 @@ class TestqueryString(TestcaseBase):
assert len(res) == nb
@pytest.mark.tags(CaseLabel.L0)
@pytest.mark.tags(CaseLabel.L2)
def test_query_with_create_diskann_index(self):
"""
target: test query after create diskann index
@ -1691,7 +1691,7 @@ class TestqueryString(TestcaseBase):
check_vec = vectors[0].iloc[:, [0]][0:len(int_values)].to_dict('records')
collection_w.query(term_expr, check_task=CheckTasks.check_query_results, check_items={exp_res: check_vec})
@pytest.mark.tags(CaseLabel.L0)
@pytest.mark.tags(CaseLabel.L2)
def test_query_with_create_diskann_with_string_pk(self):
"""
target: test query after create diskann index

14
tests/python_client/testcases/test_search.py
View File

@ -4222,7 +4222,7 @@ class TestsearchDiskann(TestcaseBase):
def _async(self, request):
yield request.param
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_search_with_diskann_index(self, dim, auto_id, _async):
"""
target: test delete after creating index
@ -4261,7 +4261,7 @@ class TestsearchDiskann(TestcaseBase):
"_async": _async}
)
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.parametrize("limit", [20])
@pytest.mark.parametrize("search_list", [10, 201])
def test_search_invalid_params_with_diskann_A(self, dim, auto_id, search_list, limit):
@ -4292,7 +4292,7 @@ class TestsearchDiskann(TestcaseBase):
"err_msg": "fail to search on all shard leaders"}
)
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.parametrize("limit", [6553])
@pytest.mark.parametrize("search_list", [6553, 65531])
def test_search_invalid_params_with_diskann_B(self, dim, auto_id, search_list, limit):
@ -4323,7 +4323,7 @@ class TestsearchDiskann(TestcaseBase):
"err_msg": "fail to search on all shard leaders"}
)
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.parametrize("limit", [6554])
@pytest.mark.parametrize("search_list", [6554, 65536])
def test_search_invalid_params_with_diskann_C(self, dim, auto_id, search_list, limit):
@ -4353,7 +4353,7 @@ class TestsearchDiskann(TestcaseBase):
check_items={"err_code": 1,
"err_msg": "fail to search on all shard leaders"}
)
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_search_with_diskann_with_string_pk(self, dim):
"""
target: test delete after creating index
@ -4385,7 +4385,7 @@ class TestsearchDiskann(TestcaseBase):
)
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_search_with_delete_data(self, dim, auto_id, _async):
"""
target: test delete after creating index
@ -4427,7 +4427,7 @@ class TestsearchDiskann(TestcaseBase):
)
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.tags(CaseLabel.L2)
def test_search_with_diskann_and_more_index(self, dim, auto_id, _async):
"""
target: test delete after creating index