Update pymilvus version and add cases (#23521)

Signed-off-by: nico <cheng.yuan@zilliz.com>
pull/23540/head
nico 2023-04-19 19:02:35 +08:00 committed by GitHub
parent 7a6b435b67
commit 1cc8e6499e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 35 additions and 3 deletions

View File

@ -12,7 +12,7 @@ allure-pytest==2.7.0
pytest-print==0.2.1
pytest-level==0.1.1
pytest-xdist==2.5.0
pymilvus==2.4.0.dev7
pymilvus==2.4.0.dev11
pytest-rerunfailures==9.1.1
git+https://github.com/Projectplace/pytest-tags
ndg-httpsclient

View File

@ -868,7 +868,7 @@ class TestCollectionSearchInvalid(TestcaseBase):
"err_msg": f"Field {output_fields[-1]} not exist"})
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.parametrize("ignore_growing", ct.get_invalid_strs[:8])
@pytest.mark.parametrize("ignore_growing", ct.get_invalid_strs[2:8])
def test_search_invalid_ignore_growing_param(self, ignore_growing):
"""
target: test search ignoring growing segment
@ -877,7 +877,7 @@ class TestCollectionSearchInvalid(TestcaseBase):
3. search with param ignore_growing invalid
expected: raise exception
"""
if ignore_growing == 1:
if ignore_growing is None or ignore_growing == "":
pytest.skip("number is valid")
# 1. create a collection
collection_w = self.init_collection_general(prefix, True)[0]
@ -3236,6 +3236,38 @@ class TestCollectionSearch(TestcaseBase):
for ids in res[0].ids:
assert ids < 10000
@pytest.mark.tags(CaseLabel.L1)
def test_search_ignore_growing_two(self, nq, dim, _async):
"""
target: test search ignoring growing segment
method: 1. create a collection, insert data, create index and load
2. insert data again
3. search with param ignore_growing=True(outside search_params)
expected: searched successfully
"""
# 1. create a collection
collection_w = self.init_collection_general(prefix, True, dim=dim)[0]
# 2. insert data again
data = cf.gen_default_dataframe_data(dim=dim, start=10000)
collection_w.insert(data)
# 3. search with param ignore_growing=True
search_params = {"metric_type": "L2", "params": {"nprobe": 10}}
vector = [[random.random() for _ in range(dim)] for _ in range(nq)]
res = collection_w.search(vector[:nq], default_search_field, search_params, default_limit,
default_search_exp, _async=_async,
ignore_growing=True,
check_task=CheckTasks.check_search_results,
check_items={"nq": nq,
"limit": default_limit,
"_async": _async})[0]
if _async:
res.done()
res = res.result()
for ids in res[0].ids:
assert ids < 10000
@pytest.mark.tags(CaseLabel.L1)
@pytest.mark.parametrize("name", ["_co11ection", "co11_ection"])
@pytest.mark.parametrize("index_name", ["_1ndeX", "In_0"])