Test --resume-from

flatironinstitute · Nov 21, 2024 · 1cab0f5 · 1cab0f5
1 parent e825713
commit 1cab0f5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/tests/test_slurm/run.sh b/tests/test_slurm/run.sh
@@ -12,6 +12,14 @@ salloc -n 2 disBatch Tasks
 [[ -f A.txt && -f B.txt && -f C.txt ]]
 success=$?
 
+# Add a task and check that we can resume
+rm A.txt B.txt C.txt
+echo "touch D.txt" >> Tasks
+salloc -n 2 disBatch Tasks -r Tasks*_status.txt
+
+[[ -f D.txt && ! -f A.txt && ! -f B.txt && ! -f C.txt ]]
+success=$((success + $?))
+
 # Test empty task file
 salloc -n 2 disBatch /dev/null
 success=$((success + $?))

diff --git a/tests/test_ssh/run.sh b/tests/test_ssh/run.sh
@@ -12,6 +12,14 @@ disBatch -s localhost:2 Tasks
 [[ -f A.txt && -f B.txt && -f C.txt ]]
 success=$?
 
+# Add a task and check that we can resume
+rm A.txt B.txt C.txt
+echo "touch D.txt" >> Tasks
+disBatch -s localhost:2 Tasks -r Tasks*_status.txt
+
+[[ -f D.txt && ! -f A.txt && ! -f B.txt && ! -f C.txt ]]
+success=$((success + $?))
+
 # Test empty task file
 disBatch -s localhost:2 /dev/null
 success=$((success + $?))