TST: Ensure prng only produces a zero once

HPC · May 16, 2019 · 81eebc5 · 81eebc5
1 parent 52ee38c
commit 81eebc5
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/tests/prng.bats b/tests/prng.bats
@@ -6,7 +6,7 @@ true
 # shellcheck source=../parallel_opts.sh
 . "${BATS_TEST_DIRNAME}/../examples/script_that_sometimes_fails.sh"
 
-@test 'prng fails at least once' {
+@test 'prng fails at least 4 times' {
     # shellcheck disable=2034
     SLURM_NTASKS=5
 
@@ -16,10 +16,14 @@ true
 	sum=0
 	for ID in {0..4}
 	do
-	    result[ID]=$(prng)
-	    sum+=${result[ID]}
+	    x=$(prng)
+	    if (( x % SLURM_NTASKS != 0 ))
+	    then
+		sum=$(( sum + 1 ))
+	    fi
+	    result[ID]=$x
 	done
-	echo "seed: $seed, prng: ${result[*]}"
-	[[ $sum -gt 0 ]]
+	echo "seed: $seed, prng: ${result[*]}, sum: $sum"
+	[[ $sum -ge 4 ]]
     done
 }