test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-02 11:59:00 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

test: add different nq for restful v2 test (#33456)

Browse Source 
Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
This commit is contained in:
zhuwenxing 2024-05-31 13:57:45 +08:00 committed by GitHub
parent 3336b91ce6
commit 6b51bae5be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/restful_client_v2/testcases/test_vector_operations.py
View File

@ -750,8 +750,9 @@ class TestSearchVector(TestBase):
@pytest.mark.parametrize("enable_dynamic_schema", [True])
@pytest.mark.parametrize("nb", [3000])
@pytest.mark.parametrize("dim", [128])
@pytest.mark.parametrize("nq", [1, 2])
def test_search_vector_with_float_vector_datatype(self, nb, dim, insert_round, auto_id,
is_partition_key, enable_dynamic_schema):
is_partition_key, enable_dynamic_schema, nq):
"""
Insert a vector with a simple payload
"""
@ -812,7 +813,7 @@ class TestSearchVector(TestBase):
# search data
payload = {
"collectionName": name,
"data": [gen_vector(datatype="FloatVector", dim=dim)],
"data": [gen_vector(datatype="FloatVector", dim=dim) for _ in range(nq)],
"filter": "word_count > 100",
"groupingField": "user_id",
"outputFields": ["*"],
@ -827,7 +828,7 @@ class TestSearchVector(TestBase):
}
rsp = self.vector_client.vector_search(payload)
assert rsp['code'] == 0
assert len(rsp['data']) == 100
assert len(rsp['data']) == 100 * nq
@pytest.mark.parametrize("insert_round", [1, 10])