def test_attributes_from_environment_variables():
""" Test that the SLURM cluster environment takes the attributes from the environment variables. """
env = SLURMEnvironment()
assert env.master_address() == "1.1.1.1"
assert env.master_port() == 15000 + 1234
assert env.world_size() is None
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 = SLURMEnvironment()
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()
with pytest.raises(KeyError):
# node_rank is required to be passed as env variable
env.node_rank()
def test_attributes_from_environment_variables(caplog):
"""Test that the SLURM cluster environment takes the attributes from the environment variables."""
env = SLURMEnvironment()
assert env.master_address() == "1.1.1.1"
assert env.master_port() == 15000 + 1234
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
