Skip to content

Commit

Permalink
Use new version of iterative_ensemble_smoother
Browse files Browse the repository at this point in the history
Returns all correlated pairs and not just those deemed
significantly correlated.
Therefore, some tests need to be updated.
  • Loading branch information
dafeda committed Jan 3, 2025
1 parent 987bb0b commit 6a82399
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 8 deletions.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ dependencies = [
"filelock",
"httpx",
"humanize",
"iterative_ensemble_smoother<0.3.0",
"iterative_ensemble_smoother>=0.3.0",
"jinja2 >= 2.10",
"lark",
"lxml",
Expand Down
9 changes: 2 additions & 7 deletions tests/ert/ui_tests/cli/analysis/test_adaptive_localization.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,11 +49,6 @@ def test_that_adaptive_localization_with_cutoff_1_equals_ensemble_prior():
f.writelines(lines)
prior_ensemble, posterior_ensemble = run_cli_ES_with_case("poly_localization_1.ert")

with pytest.raises(
FileNotFoundError, match="No cross-correlation data available at"
):
prior_ensemble.load_cross_correlations()

prior_sample = prior_ensemble.load_parameters("COEFFS")["values"]
posterior_sample = posterior_ensemble.load_parameters("COEFFS")["values"]
# Check prior and posterior samples are equal
Expand Down Expand Up @@ -300,8 +295,8 @@ def test_that_posterior_generalized_variance_increases_in_cutoff():
)

cross_correlations = prior_ensemble_cutoff1.load_cross_correlations()
assert all(cross_correlations.parameter.to_numpy() == ["a", "b"])
assert cross_correlations["COEFFS"].values.shape == (2, 5)
assert all(cross_correlations.parameter.to_numpy() == ["a", "b", "c"])
assert cross_correlations["COEFFS"].values.shape == (3, 5)
assert (
(cross_correlations["COEFFS"].values >= -1)
& (cross_correlations["COEFFS"].values <= 1)
Expand Down

0 comments on commit 6a82399

Please sign in to comment.