未验证 提交 848db338 编写于 作者: Z zhuwenxing 提交者: GitHub

Update search testcases (#7363)

Signed-off-by: Nzhuwenxing <wenxing.zhu@zilliz.com>
上级 bde83941
......@@ -311,8 +311,8 @@ class TestReleaseAdvanced:
connect.load_partitions(collection, [default_tag])
res = connect.search(collection, query, _async=True)
connect.release_partitions(collection, [default_tag])
res = connect.search(collection, default_single_query)
assert len(res[0]) == 0
with pytest.raises(Exception) as e:
res = connect.search(collection, default_single_query)
@pytest.mark.tags(CaseLabel.tags_smoke)
def test_release_collection_during_searching_A(self, connect, collection):
......
......@@ -1223,7 +1223,7 @@ class TestSearchDSL(object):
collection_term = gen_unique_str("term")
connect.create_collection(collection_term, term_fields)
term_entities = add_field(entities, field_name="term")
ids = connect.insert(collection_term, term_entities)
ids = connect.insert(collection_term, term_entities).primary_keys
assert len(ids) == default_nb
connect.flush([collection_term])
# count = connect.count_entities(collection_term)
......
......@@ -1512,7 +1512,6 @@ class TestCollectionSearch(TestcaseBase):
assert abs(res[0]._distances[0] - min(distance_0, distance_1)) <= epsilon
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.xfail(expression="500 <= int64 < 1000", reason="issue:7142")
@pytest.mark.parametrize("expression", cf.gen_normal_expressions())
def test_search_with_expression(self, dim, expression, _async):
"""
......@@ -1526,10 +1525,10 @@ class TestCollectionSearch(TestcaseBase):
nb, dim=dim,
is_index=True)
# filter result with expression in colllection
# filter result with expression in collection
_vectors = _vectors[0]
expression = expression.replace("&&", "and").replace("||", "or")
filter_ids = []
expression = expression.replace("&&", "and").replace("||", "or")
filter_ids = []
for i, _id in enumerate(insert_ids):
int64 = _vectors.int64[i]
float = _vectors.float[i]
......@@ -1559,10 +1558,9 @@ class TestCollectionSearch(TestcaseBase):
filter_ids_set = set(filter_ids)
for hits in search_res:
ids = hits.ids
assert set(ids).issubset(filter_ids_set)
assert set(ids).issubset(filter_ids_set)
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.xfail(expression=f"500 <= {default_float_field_name} <= 1000", reason="issue:7142")
@pytest.mark.parametrize("expression", cf.gen_normal_expressions_field(default_float_field_name))
def test_search_with_expression_auto_id(self, dim, expression, _async):
"""
......@@ -1578,14 +1576,14 @@ class TestCollectionSearch(TestcaseBase):
is_index=True)
# filter result with expression in colllection
# filter result with expression in collection
_vectors = _vectors[0]
expression = expression.replace("&&", "and").replace("||", "or")
filter_ids = []
for i, _id in enumerate(insert_ids):
exec(f"{default_float_field_name} = _vectors.{default_float_field_name}[i]")
if not expression or eval(expression):
filter_ids.append(_id)
filter_ids.append(_id)
# 2. create index
index_param = {"index_type": "IVF_FLAT", "metric_type": "L2", "params": {"nlist": 100}}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册