diff --git a/tests/workflows/ahm_analysis/test_ahm_analysis.py b/tests/workflows/ahm_analysis/test_ahm_analysis.py index bdcd572f..e6fdf763 100644 --- a/tests/workflows/ahm_analysis/test_ahm_analysis.py +++ b/tests/workflows/ahm_analysis/test_ahm_analysis.py @@ -17,7 +17,8 @@ def test_make_update_log_df(snake_oil_facade, snapshot): snapshots are correct, they are just documenting the current behavior. """ with open_storage(snake_oil_facade.enspath, "w") as storage: - prior_ens = storage.get_ensemble_by_name("default") + experiment = storage.get_experiment_by_name("default_experiment_name") + prior_ens = experiment.get_ensemble_by_name("default") posterior_ens = storage.create_ensemble( prior_ens.experiment_id, ensemble_size=prior_ens.ensemble_size, diff --git a/tests/workflows/ahm_analysis/test_integration.py b/tests/workflows/ahm_analysis/test_integration.py index c4c9b81a..ef21bb40 100644 --- a/tests/workflows/ahm_analysis/test_integration.py +++ b/tests/workflows/ahm_analysis/test_integration.py @@ -14,10 +14,11 @@ def test_ahmanalysis_run(snake_oil_facade): """test data_set with only scalar parameters""" with open_storage(snake_oil_facade.enspath, "w") as storage: + experiment = storage.get_experiment_by_name("default_experiment_name") snake_oil_facade.run_ertscript( ahmanalysis.AhmAnalysisJob, storage, - storage.get_ensemble_by_name("default"), + experiment.get_ensemble_by_name("default"), ) # assert that this returns/generates a KS csv file @@ -70,10 +71,11 @@ def test_ahmanalysis_run_deactivated_obs(snake_oil_facade, snapshot, caplog): with open_storage(snake_oil_facade.enspath, "w") as storage, caplog.at_level( logging.WARNING ): + experiment = storage.get_experiment_by_name("default_experiment_name") snake_oil_facade.run_ertscript( ahmanalysis.AhmAnalysisJob, storage, - storage.get_ensemble_by_name("default"), + experiment.get_ensemble_by_name("default"), ) assert "Analysis failed for" in caplog.text