From bacef01940fbd57665e010b6458dc47942fcf8a6 Mon Sep 17 00:00:00 2001 From: zhuwenxing Date: Tue, 9 Nov 2021 19:08:17 +0800 Subject: [PATCH] Add testcase for utility function (#11490) Signed-off-by: zhuwenxing --- tests/python_client/testcases/test_utility.py | 36 ++++++++++++++++++- 1 file changed, 35 insertions(+), 1 deletion(-) diff --git a/tests/python_client/testcases/test_utility.py b/tests/python_client/testcases/test_utility.py index 8bf803aa3c..6758d0c771 100644 --- a/tests/python_client/testcases/test_utility.py +++ b/tests/python_client/testcases/test_utility.py @@ -1345,4 +1345,38 @@ class TestUtilityAdvanced(TestcaseBase): for t in threads: t.join() log.debug(self.utility_wrap.list_collections()[0]) - + + @pytest.mark.tags(CaseLabel.L1) + def test_get_query_segment_info_empty_collection(self): + """ + target: test getting query segment info of empty collection + method: init a collection and get query segment info + expected: length of segment is 0 + """ + c_name = cf.gen_unique_str(prefix) + collection_w = self.init_collection_wrap(name=c_name) + collection_w.load() + res, _ = self.utility_wrap.get_query_segment_info(c_name) + assert len(res) == 0 + + @pytest.mark.tags(CaseLabel.L1) + def test_get_query_segment_info(self): + """ + target: test getting query segment info of collection with data + method: init a collection, insert data and get query segment info + expected: + 1. length of segment is greater than 0 + 2. the sum num_rows of each segment is equal to num of entities + """ + c_name = cf.gen_unique_str(prefix) + collection_w = self.init_collection_wrap(name=c_name) + nb = 3000 + df = cf.gen_default_dataframe_data(nb) + collection_w.insert(df) + collection_w.load() + res, _ = self.utility_wrap.get_query_segment_info(c_name) + assert len(res) > 0 + cnt = 0 + for r in res: + cnt += r.num_rows + assert cnt == nb