Skip to content

Commit 3eb3270

Browse files
authored
Merge pull request #168 from pyiron/add_write_test
Add write test
2 parents fa14c0d + 4cb1cdb commit 3eb3270

1 file changed

Lines changed: 40 additions & 0 deletions

File tree

tests/test_queueadapter.py

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -369,6 +369,10 @@ def test_submit_job_remote(self):
369369
with self.assertRaises(NotImplementedError):
370370
self.remote.submit_job(queue="remote", dependency_list=[])
371371

372+
def test_submit_job_empty_working_directory(self):
373+
with self.assertRaises(ValueError):
374+
self.slurm.submit_job(working_directory=" ")
375+
372376
def test_memory_string_comparison(self):
373377
self.assertEqual(BasisQueueAdapter._value_in_range(1023, value_min="1K"), "1K")
374378
self.assertEqual(BasisQueueAdapter._value_in_range(1035, value_min="1K"), 1035)
@@ -412,3 +416,39 @@ def test_memory_string_comparison(self):
412416
),
413417
"60G",
414418
)
419+
420+
def test_write_queue(self):
421+
with self.assertRaises(ValueError):
422+
self.slurm._adapter._write_queue_script(
423+
queue=None,
424+
job_name=None,
425+
working_directory=None,
426+
cores=None,
427+
memory_max=None,
428+
run_time_max=None,
429+
command=None
430+
)
431+
self.slurm._adapter._write_queue_script(
432+
queue="slurm",
433+
job_name=None,
434+
working_directory=None,
435+
cores=None,
436+
memory_max=None,
437+
run_time_max=None,
438+
command="echo \"hello\""
439+
)
440+
with open("run_queue.sh", "r") as f:
441+
content = f.read()
442+
output = """\
443+
#!/bin/bash
444+
#SBATCH --output=time.out
445+
#SBATCH --job-name=None
446+
#SBATCH --chdir=.
447+
#SBATCH --get-user-env=L
448+
#SBATCH --partition=slurm
449+
#SBATCH --time=4320
450+
#SBATCH --cpus-per-task=10
451+
452+
echo \"hello\""""
453+
self.assertEqual(content, output)
454+
os.remove("run_queue.sh")

0 commit comments

Comments
 (0)