Improve AL tests

equinor · Oct 13, 2023 · a12530e · a12530e
1 parent 343f552
commit a12530e
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 65 deletions.
diff --git a/test-data/poly_example/poly_loc_0.ert b/test-data/poly_example/poly_loc_0.ert
diff --git a/test-data/poly_example/poly_loc_1.ert b/test-data/poly_example/poly_loc_1.ert
diff --git a/test-data/poly_example/poly_no_loc.ert b/test-data/poly_example/poly_no_loc.ert
diff --git a/tests/unit_tests/analysis/test_adaptive_localization.py b/tests/unit_tests/analysis/test_adaptive_localization.py
@@ -46,7 +46,7 @@ def run_cli_ES_with_case(poly_config):
 def test_that_adaptive_localization_with_cutoff_1_equals_ensemble_prior(copy_case):
     copy_case("poly_example")
     random_seed_line = "RANDOM_SEED 1234\n\n"
-    set_adaptive_localization = dedent(
+    set_adaptive_localization_1 = dedent(
         """
         ANALYSIS_SET_VAR STD_ENKF LOCALIZATION True
         ANALYSIS_SET_VAR STD_ENKF LOCALIZATION_CORRELATION_THRESHOLD 1.0
@@ -56,7 +56,7 @@ def test_that_adaptive_localization_with_cutoff_1_equals_ensemble_prior(copy_cas
     with open("poly.ert", "r+") as f:
         lines = f.readlines()
         lines.insert(2, random_seed_line)
-        lines.insert(10, set_adaptive_localization)
+        lines.insert(10, set_adaptive_localization_1)
 
     with open("poly_loc_1.ert", "w") as f:
         f.writelines(lines)
@@ -74,6 +74,26 @@ def test_that_adaptive_localization_with_cutoff_0_equals_ESupdate(copy_case):
     """
     copy_case("poly_example")
 
+    random_seed_line = "RANDOM_SEED 1234\n\n"
+    set_adaptive_localization_0 = dedent(
+        """
+        ANALYSIS_SET_VAR STD_ENKF LOCALIZATION True
+        ANALYSIS_SET_VAR STD_ENKF LOCALIZATION_CORRELATION_THRESHOLD 0.0
+        """
+    )
+
+    with open("poly.ert", "r+") as f:
+        lines = f.readlines()
+        lines.insert(2, random_seed_line)
+
+    with open("poly_no_loc.ert", "w") as f:
+        f.writelines(lines)
+
+    lines.insert(10, set_adaptive_localization_0)
+
+    with open("poly_loc_0.ert", "w") as f:
+        f.writelines(lines)
+
     prior_sample_loc0, posterior_sample_loc0 = run_cli_ES_with_case("poly_loc_0.ert")
     prior_sample_noloc, posterior_sample_noloc = run_cli_ES_with_case("poly_no_loc.ert")