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

Update search testcases (#7363) 

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
pull/7323/head
zhuwenxing 2021-08-30 20:12:00 +08:00 committed by GitHub
parent bde8394198
commit 848db33848
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/python_client/testcases/collection/test_load_collection.py
View File

@ -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):

2
tests/python_client/testcases/entity/test_search.py
View File

@ -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)

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

@ -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}}