SMART-Lab / smartdispatch

        assert_equal(job_generator.pbs_list[0].modules, self.modules)

    def test_generate_pbs2_gpu(self):

        # Test nb_gpus_per_command argument

        # Should needs two PBS file

        command_params = {'nb_gpus_per_command': self.gpus // 2}

        job_generator = JobGenerator(self.queue_gpu, self.commands, command_params=command_params)

        assert_equal(len(job_generator.pbs_list), 2)

        assert_equal(job_generator.pbs_list[0].commands, self.commands[:2])

        assert_equal(job_generator.pbs_list[1].commands, self.commands[2:])

    def test_generate_pbs4_gpu(self):

        # Should needs four PBS files

        command_params = {'nb_gpus_per_command': self.gpus}

        job_generator = JobGenerator(self.queue_gpu, self.commands, command_params=command_params)

        assert_equal(len(job_generator.pbs_list), 4)

        assert_equal([pbs.commands[0] for pbs in job_generator.pbs_list], self.commands)

        # Since queue has gpus it should be specified in PBS resource `nodes`

        assert_equal(job_generator.pbs_list[0].prolog, self.prolog)

        assert_equal(job_generator.pbs_list[0].epilog, self.epilog)

    def test_generate_pbs2_cpu(self):

        # Should needs two PBS file

        command_params = {'nb_cores_per_command': self.cores // 2}

        job_generator = JobGenerator(self.queue, self.commands, command_params=command_params)

        assert_equal(len(job_generator.pbs_list), 2)

        assert_equal(job_generator.pbs_list[0].commands, self.commands[:2])

        assert_equal(job_generator.pbs_list[1].commands, self.commands[2:])

    def test_generate_pbs4_cpu(self):

        # Should needs four PBS file

        command_params = {'nb_cores_per_command': self.cores}

        job_generator = JobGenerator(self.queue, self.commands, command_params=command_params)

        assert_equal(len(job_generator.pbs_list), 4)

        assert_equal([pbs.commands[0] for pbs in job_generator.pbs_list], self.commands)

        # Since queue has no gpus it should not be specified in PBS resource `nodes`

        self.queue = Queue(self.name, self.cluster_name, self.walltime, self.cores, 0, self.mem_per_node, self.modules)

        self.queue_gpu = Queue(self.name, self.cluster_name, self.walltime, self.cores, self.gpus, self.mem_per_node, self.modules)

        self.commands = ["echo 1", "echo 2", "echo 3", "echo 4"]

        self.prolog = ["echo prolog"]

        self.epilog = ["echo epilog"]

    def tearDown(self):

        shutil.rmtree(self.testing_dir)

    def test_generate_pbs(self):

        job_generator = JobGenerator(self.queue, self.commands, prolog=self.prolog, epilog=self.epilog)

        # Test nb_cores_per_command argument