def test_attributes_from_environment_variables(): """ Test that the torchelastic cluster environment takes the attributes from the environment variables. """ env = TorchElasticEnvironment() assert env.master_address() == "1.2.3.4" assert env.master_port() == 500 assert env.world_size() == 20 assert env.local_rank() == 2 assert env.node_rank() == 3
def test_default_attributes(): """ Test the default attributes when no environment variables are set. """ env = TorchElasticEnvironment() assert env.creates_children() assert env.master_address() == "127.0.0.1" assert env.master_port() == 12910 assert env.world_size() is None with pytest.raises(KeyError): # local rank is required to be passed as env variable env.local_rank() assert env.node_rank() == 0
def test_attributes_from_environment_variables(caplog): """ Test that the torchelastic cluster environment takes the attributes from the environment variables. """ env = TorchElasticEnvironment() assert env.master_address() == "1.2.3.4" assert env.master_port() == 500 assert env.world_size() == 20 assert env.global_rank() == 1 assert env.local_rank() == 2 assert env.node_rank() == 3 # setter should be no-op with caplog.at_level(logging.DEBUG, logger="pytorch_lightning.plugins.environments"): env.set_global_rank(100) assert env.global_rank() == 1 assert "setting global rank is not allowed" in caplog.text caplog.clear() with caplog.at_level(logging.DEBUG, logger="pytorch_lightning.plugins.environments"): env.set_world_size(100) assert env.world_size() == 20 assert "setting world size is not allowed" in caplog.text