test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-01 03:18:29 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

[test] Update query with bool field case (#19133)

Browse Source 
Signed-off-by: ThreadDao <yufen.zong@zilliz.com>

Signed-off-by: ThreadDao <yufen.zong@zilliz.com>
This commit is contained in:
ThreadDao 2022-09-14 14:48:32 +08:00 committed by GitHub
parent 086eb92ade
commit 25f5ff36b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -285,18 +285,24 @@ class TestQueryParams(TestcaseBase):
assert self.collection_wrap.num_entities == ct.default_nb
self.collection_wrap.load()
# Now don't support output bool field
# res, _ = self.collection_wrap.query(default_term_expr, output_fields=[ct.default_bool_field_name])
# assert set(res[0].keys()) == {ct.default_int64_field_name, ct.default_bool_field_name}
# output bool field
res, _ = self.collection_wrap.query(default_term_expr, output_fields=[ct.default_bool_field_name])
assert set(res[0].keys()) == {ct.default_int64_field_name, ct.default_bool_field_name}
exprs = [f'{ct.default_bool_field_name} in [false]',
f'{ct.default_bool_field_name} in [True]',
f'{ct.default_bool_field_name} == true',
f'{ct.default_bool_field_name} == False']
# exprs.append(f'{ct.default_bool_field_name} in [0]')
# not support filter bool field with expr 'bool in [0/ 1]'
not_support_expr = f'{ct.default_bool_field_name} in [0]'
error = {ct.err_code: 1, ct.err_msg: 'error: value \"0\" in list cannot be casted to Bool'}
self.collection_wrap.query(not_support_expr, output_fields=[ct.default_bool_field_name],
check_task=CheckTasks.err_res, check_items=error)
# filter bool field by bool term expr
for bool_value in [True, False]:
exprs = [f'{ct.default_bool_field_name} in [{bool_value}]', f'{ct.default_bool_field_name} == {bool_value}']
for expr in exprs:
res, _ = self.collection_wrap.query(expr)
res, _ = self.collection_wrap.query(expr, output_fields=[ct.default_bool_field_name])
assert len(res) == ct.default_nb / 2
for _r in res:
assert _r[ct.default_bool_field_name] == bool_value
@pytest.mark.tags(CaseLabel.L1)
def test_query_expr_by_int8_field(self):