@@ -40,7 +40,7 @@ def test_map_basin_region_wat():
40
40
with patch (
41
41
"message_ix_models.util.private_data_path" , return_value = "path/to/file"
42
42
), patch ("pandas.read_csv" , return_value = df_sw ):
43
- context [' time' ] = ' year'
43
+ context [" time" ] = " year"
44
44
result = map_basin_region_wat (context )
45
45
46
46
# Assert the results
@@ -98,7 +98,7 @@ def test_add_water_supply():
98
98
"message_ix_models.util.private_data_path" , return_value = "path/to/file"
99
99
), patch ("pandas.read_csv" , return_value = df_node ), patch (
100
100
"message_ix_models.model.water.water_supply.map_basin_region_wat" ,
101
- return_value = df_sw # Adjust this import
101
+ return_value = df_sw , # Adjust this import
102
102
):
103
103
# Call the function to be tested
104
104
result = add_water_supply (context )
@@ -137,7 +137,7 @@ def test_add_e_flow():
137
137
# Mock the function 'read_water_availability' to return the mocked DataFrame
138
138
with patch (
139
139
"message_ix_models.model.water.demands.read_water_availability" ,
140
- return_value = (df_sw , df_sw )
140
+ return_value = (df_sw , df_sw ),
141
141
), patch (
142
142
"message_ix_models.util.private_data_path" , return_value = "path/to/file"
143
143
), patch (
0 commit comments