예제 #1
0
    def test_all_config_entries_are_set(self):
        hook_manager = HookManager(
            workflow_list=self.res_config.ert_workflow_list,
            config_dict=self.config_data)
        list_file = hook_manager.getRunpathListFile()
        conf_dir = self.config_data[ConfigKeys.CONFIG_DIRECTORY]
        self.assertEqual(
            list_file,
            os.path.join(conf_dir, self.config_data[ConfigKeys.RUNPATH_FILE]))

        self.assertEqual(len(hook_manager), 2)
        qc_workflow = hook_manager[0]
        self.assertEqual(
            qc_workflow.getWorkflow().src_file,
            self.config_data[ConfigKeys.QC_WORKFLOW_KEY])
        self.assertEqual(
            qc_workflow.getRunMode(),
            HookRuntime.POST_SIMULATION)

        magic_workflow = hook_manager[1]
        self.assertEqual(
            magic_workflow.getWorkflow().src_file,
            os.path.join(conf_dir, self.config_data[ConfigKeys.LOAD_WORKFLOW]))
        self.assertEqual(
            magic_workflow.getRunMode(),
            HookRuntime.PRE_SIMULATION)
예제 #2
0
    def test_different_hook_workflow_gives_not_equal_hook_managers(self):
        res_config2 = ResConfig(user_config_file=self.filename)
        hook_manager1 = HookManager(
            workflow_list=self.res_config.ert_workflow_list,
            config_dict=self.config_data)
        hook_manager2 = HookManager(
            workflow_list=res_config2.ert_workflow_list,
            config_dict=self.remove_key(ConfigKeys.HOOK_WORKFLOW_KEY))

        self.assertNotEqual(hook_manager1, hook_manager2)
예제 #3
0
    def test_different_runpath_gives_not_equal_hook_managers(self):
        res_config2 = ResConfig(user_config_file=self.filename)
        hook_manager1 = HookManager(
            workflow_list=self.res_config.ert_workflow_list,
            config_dict=self.config_data)
        hook_manager2 = HookManager(
            workflow_list=res_config2.ert_workflow_list,
            config_dict=self.set_key(ConfigKeys.RUNPATH_FILE, "runpath2"))

        self.assertNotEqual(hook_manager1, hook_manager2)
예제 #4
0
    def test_old_and_new_constructor_creates_equal_config(self):
        res_config2 = ResConfig(user_config_file=self.filename)
        old = res_config2.hook_manager
        new = HookManager(
            workflow_list=self.res_config.ert_workflow_list,
            config_dict=self.config_data)

        self.assertEqual(old, new)
예제 #5
0
파일: res_config.py 프로젝트: ManInFez/ert
    def _alloc_from_dict(self, config_dict, throw_on_error=True):
        # treat the default config dir
        config_dir = os.getcwd()
        if ConfigKeys.CONFIG_DIRECTORY in config_dict:
            config_dir = config_dict[ConfigKeys.CONFIG_DIRECTORY]
        config_dict[ConfigKeys.CONFIG_DIRECTORY] = config_dir

        subst_config = SubstConfig(config_dict=config_dict)
        site_config = SiteConfig(config_dict=config_dict)
        rng_config = RNGConfig(config_dict=config_dict)
        analysis_config = AnalysisConfig(config_dict=config_dict)
        ecl_config = EclConfig(config_dict=config_dict)
        log_config = LogConfig(config_dict=config_dict)
        queue_config = QueueConfig(config_dict=config_dict)

        ert_workflow_list = ErtWorkflowList(
            subst_list=subst_config.subst_list, config_dict=config_dict
        )

        hook_manager = HookManager(
            workflow_list=ert_workflow_list, config_dict=config_dict
        )

        ert_templates = ErtTemplates(
            parent_subst=subst_config.subst_list, config_dict=config_dict
        )

        ensemble_config = EnsembleConfig(
            grid=ecl_config.getGrid(),
            refcase=ecl_config.getRefcase(),
            config_dict=config_dict,
        )

        model_config = ModelConfig(
            data_root=config_dir,
            joblist=site_config.get_installed_jobs(),
            last_history_restart=ecl_config.getLastHistoryRestart(),
            refcase=ecl_config.getRefcase(),
            config_dict=config_dict,
        )

        return [
            subst_config,
            site_config,
            rng_config,
            analysis_config,
            ert_workflow_list,
            hook_manager,
            ert_templates,
            ecl_config,
            ensemble_config,
            model_config,
            log_config,
            queue_config,
        ], config_dir
