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

Fix the check of search pagination (#22713) 

Signed-off-by: nico <cheng.yuan@zilliz.com>
pull/22662/head
NicoYuan1986 2023-03-14 10:09:57 +08:00 committed by GitHub
parent c27561173d
commit ebc173cfb8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/python_client/testcases/test_search.py
View File

@ -3898,7 +3898,7 @@ class TestsearchPagination(TestcaseBase):
res.done()
res = res.result()
res_distance = res[0].distances[offset:]
assert sorted(search_res[0].distances) == sorted(res_distance)
assert sorted(search_res[0].distances, key=numpy.float32) == sorted(res_distance, key=numpy.float32)
assert set(search_res[0].ids) == set(res[0].ids[offset:])
@pytest.mark.tags(CaseLabel.L1)
@ -3937,7 +3937,7 @@ class TestsearchPagination(TestcaseBase):
res.done()
res = res.result()
res_distance = res[0].distances[offset:]
assert sorted(search_res[0].distances) == sorted(res_distance)
assert sorted(search_res[0].distances, key=numpy.float32) == sorted(res_distance, key=numpy.float32)
assert set(search_res[0].ids) == set(res[0].ids[offset:])
@pytest.mark.tags(CaseLabel.L1)
@ -4004,7 +4004,7 @@ class TestsearchPagination(TestcaseBase):
res.done()
res = res.result()
res_distance = res[0].distances[offset:]
assert sorted(search_res[0].distances) == sorted(res_distance)
assert sorted(search_res[0].distances, key=numpy.float32) == sorted(res_distance, key=numpy.float32)
assert set(search_res[0].ids) == set(res[0].ids[offset:])
@pytest.mark.tags(CaseLabel.L2)
@ -4101,7 +4101,7 @@ class TestsearchPagination(TestcaseBase):
res.done()
res = res.result()
res_distance = res[0].distances[offset:]
assert sorted(search_res[0].distances, key=float) == sorted(res_distance, key=float)
assert sorted(search_res[0].distances, key=numpy.float32) == sorted(res_distance, key=numpy.float32)
assert set(search_res[0].ids) == set(res[0].ids[offset:])
@pytest.mark.tags(CaseLabel.L2)
@ -4272,7 +4272,7 @@ class TestsearchPagination(TestcaseBase):
res.done()
res = res.result()
res_distance = res[0].distances[offset:]
# assert sorted(search_res[0].distances, key=float) == sorted(res_distance, key=float)
assert sorted(search_res[0].distances, key=numpy.float32) == sorted(res_distance, key=numpy.float32)
assert set(search_res[0].ids) == set(res[0].ids[offset:])