From 81eebc5d2c60455427a02d6009d74363bcb43736 Mon Sep 17 00:00:00 2001
From: Pariksheet Nanda <pariksheet.nanda@uconn.edu>
Date: Thu, 16 May 2019 16:42:27 -0400
Subject: [PATCH] TST: Ensure prng only produces a zero once

---
 tests/prng.bats | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/tests/prng.bats b/tests/prng.bats
index 5e3b413..7fc3ced 100644
--- 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
 }