未验证 提交 769aa227 编写于 作者: D del-zhenwu 提交者: GitHub

[skip ci] add some dsl cases (#3111)

* [skip ci] add some dsl cases
Signed-off-by: Nzw <zw@milvus.io>

* [skip ci] add some dsl cases
Signed-off-by: Nzw <zw@milvus.io>
Co-authored-by: Nzw <zw@milvus.io>
Co-authored-by: NWang XiangYu <xy.wang@zilliz.com>
上级 62841c68
......@@ -890,6 +890,77 @@ class TestSearchDSL(object):
res = connect.search(collection, query)
"""
******************************************************************
# The following cases are used to build valid query expr
******************************************************************
"""
def test_query_term_value_not_in(self, connect, collection):
'''
method: build query with vector and term expr, with no term can be filtered
expected: filter pass
'''
entities, ids = init_data(connect, collection)
expr = gen_default_term_expr(values=[100000])
query = update_query_expr(default_query, expr=expr):
res = connect.search(collection, query)
# TODO:
def test_query_term_value_all_in(self, connect, collection):
'''
method: build query with vector and term expr, with all term can be filtered
expected: filter pass
'''
entities, ids = init_data(connect, collection)
expr = gen_default_term_expr(values=1)
query = update_query_expr(default_query, expr=expr):
res = connect.search(collection, query)
# TODO:
def test_query_term_values_not_in(self, connect, collection):
'''
method: build query with vector and term expr, with no term can be filtered
expected: filter pass
'''
entities, ids = init_data(connect, collection)
expr = gen_default_term_expr(values=[i for i in range(100000, 100010)])
query = update_query_expr(default_query, expr=expr):
res = connect.search(collection, query)
# TODO:
def test_query_term_values_all_in(self, connect, collection):
'''
method: build query with vector and term expr, with all term can be filtered
expected: filter pass
'''
entities, ids = init_data(connect, collection)
expr = gen_default_term_expr()
query = update_query_expr(default_query, expr=expr):
res = connect.search(collection, query)
# TODO:
def test_query_term_values_parts_in(self, connect, collection):
'''
method: build query with vector and term expr, with parts of term can be filtered
expected: filter pass
'''
entities, ids = init_data(connect, collection)
expr = gen_default_term_expr(values=[i for i in range(nb/2, nb+nb/2)])
query = update_query_expr(default_query, expr=expr):
res = connect.search(collection, query)
# TODO:
def test_query_term_values_repeat(self, connect, collection):
'''
method: build query with vector and term expr, with the same values
expected: filter pass
'''
entities, ids = init_data(connect, collection)
expr = gen_default_term_expr(values=[1 for i in range(1, nb)])
query = update_query_expr(default_query, expr=expr):
res = connect.search(collection, query)
# TODO:
"""
******************************************************************
# The following cases are used to test `search` function
......
......@@ -293,13 +293,17 @@ def update_query_expr(src_query, keep_old=True, expr=None):
return tmp_query
def gen_default_term_expr():
expr = {"term": {"int64": {"values": [i for i in range(nb/2)]}}}
def gen_default_term_expr(values=None):
if values is None:
values = [i for i in range(nb/2)]
expr = {"term": {"int64": {"values": values}}}
return expr
def gen_default_range_expr():
expr = {"range": {"int64": {"ranges": {"GT": 1, "LT": nb/2}}}}
def gen_default_range_expr(ranges=None):
if ranges is None:
ranges = {"GT": 1, "LT": nb/2}
expr = {"range": {"int64": {"ranges": ranges}}}
return expr
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册