From 7b8976a030a89f9965ec560264639077c3990de9 Mon Sep 17 00:00:00 2001 From: "Yngve S. Kristiansen" Date: Tue, 1 Oct 2024 09:09:16 +0200 Subject: [PATCH] Fixup test_storage_migration --- .../test_that_storage_matches/summary_data | 2 +- tests/ert/unit_tests/storage/test_storage_migration.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/ert/unit_tests/storage/snapshots/test_storage_migration/test_that_storage_matches/summary_data b/tests/ert/unit_tests/storage/snapshots/test_storage_migration/test_that_storage_matches/summary_data index 38b654b284c..ddb87262c2b 100644 --- a/tests/ert/unit_tests/storage/snapshots/test_storage_migration/test_that_storage_matches/summary_data +++ b/tests/ert/unit_tests/storage/snapshots/test_storage_migration/test_that_storage_matches/summary_data @@ -1,4 +1,4 @@ -time,name,realization,values +time,response_key,realization,values 1996-01-02,FOPR,0,1.1 1996-01-02,FOPR,1,1.1 1996-01-02,FOPR,2,1.1 diff --git a/tests/ert/unit_tests/storage/test_storage_migration.py b/tests/ert/unit_tests/storage/test_storage_migration.py index 8e458c1fccd..4d526045845 100644 --- a/tests/ert/unit_tests/storage/test_storage_migration.py +++ b/tests/ert/unit_tests/storage/test_storage_migration.py @@ -169,9 +169,9 @@ def test_that_storage_matches( tuple(ensemble.get_realization_list_with_responses("summary")), ) snapshot.assert_match( - summary_data.sort("time", "name", "realization") + summary_data.sort("time", "response_key", "realization") .to_pandas() - .set_index(["time", "name", "realization"]) + .set_index(["time", "response_key", "realization"]) .transform(np.sort) .to_csv(), "summary_data", @@ -184,6 +184,8 @@ def test_that_storage_matches( "observations", ) + assert ensemble.get_summary_keyset() == ["FOPR"] + @pytest.mark.integration_test @pytest.mark.usefixtures("copy_shared")