Skip to content

Commit

Permalink
Add test for migration from version 1
Browse files Browse the repository at this point in the history
  • Loading branch information
oyvindeide committed Sep 20, 2023
1 parent 73d98ff commit 1072844
Show file tree
Hide file tree
Showing 3 changed files with 28 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/ert/storage/migration/response_info.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def migrate(path: Path) -> None:
response_info = {}
for response in ens_config.response_configuration:
response_info[response.name] = response.to_dict()
with open(experiment / "response.json", "w", encoding="utf-8") as fout:
with open(experiment / "responses.json", "w", encoding="utf-8") as fout:
fout.write(json.dumps(response_info))
with open(path / "index.json", encoding="utf-8") as f:
index_json = json.load(f)
Expand Down
2 changes: 1 addition & 1 deletion test-data/block_storage
Submodule block_storage updated 54 files
+1 −0 version-1/poly_example/.ert_runpath_list
+2 −0 version-1/poly_example/ASSERT_RUNPATH_FILE
+1 −0 version-1/poly_example/POLY_EVAL
+62 −0 version-1/poly_example/README.md
+1 −0 version-1/poly_example/TEST_RUNPATH_FILE
+34 −0 version-1/poly_example/assert_runpath_file.py
+5 −0 version-1/poly_example/coeff.tmpl
+3 −0 version-1/poly_example/coeff_priors
+5 −0 version-1/poly_example/observations
+17 −0 version-1/poly_example/poly-ies.ert
+17 −0 version-1/poly_example/poly.ert
+19 −0 version-1/poly_example/poly_eval.py
+5 −0 version-1/poly_example/poly_obs_data.txt
+1 −0 version-1/poly_example/storage/.fs_version
+1 −0 version-1/poly_example/storage/ensembles/3a6fc805-f2b0-4dc7-9621-cd97e466cfa4/experiment/state_map.json
+ version-1/poly_example/storage/ensembles/3a6fc805-f2b0-4dc7-9621-cd97e466cfa4/experiment/time_map
+1 −0 version-1/poly_example/storage/ensembles/3a6fc805-f2b0-4dc7-9621-cd97e466cfa4/index.json
+ version-1/poly_example/storage/ensembles/3a6fc805-f2b0-4dc7-9621-cd97e466cfa4/realization-0/COEFFS.nc
+ version-1/poly_example/storage/ensembles/3a6fc805-f2b0-4dc7-9621-cd97e466cfa4/realization-0/POLY_RES.nc
+1 −0 version-1/poly_example/storage/experiments/8eb2ad69-b7ea-4f9c-9d67-2047c450eeb9/gen-kw-priors.json
+1 −0 version-1/poly_example/storage/experiments/8eb2ad69-b7ea-4f9c-9d67-2047c450eeb9/parameter.json
+1 −0 version-1/poly_example/storage/index.json
+1 −0 version-2/snake_oil/.ert_runpath_list
+1 −0 version-2/snake_oil/SEED
+4 −0 version-2/snake_oil/jobs/SNAKE_OIL_DIFF
+4 −0 version-2/snake_oil/jobs/SNAKE_OIL_NPV
+4 −0 version-2/snake_oil/jobs/SNAKE_OIL_SIMULATOR
+29 −0 version-2/snake_oil/jobs/snake_oil_diff.py
+104 −0 version-2/snake_oil/jobs/snake_oil_npv.py
+204 −0 version-2/snake_oil/jobs/snake_oil_simulator.py
+56 −0 version-2/snake_oil/observations/observations.txt
+4 −0 version-2/snake_oil/observations/wpr_diff_obs.txt
+11 −0 version-2/snake_oil/parameters/snake_oil_parameters.txt
+3 −0 version-2/snake_oil/refcase/SNAKE_OIL_FIELD.SMSPEC
+3 −0 version-2/snake_oil/refcase/SNAKE_OIL_FIELD.UNSMRY
+1 −0 version-2/snake_oil/refcase/refcase_readme.txt
+1 −0 version-2/snake_oil/refcase/seed.txt
+10 −0 version-2/snake_oil/refcase/snake_oil_params.txt
+2,000 −0 version-2/snake_oil/refcase/time_map.txt
+35 −0 version-2/snake_oil/snake_oil.ert
+1 −0 version-2/snake_oil/storage/snake_oil/ensemble/.fs_version
+1 −0 ...ake_oil/storage/snake_oil/ensemble/ensembles/1e79599b-b386-4d97-a0e2-8201d6222a88/experiment/state_map.json
+1 −0 version-2/snake_oil/storage/snake_oil/ensemble/ensembles/1e79599b-b386-4d97-a0e2-8201d6222a88/index.json
+ ...orage/snake_oil/ensemble/ensembles/1e79599b-b386-4d97-a0e2-8201d6222a88/realization-0/SNAKE_OIL_GPR_DIFF.nc
+ ...orage/snake_oil/ensemble/ensembles/1e79599b-b386-4d97-a0e2-8201d6222a88/realization-0/SNAKE_OIL_OPR_DIFF.nc
+ .../storage/snake_oil/ensemble/ensembles/1e79599b-b386-4d97-a0e2-8201d6222a88/realization-0/SNAKE_OIL_PARAM.nc
+ ...orage/snake_oil/ensemble/ensembles/1e79599b-b386-4d97-a0e2-8201d6222a88/realization-0/SNAKE_OIL_WPR_DIFF.nc
+ ...nake_oil/storage/snake_oil/ensemble/ensembles/1e79599b-b386-4d97-a0e2-8201d6222a88/realization-0/summary.nc
+1 −0 version-2/snake_oil/storage/snake_oil/ensemble/experiments/2598a405-3733-42ef-8a44-46fa43bfe18f/parameter.json
+1 −0 ...e_oil/storage/snake_oil/ensemble/experiments/2598a405-3733-42ef-8a44-46fa43bfe18f/simulation_arguments.json
+10 −0 ...nake_oil/storage/snake_oil/ensemble/experiments/2598a405-3733-42ef-8a44-46fa43bfe18f/snake_oil_template.txt
+1 −0 version-2/snake_oil/storage/snake_oil/ensemble/index.json
+1 −0 version-2/snake_oil/templates/seed_template.txt
+10 −0 version-2/snake_oil/templates/snake_oil_template.txt
26 changes: 26 additions & 0 deletions tests/unit_tests/storage/migration/test_version_1.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
import pytest
import json

from ert.config import ErtConfig
from ert.storage.local_storage import local_storage_set_ert_config
from ert.storage import open_storage


@pytest.fixture(scope="module", autouse=True)
def set_ert_config(block_storage_path):
ert_config = ErtConfig.from_file(
str(block_storage_path / "version-1/poly_example/poly.ert")
)
yield local_storage_set_ert_config(ert_config)
local_storage_set_ert_config(None)


def test_migrate_gen_kw(setup_case, set_ert_config):
setup_case("block_storage/version-1/poly_example", "poly.ert")
with open_storage("storage", "w") as storage:
assert len(list(storage.experiments)) == 1
experiment = list(storage.experiments)[0]
param_info = json.loads(
(experiment._path / "parameter.json").read_text(encoding="utf-8")
)
assert "COEFFS" in param_info

0 comments on commit 1072844

Please sign in to comment.