Modify test case of load balance (#23234)

Signed-off-by: nico <cheng.yuan@zilliz.com>
pull/23299/head
NicoYuan1986 2023-04-07 19:12:31 +08:00 committed by GitHub
parent 9f127dae47
commit fe0d12c514
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions

View File

@ -1807,6 +1807,7 @@ class TestUtilityAdvanced(TestcaseBase):
# load balance
self.utility_wrap.load_balance(collection_w.name, src_node_id, des_node_ids, sealed_segment_ids)
# get segments distribution after load balance
time.sleep(0.5)
res, _ = self.utility_wrap.get_query_segment_info(c_name)
segment_distribution = cf.get_segment_distribution(res)
sealed_segment_ids_after_load_banalce = segment_distribution[src_node_id]["sealed"]