Fixup test_storage_migration

equinor · Oct 1, 2024 · 7b8976a · 7b8976a
1 parent 7665159
commit 7b8976a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/...nit_tests/storage/snapshots/test_storage_migration/test_that_storage_matches/summary_data b/...nit_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
@@ -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")