Added test for slurm integration

Author: aalitaiga

test_pbs_slurm.py

@@ -0,0 +1,73 @@
+import time
+import unittest
+
+
+from subprocess import Popen, PIPE
+from nose.tools import assert_equal, assert_true
+
+pbs_string = """\
+#!/usr/bin/env /bin/bash
+
+#PBS -N arrayJob
+#PBS -o arrayJob_%A_%a.out
+#PBS -t 1-5
+#PBS -l walltime=01:00:00
+#PBS -l naccelerators=1
+#PBS -l proc={cons}
+#PBS -M {mail}
+#SBATCH --qos {priority}
+
+######################
+# Begin work section #
+######################
+
+echo "My SLURM_ARRAY_JOB_ID:" $SLURM_ARRAY_JOB_ID
+echo "My SLURM_ARRAY_TASK_ID: " $SLURM_ARRAY_TASK_ID
+nvidia-smi
+"""
+
+sbatch_string = """\
+#!/usr/bin/env -i /bin/zsh
+
+#SBATCH --job-name=arrayJob
+#SBATCH --output=arrayJob_%A_%a.out
+#SBATCH --array=0-5
+#SBATCH --time=01:00:00
+#SBATCH --gres=gpu
+#SBATCH --constraint=gpu6gb
+
+######################
+# Begin work section #
+######################
+
+echo "My SLURM_ARRAY_JOB_ID:" $SLURM_ARRAY_JOB_ID
+echo "My SLURM_ARRAY_TASK_ID: " $SLURM_ARRAY_TASK_ID
+nvidia-smi
+"""
+
+class TestSlurm(unittest.TestCase):
+
+    def test_pbs_slurm(self):
+        priorities = ['unkillable', 'high', 'low']
+        # doesn't test 12 and 24 to avoid killing jobs just for testing
+        constraints = ['gpulowmem', 'gpu6gb', 'gpu8gb']
+        mem = ['2G', '4G', '6G']
+        mail = 'adrien.alitaiga@gmail.com'
+        # gpus = ['titanblack']
+
+        for priority in priorities:
+            for cons in constraints:
+                for m in mem:
+                    string = pbs_string.format(**{
+                        'cons': cons,
+                        'mail': mail,
+                        'priority': priority
+                    })
+                    with open("test.pbs", "w") as text_file:
+                        text_file.write(string)
+                    process = Popen("sbatch test.pbs", stdout=PIPE, stderr=PIPE, shell=True)
+                    time.sleep(1)
+
+                    stdout, stderr = process.communicate()
+                    print stdout
+                    assert_true("Submitted batch job" in stdout)