Remove suppress statement by providing a correct design matrix

Author: xjules

tests/ert/ui_tests/cli/analysis/test_design_matrix.py

@@ -52,8 +52,6 @@ def test_run_poly_example_with_design_matrix():
             dedent(
                 """\
                 #!/usr/bin/env python
-                import numpy as np
-                import sys
                 import json
 
                 def _load_coeffs(filename):
@@ -141,8 +139,6 @@ def test_run_poly_example_with_design_matrix_and_genkw_merge(default_values, err
             dedent(
                 """\
                 #!/usr/bin/env python
-                import numpy as np
-                import sys
                 import json
 
                 def _load_coeffs(filename):

tests/ert/unit_tests/config/test_analysis_config.py

@@ -1,7 +1,7 @@
-from contextlib import suppress
 from textwrap import dedent
 
 import hypothesis.strategies as st
+import pandas as pd
 import pytest
 from hypothesis import given
 
@@ -16,33 +16,43 @@
 
 
 def test_analysis_config_from_file_is_same_as_from_dict(monkeypatch, tmp_path):
-    with open(tmp_path / "my_design_matrix.xlsx", "w", encoding="utf-8"):
-        pass
-    monkeypatch.chdir(tmp_path)
-    with suppress(ConfigValidationError):
-        assert ErtConfig.from_file_contents(
-            dedent(
-                """
-                    NUM_REALIZATIONS 10
-                    MIN_REALIZATIONS 10
-                    ANALYSIS_SET_VAR STD_ENKF ENKF_TRUNCATION 0.8
-                    DESIGN_MATRIX my_design_matrix.xlsx DESIGN_SHEET:my_sheet DEFAULT_SHEET:my_default_sheet
-                    """
-            )
-        ).analysis_config == AnalysisConfig.from_dict(
+    with pd.ExcelWriter(tmp_path / "my_design_matrix.xlsx") as xl_write:
+        design_matrix_df = pd.DataFrame(
             {
-                ConfigKeys.NUM_REALIZATIONS: 10,
-                ConfigKeys.MIN_REALIZATIONS: "10",
-                ConfigKeys.ANALYSIS_SET_VAR: [
-                    ("STD_ENKF", "ENKF_TRUNCATION", 0.8),
-                ],
-                ConfigKeys.DESIGN_MATRIX: [
-                    "my_design_matrix.xlsx",
-                    "DESIGN_SHEET:my_sheet",
-                    "DEFAULT_SHEET:my_default_sheet",
-                ],
+                "REAL": [0, 1, 2],
+                "a": [1, 2, 3],
+                "b": [0, 2, 0],
             }
         )
+        default_sheet_df = pd.DataFrame([["a", 1], ["b", 4]])
+        design_matrix_df.to_excel(xl_write, index=False, sheet_name="my_sheet")
+        default_sheet_df.to_excel(
+            xl_write, index=False, sheet_name="my_default_sheet", header=False
+        )
+    monkeypatch.chdir(tmp_path)
+    assert ErtConfig.from_file_contents(
+        dedent(
+            """
+                NUM_REALIZATIONS 3
+                MIN_REALIZATIONS 3
+                ANALYSIS_SET_VAR STD_ENKF ENKF_TRUNCATION 0.8
+                DESIGN_MATRIX my_design_matrix.xlsx DESIGN_SHEET:my_sheet DEFAULT_SHEET:my_default_sheet
+                """
+        )
+    ).analysis_config == AnalysisConfig.from_dict(
+        {
+            ConfigKeys.NUM_REALIZATIONS: 3,
+            ConfigKeys.MIN_REALIZATIONS: "3",
+            ConfigKeys.ANALYSIS_SET_VAR: [
+                ("STD_ENKF", "ENKF_TRUNCATION", 0.8),
+            ],
+            ConfigKeys.DESIGN_MATRIX: [
+                "my_design_matrix.xlsx",
+                "DESIGN_SHEET:my_sheet",
+                "DEFAULT_SHEET:my_default_sheet",
+            ],
+        }
+    )
 
 
 @pytest.mark.filterwarnings(