|
@@ 67-74 (lines=8) @@
|
| 64 |
|
job_generator = JobGenerator(self.queue, self.commands, command_params=command_params) |
| 65 |
|
assert_equal(len(job_generator.pbs_list), 4) |
| 66 |
|
assert_equal([pbs.commands[0] for pbs in job_generator.pbs_list], self.commands) |
| 67 |
|
|
| 68 |
|
# Since queue has no gpus it should not be specified in PBS resource `nodes` |
| 69 |
|
assert_true('gpus' not in job_generator.pbs_list[0].resources['nodes']) |
| 70 |
|
|
| 71 |
|
# Test modules to load |
| 72 |
|
# Check if needed modules for this queue are included in the PBS file |
| 73 |
|
assert_equal(job_generator.pbs_list[0].modules, self.modules) |
| 74 |
|
|
| 75 |
|
def test_generate_pbs2_gpu(self): |
| 76 |
|
# Test nb_gpus_per_command argument |
| 77 |
|
# Should needs two PBS file |
|
@@ 45-51 (lines=7) @@
|
| 42 |
|
|
| 43 |
|
def test_generate_pbs(self): |
| 44 |
|
job_generator = JobGenerator(self.queue, self.commands, prolog=self.prolog, epilog=self.epilog) |
| 45 |
|
|
| 46 |
|
# Test nb_cores_per_command argument |
| 47 |
|
# Should needs one PBS file |
| 48 |
|
assert_equal(len(job_generator.pbs_list), 1) |
| 49 |
|
assert_equal(job_generator.pbs_list[0].commands, self.commands) |
| 50 |
|
assert_equal(job_generator.pbs_list[0].prolog, self.prolog) |
| 51 |
|
assert_equal(job_generator.pbs_list[0].epilog, self.epilog) |
| 52 |
|
|
| 53 |
|
def test_generate_pbs2_cpu(self): |
| 54 |
|
# Should needs two PBS file |