From f9137d880bbb701696e69d987def1b73e6b12846 Mon Sep 17 00:00:00 2001 From: nico <109071306+NicoYuan1986@users.noreply.github.com> Date: Tue, 10 Oct 2023 11:53:34 +0800 Subject: [PATCH] Modify nightly test cases (#27517) Signed-off-by: zilliz Co-authored-by: zilliz --- tests/python_client/testcases/test_insert.py | 2 +- tests/python_client/testcases/test_search.py | 31 ++++++++++++-------- 2 files changed, 19 insertions(+), 14 deletions(-) diff --git a/tests/python_client/testcases/test_insert.py b/tests/python_client/testcases/test_insert.py index 5035995be6f36..42cbba0543b38 100644 --- a/tests/python_client/testcases/test_insert.py +++ b/tests/python_client/testcases/test_insert.py @@ -2151,7 +2151,7 @@ def test_upsert_partition_name_nonexistent(self): collection_w = self.init_collection_wrap(name=c_name) data = cf.gen_default_dataframe_data(nb=2) partition_name = "partition1" - error = {ct.err_code: 1, ct.err_msg: "partition is not exist: partition1"} + error = {ct.err_code: 15, ct.err_msg: f"partition={partition_name}: partition not found"} collection_w.upsert(data=data, partition_name=partition_name, check_task=CheckTasks.err_res, check_items=error) diff --git a/tests/python_client/testcases/test_search.py b/tests/python_client/testcases/test_search.py index 73a5c4e8f6fe8..690d794a3402e 100644 --- a/tests/python_client/testcases/test_search.py +++ b/tests/python_client/testcases/test_search.py @@ -7250,28 +7250,33 @@ def test_range_search_with_expression_large(self, dim): is_index=False)[0:4] # 2. create index - index_param = {"index_type": "IVF_FLAT", - "metric_type": "L2", "params": {"nlist": 100}} + index_param = {"index_type": "IVF_FLAT", "metric_type": "L2", "params": {"nlist": 100}} collection_w.create_index("float_vector", index_param) collection_w.load() # 3. search with expression expression = f"0 < {default_int64_field_name} < 5001" - log.info( - "test_search_with_expression: searching with expression: %s" % expression) + log.info("test_search_with_expression: searching with expression: %s" % expression) nums = 5000 vectors = [[random.random() for _ in range(dim)] for _ in range(nums)] - range_search_params = {"metric_type": "L2", "params": {"radius": 1000, - "range_filter": 0}} + # calculate the distance to make sure in range(0, 1000) + search_params = {"metric_type": "L2"} search_res, _ = collection_w.search(vectors, default_search_field, - range_search_params, default_limit, expression, - check_task=CheckTasks.check_search_results, - check_items={ - "nq": nums, - "ids": insert_ids, - "limit": default_limit, - }) + search_params, 500, expression) + for i in range(nums): + if len(search_res[i]) < 10: + assert False + for j in range(len(search_res[i])): + if search_res[i][j].distance < 0 or search_res[i][j].distance >= 1000: + assert False + # range search + range_search_params = {"metric_type": "L2", "params": {"radius": 1000, "range_filter": 0}} + search_res, _ = collection_w.search(vectors, default_search_field, + range_search_params, default_limit, expression) + for i in range(nums): + log.info(i) + assert len(search_res[i]) == default_limit @pytest.mark.tags(CaseLabel.L2) def test_range_search_with_consistency_bounded(self, nq, dim, auto_id, _async):