Update unit.jl

Author: DilumAluthge

test/unit.jl

@@ -1 +1,26 @@
+@testset "get_slurm_ntasks_int()" begin
+  c = withenv(SLURM_NTASKS => "56") do
+    SlurmClusterManager.get_slurm_jobid_int()
+  end
+  @test c == 34
 
+  withenv("SLURM_NTASKS" => nothing) do
+    @test_throws ErrorException SlurmClusterManager.get_slurm_jobid_int()
+  end
+end
+
+@testset "get_slurm_jobid_int()" begin
+  a = withenv("SLURM_JOB_ID" => "12", SLURM_JOBID => nothing) do
+    SlurmClusterManager.get_slurm_jobid_int()
+  end
+  @test a == 12
+
+  b = withenv("SLURM_JOB_ID" => nothing, SLURM_JOBID => "34") do
+    SlurmClusterManager.get_slurm_jobid_int()
+  end
+  @test a == 34
+  
+  withenv("SLURM_JOB_ID" => nothing, SLURM_JOBID => nothing) do
+    @test_throws ErrorException SlurmClusterManager.get_slurm_jobid_int()
+  end
+end