format

jdebacker · Aug 17, 2024 · a00db08 · a00db08
1 parent a3652ab
commit a00db08
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 8 deletions.
diff --git a/tests/test_output_plots.py b/tests/test_output_plots.py
@@ -50,7 +50,9 @@
     )
 else:
     reform_params = utils.safe_read_pickle(
-        os.path.join(CUR_PATH, "test_io_data", "model_params_baseline_v312.pkl")
+        os.path.join(
+            CUR_PATH, "test_io_data", "model_params_baseline_v312.pkl"
+        )
     )
 reform_taxfunctions = utils.safe_read_pickle(
     os.path.join(CUR_PATH, "test_io_data", "TxFuncEst_reform.pkl")
@@ -228,7 +230,11 @@ def test_plot_industry_aggregates(
 def test_plot_aggregates_save_fig(tmpdir):
     path = os.path.join(tmpdir, "test_plot.png")
     output_plots.plot_aggregates(
-        base_tpi, base_params, start_year=int(base_params.start_year), plot_type="levels", path=path
+        base_tpi,
+        base_params,
+        start_year=int(base_params.start_year),
+        plot_type="levels",
+        path=path,
     )
     img = mpimg.imread(path)
 
@@ -238,7 +244,10 @@ def test_plot_aggregates_save_fig(tmpdir):
 def test_plot_aggregates_not_a_type(tmpdir):
     with pytest.raises(AssertionError):
         output_plots.plot_aggregates(
-            base_tpi, base_params, start_year=int(base_params.start_year), plot_type="levels2"
+            base_tpi,
+            base_params,
+            start_year=int(base_params.start_year),
+            plot_type="levels2",
         )
 
 

diff --git a/tests/test_output_tables.py b/tests/test_output_tables.py
@@ -53,7 +53,9 @@
     )
 else:
     reform_params = utils.safe_read_pickle(
-        os.path.join(CUR_PATH, "test_io_data", "model_params_baseline_v312.pkl")
+        os.path.join(
+            CUR_PATH, "test_io_data", "model_params_baseline_v312.pkl"
+        )
     )
 # add investment tax credit parameter that not in cached parameters
 base_params.inv_tax_credit = np.zeros(

diff --git a/tests/test_parameter_plots.py b/tests/test_parameter_plots.py
@@ -44,9 +44,9 @@
 )
 if base_params.rho.ndim == 1:
     base_params.rho = np.tile(
-            base_params.rho.reshape(1, base_params.S),
-            (base_params.T + base_params.S, 1),
-        )
+        base_params.rho.reshape(1, base_params.S),
+        (base_params.T + base_params.S, 1),
+    )
 
 
 def test_plot_imm_rates():
@@ -107,7 +107,9 @@ def test_plot_pop_growth():
 
 
 def test_plot_pop_growth_rates_save_fig(tmpdir):
-    parameter_plots.plot_pop_growth(base_params, start_year=int(base_params.start_year), path=tmpdir)
+    parameter_plots.plot_pop_growth(
+        base_params, start_year=int(base_params.start_year), path=tmpdir
+    )
     img = mpimg.imread(os.path.join(tmpdir, "pop_growth_rates.png"))
 
     assert isinstance(img, np.ndarray)