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",
)