SMART-Lab / smartdispatch

        # Check if needed modules for this queue are included in the PBS file

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

    def test_generate_pbs2_mem(self):

        # Should need two PBS files

        command_params = {'mem_per_command': self.mem_per_node // 2}

        job_generator = JobGenerator(self.queue, self.commands, 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_mem(self):

        # Should need four PBS files

        command_params = {'mem_per_command': self.mem_per_node}

        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 need two PBS files

        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 need four PBS files

        # Check if needed modules for this queue are included in the PBS file

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

    def test_generate_pbs2_mem(self):

        # Should need two PBS files

        command_params = {'mem_per_command': self.mem_per_node // 2}

        job_generator = JobGenerator(self.queue, self.commands, 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_mem(self):

        # Should need four PBS files