Commitin

dafeda · Aug 21, 2024 · 5dc6f1f · 5dc6f1f
1 parent 324e6a0
commit 5dc6f1f
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/integration_tests/analysis/test_es_update.py b/tests/integration_tests/analysis/test_es_update.py
@@ -135,9 +135,9 @@ def sample_prior(nx, ny):
     ert_config = ErtConfig.from_file("snake_oil_surface.ert")
 
     storage = open_storage(ert_config.ens_path)
-
-    ens_prior = storage.get_ensemble_by_name("iter-0")
-    ens_posterior = storage.get_ensemble_by_name("iter-1")
+    experiment = storage.get_experiment_by_name("es")
+    ens_prior = experiment.get_ensemble_by_name("iter-0")
+    ens_posterior = experiment.get_ensemble_by_name("iter-1")
 
     # Check that surfaces defined in INIT_FILES are not changed by ERT
     surf_prior = ens_prior.load_parameters("TOP", list(range(ensemble_size)))["values"]
@@ -173,8 +173,9 @@ def test_update_multiple_param():
     ert_config = ErtConfig.from_file("snake_oil.ert")
 
     storage = open_storage(ert_config.ens_path)
-    prior_ensemble = storage.get_ensemble_by_name("iter-0")
-    posterior_ensemble = storage.get_ensemble_by_name("iter-1")
+    experiment = storage.get_experiment_by_name("es")
+    prior_ensemble = experiment.get_ensemble_by_name("iter-0")
+    posterior_ensemble = experiment.get_ensemble_by_name("iter-1")
 
     prior_array = _all_parameters(prior_ensemble, list(range(10)))
     posterior_array = _all_parameters(posterior_ensemble, list(range(10)))