From e0474f8f0f7f72327eeab5a838d8db60161f52ee Mon Sep 17 00:00:00 2001 From: Andres Algaba Date: Tue, 30 Nov 2021 11:53:55 +0100 Subject: [PATCH] Add test for `job_id` (#10774) --- tests/plugins/environments/test_slurm_environment.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/plugins/environments/test_slurm_environment.py b/tests/plugins/environments/test_slurm_environment.py index cce06e9c39..71e87adb26 100644 --- a/tests/plugins/environments/test_slurm_environment.py +++ b/tests/plugins/environments/test_slurm_environment.py @@ -27,6 +27,7 @@ def test_default_attributes(): assert env.creates_processes_externally assert env.main_address == "127.0.0.1" assert env.main_port == 12910 + assert env.job_id() is None with pytest.raises(KeyError): # world size is required to be passed as env variable env.world_size() @@ -55,6 +56,7 @@ def test_attributes_from_environment_variables(caplog): assert env.auto_requeue is True assert env.main_address == "1.1.1.1" assert env.main_port == 15000 + 1234 + assert env.job_id() == int("0001234") assert env.world_size() == 20 assert env.global_rank() == 1 assert env.local_rank() == 2