Skip to content

Commit 3e63db3

Browse files
committed
test: add part of test cases for add field
Signed-off-by: binbin lv <[email protected]>
1 parent 2e35393 commit 3e63db3

File tree

5 files changed

+782
-6
lines changed

5 files changed

+782
-6
lines changed

tests/python_client/base/client_v2_base.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1100,4 +1100,26 @@ def transfer_replica(self, client, source_group, target_group, collection_name,
11001100
source_group=source_group, target_group=target_group,
11011101
collection_name=collection_name, num_replicas=num_replicas,
11021102
**kwargs).run()
1103+
return res, check_result
1104+
1105+
@trace()
1106+
def create_field_schema(self, client, name, data_type, desc='', timeout=None, check_task=None, check_items=None, **kwargs):
1107+
timeout = TIMEOUT if timeout is None else timeout
1108+
kwargs.update({"timeout": timeout})
1109+
1110+
func_name = sys._getframe().f_code.co_name
1111+
res, check = api_request([client.create_field_schema, name, data_type, desc], **kwargs)
1112+
check_result = ResponseChecker(res, func_name, check_task, check_items, check,
1113+
**kwargs).run()
1114+
return res, check_result
1115+
1116+
@trace()
1117+
def add_collection_field(self, client, collection_name, field_name, data_type, desc="", timeout=None, check_task=None, check_items=None, **kwargs):
1118+
timeout = TIMEOUT if timeout is None else timeout
1119+
kwargs.update({"timeout": timeout})
1120+
1121+
func_name = sys._getframe().f_code.co_name
1122+
res, check = api_request([client.add_collection_field, collection_name, field_name, data_type, desc], **kwargs)
1123+
check_result = ResponseChecker(res, func_name, check_task, check_items, check,
1124+
**kwargs).run()
11031125
return res, check_result

tests/python_client/milvus_client/test_milvus_client_hybrid_search.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -367,6 +367,16 @@ def test_milvus_client_hybrid_search_default(self):
367367
"ids": insert_ids,
368368
"limit": default_limit,
369369
"pk_name": default_primary_key_field_name})
370+
self.add_collection_field(client, collection_name, field_name="field_new", data_type=DataType.INT64,
371+
nullable=True, max_length=100)
372+
self.hybrid_search(client, collection_name, [sub_search1, sub_search2], ranker, limit=default_limit,
373+
filter="field_new is null",
374+
check_task=CheckTasks.check_search_results,
375+
check_items={"enable_milvus_client_api": True,
376+
"nq": len(vectors_to_search),
377+
"ids": insert_ids,
378+
"limit": default_limit,
379+
"pk_name": default_primary_key_field_name})
370380
self.drop_collection(client, collection_name)
371381

372382
@pytest.mark.tags(CaseLabel.L1)

0 commit comments

Comments
 (0)