@@ 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 |