# Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.## Licensed under the Apache License, Version 2.0 (the "License");# you may not use this file except in compliance with the License.# You may obtain a copy of the License at## http://www.apache.org/licenses/LICENSE-2.0## Unless required by applicable law or agreed to in writing, software# distributed under the License is distributed on an "AS IS" BASIS,# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.# See the License for the specific language governing permissions and# limitations under the License.from__future__importprint_functionimportpaddle.fluidasfluidimportunittestimportpaddle.fluid.incubate.fleet.base.role_makerasrole_makerfrompaddle.fluid.incubate.fleet.parameter_server.distribute_transpilerimportfleetfrompaddle.fluid.incubate.fleet.utils.fleet_barrier_utilimportcheck_all_trainers_readyclassTestFleetUtils(unittest.TestCase):deftest_fleet_barrier(self):role=role_maker.UserDefinedRoleMaker(current_id=0,role=role_maker.Role.WORKER,worker_num=1,server_endpoints=['127.0.0.1'])fleet.init(role)check_all_trainers_ready("/ready_path/",0)if__name__=='__main__':unittest.main()