Replace usage of a deprecated function in a test

equinor · Mar 14, 2024 · c57b5e1 · c57b5e1
1 parent ab97e82
commit c57b5e1
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/tests/unit_tests/test_summary_response.py b/tests/unit_tests/test_summary_response.py
@@ -1,4 +1,3 @@
-import os
 import shutil
 from pathlib import Path
 from textwrap import dedent
@@ -26,7 +25,6 @@ def test_load_summary_response_restart_not_zero(tmpdir, snapshot, request, stora
         )
         with open("config.ert", "w", encoding="utf-8") as fh:
             fh.writelines(config)
-        cwd = Path().absolute()
         sim_path = Path("simulations") / "realization-0" / "iter-0"
         ert_config = ErtConfig.from_file("config.ert")
 
@@ -49,15 +47,19 @@ def test_load_summary_response_restart_not_zero(tmpdir, snapshot, request, stora
         )
 
         create_run_path(prior, ert_config)
-        os.chdir(sim_path)
-        shutil.copy(test_path / "PRED_RUN.SMSPEC", "PRED_RUN.SMSPEC")
-        shutil.copy(test_path / "PRED_RUN.UNSMRY", "PRED_RUN.UNSMRY")
-        os.chdir(cwd)
+        shutil.copy(test_path / "PRED_RUN.SMSPEC", sim_path / "PRED_RUN.SMSPEC")
+        shutil.copy(test_path / "PRED_RUN.UNSMRY", sim_path / "PRED_RUN.UNSMRY")
 
         facade = LibresFacade.from_config_file("config.ert")
         facade.load_from_forward_model(ensemble, [True], 0)
 
+        df = ensemble.load_responses("summary", (0,)).to_dataframe()
+        df = df.unstack(level="name")
+        df.columns = [col[1] for col in df.columns.values]
+        df.index = df.index.rename(
+            {"time": "Date", "realization": "Realization"}
+        ).reorder_levels(["Realization", "Date"])
         snapshot.assert_match(
-            ensemble.load_all_summary_data().dropna().iloc[:, :15].to_csv(),
+            df.dropna().iloc[:, :15].to_csv(),
             "summary_restart",
         )