SlurmSpawner: req_gres: add test

jupyterhub · Jul 1, 2020 · 50da68f · 50da68f
1 parent 8d85409
commit 50da68f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/batchspawner/tests/test_spawners.py b/batchspawner/tests/test_spawners.py
@@ -375,13 +375,15 @@ def test_slurm(db, io_loop):
         'req_prologue': 'PROLOGUE',
         'req_epilogue': 'EPILOGUE',
         'req_reservation': 'RES123',
+        'req_gres': 'GRES123',
         }
     batch_script_re_list = [
         re.compile(r'PROLOGUE.*srun batchspawner-singleuser singleuser_command.*EPILOGUE', re.S),
         re.compile(r'^#SBATCH \s+ --cpus-per-task=5', re.X|re.M),
         re.compile(r'^#SBATCH \s+ --time=3-05:10:10', re.X|re.M),
         re.compile(r'^#SBATCH \s+ some_option_asdf', re.X|re.M),
         re.compile(r'^#SBATCH \s+ --reservation=RES123', re.X|re.M),
+        re.compile(r'^#SBATCH \s+ --gres=GRES123', re.X|re.M),
         ]
     from .. import SlurmSpawner
     run_spawner_script(db, io_loop, SlurmSpawner, normal_slurm_script,