fixing E501 line too long linting issues in tests

tests/conftest.py

@@ -34,7 +34,10 @@
 
 
 def refresh_databases() -> None:
-    """make new databases from source for testing...  removes possibility of contamination by earlier runs"""
+    """
+    make new databases from source for testing...  removes possibility of contamination by earlier
+    runs
+    """
     data_output_path = Path(__file__).parent / 'testing_outputs'
     data_source_path = Path(__file__).parent / 'testing_data'
     databases = (

tests/test_cli.py

@@ -489,7 +489,8 @@ def test_cli_validate_fails_if_solver_missing(
     tmp_path: Path, monkeypatch: pytest.MonkeyPatch
 ) -> None:
     """
-    Test that the validate command fails with SolverNotAvailableError if the configured solver is missing.
+    Test that the validate command fails with SolverNotAvailableError if the configured solver is
+    missing.
     """
     db_path = Path(__file__).parent / 'testing_outputs' / 'utopia.sqlite'
     test_config_path = create_config_with_solver(tmp_path, db_path, 'nonexistent_solver')
@@ -516,7 +517,8 @@ def test_cli_validate_fails_if_solver_missing(
 
 def test_cli_run_fails_if_solver_missing(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
     """
-    Test that the run command fails with SolverNotAvailableError if the configured solver is missing.
+    Test that the run command fails with SolverNotAvailableError if the configured solver is
+    missing.
     """
     db_path = Path(__file__).parent / 'testing_outputs' / 'utopia.sqlite'
     test_config_path = create_config_with_solver(tmp_path, db_path, 'another_nonexistent_solver')

tests/test_emission_results.py

@@ -86,7 +86,8 @@ def test_emissions(solved_connection: SolvedConnection) -> None:
     emis = (
         con.cursor()
         .execute(
-            f"SELECT SUM(emission) FROM main.output_emission WHERE tech LIKE '{tech}' AND tech != 'TechEmbodied' AND period == 2000"
+            f"SELECT SUM(emission) FROM main.output_emission WHERE tech LIKE '{tech}' AND "
+            f"tech != 'TechEmbodied' AND period == 2000"
         )
         .fetchone()[0]
     )
@@ -112,7 +113,8 @@ def test_emissions_costs_undiscounted(
     ec = (
         con.cursor()
         .execute(
-            f"SELECT SUM(emiss) FROM main.output_cost WHERE tech LIKE '{tech}' AND tech != 'TechEmbodied' AND period == 2000"
+            f"SELECT SUM(emiss) FROM main.output_cost WHERE tech LIKE '{tech}' AND "
+            f"tech != 'TechEmbodied' AND period == 2000"
         )
         .fetchone()[0]
     )
@@ -139,7 +141,8 @@ def test_emissions_costs_discounted(
     ec = (
         con.cursor()
         .execute(
-            f"SELECT SUM(d_emiss) FROM main.output_cost WHERE tech LIKE '{tech}' AND tech != 'TechEmbodied' AND period == 2000"
+            f"SELECT SUM(d_emiss) FROM main.output_cost WHERE tech LIKE '{tech}' AND "
+            f"tech != 'TechEmbodied' AND period == 2000"
         )
         .fetchone()[0]
     )
@@ -158,13 +161,15 @@ def test_emissions_costs_discounted(
 )
 def test_embodied_emissions(solved_connection: SolvedConnection) -> None:
     """
-    Test that the embodied emissions from each technology archetype are correct, and check total emissions
+    Test that the embodied emissions from each technology archetype are correct, and check total
+    emissions
     """
     con, name, tech, emis_target = solved_connection
     emis = (
         con.cursor()
         .execute(
-            f"SELECT SUM(emission) FROM main.output_emission WHERE tech LIKE '{tech}' AND period == 2000"
+            f"SELECT SUM(emission) FROM main.output_emission WHERE tech LIKE '{tech}' AND "
+            f'period == 2000'
         )
         .fetchone()[0]
     )
@@ -196,7 +201,8 @@ def test_embodied_emissions_costs_undiscounted(
     )
     cost_target = 0.7 * emis_target  # emission cost x embodied emissions
     assert ec == pytest.approx(cost_target), (
-        f'{name} undiscounted embodied emission costs were incorrect. Should be {cost_target}, got {ec}'
+        f'{name} undiscounted embodied emission costs were incorrect. Should be {cost_target}, '
+        f'got {ec}'
     )
 
 
@@ -238,13 +244,15 @@ def test_embodied_emissions_costs_discounted(
 )
 def test_endoflife_emissions(solved_connection: SolvedConnection) -> None:
     """
-    Test that the end of life emissions from each technology archetype are correct, and check total emissions
+    Test that the end of life emissions from each technology archetype are correct, and check total
+    emissions
     """
     con, name, tech, emis_target = solved_connection
     emis = (
         con.cursor()
         .execute(
-            f"SELECT SUM(emission) FROM main.output_emission WHERE tech LIKE '{tech}' AND period == 2005"
+            f"SELECT SUM(emission) FROM main.output_emission WHERE tech LIKE '{tech}' AND "
+            f'period == 2005'
         )
         .fetchone()[0]
     )
@@ -276,7 +284,8 @@ def test_endoflife_emissions_costs_undiscounted(
     )
     cost_target = 0.7 * emis_target  # emission cost x end of life emissions
     assert ec == pytest.approx(cost_target), (
-        f'{name} undiscounted end of life emission costs were incorrect. Should be {cost_target}, got {ec}'
+        f'{name} undiscounted end of life emission costs were incorrect. Should be {cost_target}, '
+        f'got {ec}'
     )
 
 
@@ -331,7 +340,8 @@ def test_curtailment(solved_connection: SolvedConnection) -> None:
     curt = (
         con.cursor()
         .execute(
-            f"SELECT SUM(curtailment) FROM main.output_curtailment WHERE tech LIKE '{tech}' AND period == 2000"
+            f"SELECT SUM(curtailment) FROM main.output_curtailment WHERE tech LIKE '{tech}' AND "
+            f'period == 2000'
         )
         .fetchone()[0]
     )

tests/test_full_runs.py

@@ -66,7 +66,8 @@ def test_against_legacy_outputs(
         len(tuple(efficiency_param.sparse_iterkeys())) == expected_vals[ExpectedVals.EFF_INDEX_SIZE]
     ), 'should match legacy numbers'
 
-    # check the size of the domain.  NOTE:  The build of the domain here may be "expensive" for large models
+    # check the size of the domain.  NOTE:  The build of the domain here may be "expensive" for
+    # large models
     assert (
         len(efficiency_param.index_set().domain) == expected_vals[ExpectedVals.EFF_DOMAIN_SIZE]
     ), 'should match legacy numbers'

tests/test_linked_tech.py

@@ -46,7 +46,8 @@ def test_linked_tech(
         'the linked processes should remove have an aggregate -30 units of co2 emissions'
     )
 
-    # check the flow out of captured carbon from the driven tech, which should output the captured carbon
+    # check the flow out of captured carbon from the driven tech, which should output the captured
+    # carbon
     flow_out = conn.execute(
         "SELECT SUM(flow) FROM output_flow_out WHERE tech = 'CCS' and output_comm = 'CO2_CAP'"
     ).fetchone()[0]

tests/test_material_results.py

@@ -66,7 +66,8 @@ def test_flows(solved_connection: tuple[sqlite3.Connection, str, str, int, float
         'SELECT SUM(flow) FROM main.output_flow_out WHERE tech = ? AND period = ?',
         (tech, period),
     ).fetchone()
-    # If the query returns no rows, row will be None. If it finds rows but the sum is NULL, row[0] will be None.
+    # If the query returns no rows, row will be None. If it finds rows but the sum is NULL, row[0]
+    # will be None.
     flow = row[0] if row and row[0] is not None else 0.0
 
     assert flow == pytest.approx(

tests/test_set_consistency.py

@@ -63,7 +63,8 @@ def test_set_consistency(
             overage_in_model[set_name] = s - cached_set
             shortage_in_model[set_name] = cached_set - s
     missing_in_model = cached_sets.keys() - model_sets.keys()
-    # drop any set that has "_index" in the name as they are no longer reported by newer version of pyomo
+    # drop any set that has "_index" in the name as they are no longer reported by newer version of
+    # pyomo
     missing_in_model = {s for s in missing_in_model if '_index' not in s and '_domain' not in s}
 
     if overage_in_model:

tests/test_storage.py

@@ -86,7 +86,8 @@ def test_state_sequencing(system_test_run: tuple[str, Any, TemoaModel, Any]) ->
         next_state = model.v_storage_level[r, p, s_next, d_next, t, v].value  # type: ignore [attr-defined] # I can't figure out how to get mypy to see value through the pyomo stubs
 
         assert state + charge - discharge == pytest.approx(next_state, abs=1e-5), (
-            f'model fails to correctly sequence storage states {r, p, s, t, v} sequenced {s, d} to {s_next, d_next}'
+            f'model fails to correctly sequence storage states {r, p, s, t, v} sequenced {s, d} '
+            f'to {s_next, d_next}'
         )
 
 

tests/test_tech_activity_vectors.py

@@ -20,7 +20,8 @@ def test__vector_engine() -> None:
     }
     var_map = {'dog': ['red', 'blue'], 'pig': ['yellow', 'green'], 'cat': ['blue', 'gold']}
     tech_sizes = {k: len(v) for k, v in var_map.items()}
-    # below is just to show the mapping back to variables.... test just want the coefficients from res_values
+    # below is just to show the mapping back to variables.... test just want the coefficients from
+    # res_values
     # res = [
     #     {'red': 0.25, 'blue': 0.25, 'yellow': 0.25, 'green': 0.25},
     #     {'red': -0.25, 'blue': -0.25, 'yellow': -0.25, 'green': -0.25},

tests/utilities/capture_set_sizes_for_cache.py

@@ -15,9 +15,10 @@
 logger = logging.getLogger(__name__)
 
 print(
-    'WARNING:  Continuing to execute this file will update the cached values for the set sizes for US_9R model in '
-    'the testing_data folder from the sqlite databases in the same folder.  This should only need to be done if the '
-    'schema or model have changed and that database has been updated.'
+    'WARNING:  Continuing to execute this file will update the cached values for the set sizes '
+    'for US_9R model in the testing_data folder from the sqlite databases in the same folder.  '
+    'This should only need to be done if the schema or model have changed and that database has '
+    'been updated.'
 )
 
 t = input('Type "Y" to continue, any other key to exit now.')