Add mutation of response keys to stateful storage test

yngve-sk · Oct 23, 2024 · 81463d6 · 81463d6
1 parent fe8453e
commit 81463d6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/ert/unit_tests/storage/test_local_storage.py b/tests/ert/unit_tests/storage/test_local_storage.py
@@ -754,6 +754,16 @@ def save_summary(self, model_ensemble: Ensemble, summary_data):
 
         model_ensemble.response_values[summary.name] = ds
 
+        model_experiment = self.model[storage_experiment.id]
+        response_keys = set(ds["response_key"].unique())
+
+        model_smry_config = next(
+            config for config in model_experiment.responses if config.name == "summary"
+        )
+        model_smry_config.keys = sorted(
+            set(model_smry_config.keys).union(response_keys)
+        )
+
     @rule(model_ensemble=ensembles)
     def get_responses(self, model_ensemble: Ensemble):
         storage_ensemble = self.storage.get_ensemble(model_ensemble.uuid)