| 17 | namespace { |
| 18 | class FederatedCommTest : public SocketTest {}; |
| 19 | auto MakeConfig(std::string host, std::int32_t port, std::int32_t world, std::int32_t rank) { |
| 20 | Json config{Object{}}; |
| 21 | config["federated_server_address"] = host + ":" + std::to_string(port); |
| 22 | config["federated_world_size"] = Integer{world}; |
| 23 | config["federated_rank"] = Integer{rank}; |
| 24 | return config; |
| 25 | } |
| 26 | } // namespace |
| 27 | |
| 28 | TEST_F(FederatedCommTest, ThrowOnWorldSizeTooSmall) { |