def test_sensitivity_fast_run_and_export( workspace_integration, sensitivity_ensemble, stages_config, sensitivity_fast_experiment_config, gaussian_parameters_config, ): workspace = workspace_integration (workspace._path / _EXPERIMENTS_BASE / "sensitivity").mkdir(parents=True) experiment_run_config = ert3.config.ExperimentRunConfig( sensitivity_fast_experiment_config, stages_config, sensitivity_ensemble, gaussian_parameters_config, ) with assert_clean_workspace(workspace, allowed_files={"fast_analysis.json"}): ert3.engine.run_sensitivity_analysis( experiment_run_config, workspace, "sensitivity" ) with assert_clean_workspace(workspace, allowed_files={"data.json"}): ert3.engine.export(workspace, "sensitivity", experiment_run_config) export_data = _load_export_data(workspace, "sensitivity") assert_sensitivity_export( export_data, sensitivity_ensemble, stages_config, gaussian_parameters_config, "fast", )
def test_partial_sensitivity_run_and_export( workspace_integration, partial_sensitivity_ensemble, double_stages_config, sensitivity_oat_experiment_config, oat_compatible_record_file, gaussian_parameters_config, ): workspace = workspace_integration experiment_dir = workspace._path / _EXPERIMENTS_BASE / "partial_sensitivity" assert experiment_dir.is_dir() experiment_run_config = ert3.config.ExperimentRunConfig( sensitivity_oat_experiment_config, double_stages_config, partial_sensitivity_ensemble, gaussian_parameters_config, ) with assert_clean_workspace(workspace): transformation = ert.data.SerializationTransformation( location=oat_compatible_record_file, mime="application/json", ) get_event_loop().run_until_complete( ert3.engine.load_record( workspace, "other_coefficients", transformation, )) ert3.engine.run_sensitivity_analysis(experiment_run_config, workspace, "partial_sensitivity") with assert_clean_workspace(workspace, allowed_files={"data.json"}): ert3.engine.export(workspace, "partial_sensitivity", experiment_run_config) export_data = _load_export_data(workspace, "partial_sensitivity") assert_sensitivity_export( export_data, partial_sensitivity_ensemble, double_stages_config, gaussian_parameters_config, "one-at-a-time", )