From 9ca48b7f292422bd686605f72f49f03b627480cb Mon Sep 17 00:00:00 2001 From: zhuwenxing Date: Tue, 30 Nov 2021 20:51:34 +0800 Subject: [PATCH] Refine get query segment info testcase (#12471) Signed-off-by: zhuwenxing --- tests/python_client/testcases/test_utility.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/python_client/testcases/test_utility.py b/tests/python_client/testcases/test_utility.py index cfd043e1f..8996b654c 100644 --- a/tests/python_client/testcases/test_utility.py +++ b/tests/python_client/testcases/test_utility.py @@ -1401,9 +1401,13 @@ class TestUtilityAdvanced(TestcaseBase): collection_w.search(vectors, default_field_name, ct.default_search_params, ct.default_limit) res, _ = self.utility_wrap.get_query_segment_info(c_name) assert len(res) > 0 + segment_ids = [] cnt = 0 for r in res: - cnt += r.num_rows + log.info(f"segmentID {r.segmentID}: state: {r.state}; num_rows: {r.num_rows} ") + if r.segmentID not in segment_ids: + segment_ids.append(r.segmentID) + cnt += r.num_rows assert cnt == nb @pytest.mark.tags(CaseLabel.L1) -- GitLab