예제 #6
0
    def _alloc_from_content(self,
                            user_config_file=None,
                            config=None,
                            throw_on_error=True):
        if user_config_file is not None:
            # initialize configcontent if user_file provided
            parser = ConfigParser()
            config_content = self._alloc_config_content(
                user_config_file, parser)
            config_dir = config_content.getValue(ConfigKeys.CONFIG_DIRECTORY)
        else:
            config_dir = os.getcwd()
            config_content = self._build_config_content(config)

        if self.errors and throw_on_error:
            raise ValueError("Error loading configuration: " +
                             str(self._errors))

        subst_config = SubstConfig(config_content=config_content)
        site_config = SiteConfig(config_content=config_content)
        rng_config = RNGConfig(config_content=config_content)
        analysis_config = AnalysisConfig(config_content=config_content)
        ecl_config = EclConfig(config_content=config_content)
        log_config = LogConfig(config_content=config_content)
        queue_config = QueueConfig(config_content=config_content)

        ert_workflow_list = ErtWorkflowList(subst_list=subst_config.subst_list,
                                            config_content=config_content)

        hook_manager = HookManager(workflow_list=ert_workflow_list,
                                   config_content=config_content)

        ert_templates = ErtTemplates(parent_subst=subst_config.subst_list,
                                     config_content=config_content)

        ensemble_config = EnsembleConfig(config_content=config_content,
                                         grid=ecl_config.getGrid(),
                                         refcase=ecl_config.getRefcase())

        model_config = ModelConfig(
            data_root=config_dir,
            joblist=site_config.get_installed_jobs(),
            last_history_restart=ecl_config.getLastHistoryRestart(),
            refcase=ecl_config.getRefcase(),
            config_content=config_content)

        return [
            subst_config, site_config, rng_config, analysis_config,
            ert_workflow_list, hook_manager, ert_templates, ecl_config,
            ensemble_config, model_config, log_config, queue_config
        ], config_dir
예제 #7
0
    def __init__(self,
                 user_config_file=None,
                 config=None,
                 throw_on_error=True):

        self._errors, self._failed_keys = None, None
        self._assert_input(user_config_file, config, throw_on_error)

        if config is not None:
            config_content = self._build_config_content(config)
        elif user_config_file is not None:

            parser = ConfigParser()

            config_content = self._alloc_config_content(
                user_config_file, parser)
        else:
            raise ValueError("No config provided")

        if self.errors and throw_on_error:
            raise ValueError("Error loading configuration: " +
                             str(self._errors))

        config_dir = config_content.getValue(ConfigKeys.CONFIG_DIRECTORY)

        subst_config = SubstConfig(config_content=config_content)
        site_config = SiteConfig(config_content=config_content)
        rng_config = RNGConfig(config_content=config_content)
        analysis_config = AnalysisConfig(config_content=config_content)
        ecl_config = EclConfig(config_content=config_content)
        log_config = LogConfig(config_content=config_content)
        queue_config = QueueConfig(config_content=config_content)

        ert_workflow_list = ErtWorkflowList(
            ert_workflow_list=subst_config.subst_list,
            config_content=config_content)

        hook_manager = HookManager(workflow_list=ert_workflow_list,
                                   config_content=config_content)

        ert_templates = ErtTemplates(parent_subst=subst_config.subst_list,
                                     config_content=config_content)

        ensemble_config = EnsembleConfig(config_content=config_content,
                                         grid=ecl_config.getGrid(),
                                         refcase=ecl_config.getRefcase())

        model_config = ModelConfig(
            config_content=config_content,
            data_root=config_dir,
            joblist=site_config.get_installed_jobs(),
            last_history_restart=ecl_config.getLastHistoryRestart(),
            sched_file=ecl_config._get_sched_file(),
            refcase=ecl_config.getRefcase())

        configs = [
            subst_config, site_config, rng_config, analysis_config,
            ert_workflow_list, hook_manager, ert_templates, ecl_config,
            ensemble_config, model_config, log_config, queue_config
        ]

        c_ptr = None

        for conf in configs:
            conf.convertToCReference(None)
        c_ptr = self._alloc_full(config_dir, user_config_file, *configs)

        if c_ptr:
            super(ResConfig, self).__init__(c_ptr)
        else:
            raise ValueError(
                'Failed to construct ResConfig instance from %r.' %
                (user_config_file if user_config_file else config))