Skip to content

Commit

Permalink
Apply missing black code formatting
Browse files Browse the repository at this point in the history
was not done by ruff
  • Loading branch information
awais307 committed Nov 16, 2023
1 parent 9f23ee8 commit a1877f4
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions message_ix_models/tests/model/water/test_water_supply.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ def test_map_basin_region_wat():
with patch(
"message_ix_models.util.private_data_path", return_value="path/to/file"
), patch("pandas.read_csv", return_value=df_sw):
context['time'] = 'year'
context["time"] = "year"
result = map_basin_region_wat(context)

# Assert the results
Expand Down Expand Up @@ -98,7 +98,7 @@ def test_add_water_supply():
"message_ix_models.util.private_data_path", return_value="path/to/file"
), patch("pandas.read_csv", return_value=df_node), patch(
"message_ix_models.model.water.water_supply.map_basin_region_wat",
return_value=df_sw # Adjust this import
return_value=df_sw, # Adjust this import
):
# Call the function to be tested
result = add_water_supply(context)
Expand Down Expand Up @@ -137,7 +137,7 @@ def test_add_e_flow():
# Mock the function 'read_water_availability' to return the mocked DataFrame
with patch(
"message_ix_models.model.water.demands.read_water_availability",
return_value=(df_sw, df_sw)
return_value=(df_sw, df_sw),
), patch(
"message_ix_models.util.private_data_path", return_value="path/to/file"
), patch(
Expand Down

0 comments on commit a1877f4

Please sign in to comment.