Skip to content

Commit 062c6a2

Browse files
eilskragithub-actions[bot]
authored andcommitted
Add regression test for multi obj with constraint
(cherry picked from commit b264d73)
1 parent a92245c commit 062c6a2

1 file changed

Lines changed: 23 additions & 0 deletions

File tree

tests/everest/test_everest_storage.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,3 +111,26 @@ def test_everest_data_stored_in_ert_local_storage(
111111
]
112112

113113
assert local_storage_params == formatted_control_names
114+
115+
116+
@pytest.mark.xdist_group("math_func/config_multiobj_advanced.yml")
117+
@pytest.mark.slow
118+
def test_that_constraint_gradients_have_one_row_per_control_with_multiple_objectives(
119+
cached_example,
120+
):
121+
config_path, config_file, _, _ = cached_example(
122+
"math_func/config_multiobj_advanced.yml"
123+
)
124+
config = EverestConfig.load_file(Path(config_path) / config_file)
125+
experiment = EverestStorage.get_everest_experiment(storage_path=config.storage_dir)
126+
127+
gradients = [
128+
ensemble.batch_constraint_gradient
129+
for ensemble in experiment.ensembles
130+
if ensemble.batch_constraint_gradient is not None
131+
]
132+
133+
assert gradients
134+
for gradient in gradients:
135+
assert gradient.columns == ["batch_id", "control_name", "origin_distance"]
136+
assert gradient["control_name"].to_list() == ["point.x", "point.y", "point.z"]

0 commit comments

Comments
 (0)