create rather than read in params

Author: jdebacker

tests/test_output_plots.py

@@ -9,6 +9,7 @@
 import matplotlib.image as mpimg
 import matplotlib.pyplot as plt
 from ogcore import utils, output_plots, constants
+from ogcore.parameters import Specifications
 
 
 # Load in test results and parameters
@@ -19,42 +20,36 @@
 base_tpi = utils.safe_read_pickle(
     os.path.join(CUR_PATH, "test_io_data", "TPI_vars_baseline.pkl")
 )
-if sys.version_info[1] < 11:
-    base_params = utils.safe_read_pickle(
-        os.path.join(CUR_PATH, "test_io_data", "model_params_baseline.pkl")
-    )
-elif sys.version_info[1] == 11:
-    base_params = utils.safe_read_pickle(
-        os.path.join(
-            CUR_PATH, "test_io_data", "model_params_baseline_v311.pkl"
-        )
-    )
-else:
-    base_params = utils.safe_read_pickle(
-        os.path.join(
-            CUR_PATH, "test_io_data", "model_params_baseline_v312.pkl"
-        )
-    )
+base_params = Specifications()
+base_params.update_specifications(
+    {
+        "M": 3,
+        "gamma": [0.5, 0.5, 0.5],
+        "gamma_g": [0.0, 0.0, 0.0],
+        "epsilon": [0.5, 0.5, 0.5],
+        "I": 3,
+        "alpha_c": [0.3, 0.4, 0.3],
+        "io_matrix": np.eye(3)
+    }
+)
 reform_ss = utils.safe_read_pickle(
     os.path.join(CUR_PATH, "test_io_data", "SS_vars_reform.pkl")
 )
 reform_tpi = utils.safe_read_pickle(
     os.path.join(CUR_PATH, "test_io_data", "TPI_vars_reform.pkl")
 )
-if sys.version_info[1] < 11:
-    reform_params = utils.safe_read_pickle(
-        os.path.join(CUR_PATH, "test_io_data", "model_params_reform.pkl")
-    )
-elif sys.version_info[1] == 11:
-    reform_params = utils.safe_read_pickle(
-        os.path.join(CUR_PATH, "test_io_data", "model_params_reform_v311.pkl")
-    )
-else:
-    reform_params = utils.safe_read_pickle(
-        os.path.join(
-            CUR_PATH, "test_io_data", "model_params_baseline_v312.pkl"
-        )
-    )
+reform_params = Specifications()
+reform_params.update_specifications(
+    {
+        "M": 3,
+        "gamma": [0.5, 0.5, 0.5],
+        "gamma_g": [0.0, 0.0, 0.0],
+        "epsilon": [0.5, 0.5, 0.5],
+        "I": 3,
+        "alpha_c": [0.3, 0.4, 0.3],
+        "io_matrix": np.eye(3)
+    }
+)
 reform_taxfunctions = utils.safe_read_pickle(
     os.path.join(CUR_PATH, "test_io_data", "TxFuncEst_reform.pkl")
 )