def test_default_attributes(): """ Test the default attributes when no environment variables are set. """ env = LightningEnvironment() assert not env.creates_children() assert env.master_address() == "127.0.0.1" assert isinstance(env.master_port(), int) assert env.world_size() == 1 assert env.local_rank() == 0 assert env.node_rank() == 0
def test_manual_user_launch(environ, creates_children): """ Test that the environment switches to manual user mode when LOCAL_RANK env variable detected. """ with mock.patch.dict(os.environ, environ): env = LightningEnvironment() assert env.creates_children() == creates_children