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 testcase with query (#17137) 

Signed-off-by: jingkl <jingjing.jia@zilliz.com>
pull/17140/head
jingkl 2022-05-20 22:51:58 +08:00 committed by GitHub
parent 7659968126
commit 5344dbd2b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/python_client/requirements.txt
View File

@ -9,7 +9,7 @@ allure-pytest==2.7.0
pytest-print==0.2.1
pytest-level==0.1.1
pytest-xdist==2.2.1
pymilvus==2.1.0.dev57
pymilvus==2.1.0.dev61
pytest-rerunfailures==9.1.1
git+https://github.com/Projectplace/pytest-tags
ndg-httpsclient

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

@ -259,7 +259,11 @@ class TestQueryParams(TestcaseBase):
res = df.iloc[:2, :-1].to_dict('records')
for field in non_primary_field:
filter_values = df[field].tolist()[:2]
term_expr = f'{field} in {filter_values}'
if field is not ct.default_string_field_name:
term_expr = f'{field} in {filter_values}'
else:
term_expr = f'{field} in {filter_values}'
term_expr = term_expr.replace("'", "\"")
self.collection_wrap.query(term_expr, output_fields=["*"],
check_task=CheckTasks.check_query_results, check_items={exp_res: res})
@ -1260,7 +1264,6 @@ class TestqueryString(TestcaseBase):
collection_w.query(expression, check_task=CheckTasks.err_res,
check_items={ct.err_code: 1, ct.err_msg: "type mismatch"})
@pytest.mark.tags(CaseLabel.L1)
def test_query_string_expr_with_binary(self):
"""
@ -1279,8 +1282,8 @@ class TestqueryString(TestcaseBase):
def test_query_string_expr_with_prefixes(self):
"""
target: test query with
method: specify string primary field as output field
expected: return string primary field
method: specify string is primary field, use prefix string expr
expected: verify query successfully
"""
collection_w, vectors = self.init_collection_general(prefix, insert_data=True, primary_field=ct.default_string_field_name)[0:2]
res = vectors[0].iloc[:1, :3].to_dict('records')
@ -1288,3 +1291,16 @@ class TestqueryString(TestcaseBase):
output_fields = [default_int_field_name, default_float_field_name, default_string_field_name]
collection_w.query(expression, output_fields=output_fields,
check_task=CheckTasks.check_query_results, check_items={exp_res: res})
@pytest.mark.tags(CaseLabel.L1)
def test_query_string_with_invaild_prefix_expr(self):
"""
target: test query with
method: specify string primary field, use invaild prefix string expr
expected: raise error
"""
collection_w = self.init_collection_general(prefix, insert_data=True)[0]
expression = 'float like "0%"'
collection_w.query(expression, check_task=CheckTasks.err_res,
check_items={ct.err_code: 1, ct.err_msg: "like operation on non-text field is unsupported"}
)