@@ -59,9 +59,11 @@ def test_remove_variable_and_keeps_others(self):
5959 variable2 = "variable2"
6060 variable3 = "variable3"
6161 dataset = _xr .Dataset (
62- data_vars = dict (
63- variable1 = variable1 , variable2 = variable2 , variable3 = variable3
64- )
62+ data_vars = {
63+ "variable1" : variable1 ,
64+ "variable2" : variable2 ,
65+ "variable3" : variable3 ,
66+ }
6567 )
6668 variable_to_be_removed = [variable2 ]
6769
@@ -83,12 +85,12 @@ def test_leave_only_one_variable(self):
8385 variable3 = "variable3"
8486 variable4 = "variable4"
8587 dataset = _xr .Dataset (
86- data_vars = dict (
87- variable1 = variable1 ,
88- variable2 = variable2 ,
89- variable3 = variable3 ,
90- variable4 = variable4 ,
91- )
88+ data_vars = {
89+ " variable1" : variable1 ,
90+ " variable2" : variable2 ,
91+ " variable3" : variable3 ,
92+ " variable4" : variable4 ,
93+ }
9294 )
9395 dataset ["variable2" ].attrs = {"cf_role" : "mesh_topology" }
9496 variable_to_keep = [variable2 ]
@@ -112,12 +114,12 @@ def test_reduce_for_writing_throws_exception_for_non_existing_variable(self):
112114 variable3 = "variable3"
113115 variable4 = "variable4"
114116 dataset = _xr .Dataset (
115- data_vars = dict (
116- variable1 = variable1 ,
117- variable2 = variable2 ,
118- variable3 = variable3 ,
119- variable4 = variable4 ,
120- )
117+ data_vars = {
118+ " variable1" : variable1 ,
119+ " variable2" : variable2 ,
120+ " variable3" : variable3 ,
121+ " variable4" : variable4 ,
122+ }
121123 )
122124 dataset ["variable2" ].attrs = {"cf_role" : "mesh_topology" }
123125 logger = Mock (ILogger )
@@ -143,12 +145,12 @@ def test_leave_multiple_variables(self):
143145 variable3 = "variable3"
144146 variable4 = "variable4"
145147 dataset = _xr .Dataset (
146- data_vars = dict (
147- variable1 = variable1 ,
148- variable2 = variable2 ,
149- variable3 = variable3 ,
150- variable4 = variable4 ,
151- )
148+ data_vars = {
149+ " variable1" : variable1 ,
150+ " variable2" : variable2 ,
151+ " variable3" : variable3 ,
152+ " variable4" : variable4 ,
153+ }
152154 )
153155 dataset ["variable2" ].attrs = {"cf_role" : "mesh_topology" }
154156 variables_to_keep = [variable2 , variable4 ]
@@ -192,7 +194,11 @@ def test_list_variables_in_dataset():
192194 variable2 = "variable2"
193195 variable3 = "variable3"
194196 dataset = _xr .Dataset (
195- data_vars = dict (variable1 = variable1 , variable2 = variable2 , variable3 = variable3 )
197+ data_vars = {
198+ "variable1" : variable1 ,
199+ "variable2" : variable2 ,
200+ "variable3" : variable3 ,
201+ }
196202 )
197203
198204 # Act
@@ -227,7 +233,11 @@ def test_rename_variable_returns_dataset_without_old_variable_and_with_new_varia
227233 variable3 = "variable3"
228234 new_name = "new_name"
229235 dataset1 = _xr .Dataset (
230- data_vars = dict (variable1 = variable1 , variable2 = variable2 , variable3 = variable3 )
236+ data_vars = {
237+ "variable1" : variable1 ,
238+ "variable2" : variable2 ,
239+ "variable3" : variable3 ,
240+ }
231241 )
232242 # Act
233243 dataset2 = utilities .rename_variable (dataset1 , "variable1" , new_name )
@@ -245,11 +255,15 @@ def test_merged_dataset_is_xarray_dataset_and_contains_all_variables(self):
245255 # Arrange
246256 variable1 = "variable1"
247257 variable2 = "variable2"
248- dataset1 = _xr .Dataset (data_vars = dict (variable1 = variable1 , variable2 = variable2 ))
258+ dataset1 = _xr .Dataset (
259+ data_vars = {"variable1" : variable1 , "variable2" : variable2 }
260+ )
249261
250262 variable3 = "variable3"
251263 variable4 = "variable4"
252- dataset2 = _xr .Dataset (data_vars = dict (variable3 = variable3 , variable4 = variable4 ))
264+ dataset2 = _xr .Dataset (
265+ data_vars = {"variable3" : variable3 , "variable4" : variable4 }
266+ )
253267
254268 # Act
255269 merged_dataset = utilities .merge_datasets (dataset1 , dataset2 )
@@ -267,15 +281,21 @@ def test_merged_list_of_datasets_is_xarray_dataset_and_contains_all_variables(se
267281 # Arrange
268282 variable1 = "variable1"
269283 variable2 = "variable2"
270- dataset1 = _xr .Dataset (data_vars = dict (variable1 = variable1 , variable2 = variable2 ))
284+ dataset1 = _xr .Dataset (
285+ data_vars = {"variable1" : variable1 , "variable2" : variable2 }
286+ )
271287
272288 variable3 = "variable3"
273289 variable4 = "variable4"
274- dataset2 = _xr .Dataset (data_vars = dict (variable3 = variable3 , variable4 = variable4 ))
290+ dataset2 = _xr .Dataset (
291+ data_vars = {"variable3" : variable3 , "variable4" : variable4 }
292+ )
275293
276294 variable5 = "variable5"
277295 variable6 = "variable6"
278- dataset3 = _xr .Dataset (data_vars = dict (variable5 = variable5 , variable6 = variable6 ))
296+ dataset3 = _xr .Dataset (
297+ data_vars = {"variable5" : variable5 , "variable6" : variable6 }
298+ )
279299
280300 list_datasets = [dataset1 , dataset2 , dataset3 ]
281301 # Act
@@ -297,7 +317,7 @@ def test_get_dummy_variable(self):
297317 # Arrange
298318 variable1 = "variable1"
299319 variable2 = "variable2"
300- ds = _xr .Dataset (data_vars = dict ( variable1 = variable1 , variable2 = variable2 ) )
320+ ds = _xr .Dataset (data_vars = { " variable1" : variable1 , " variable2" : variable2 } )
301321 ds ["variable1" ].attrs = {"cf_role" : "mesh_topology" }
302322
303323 # Act
@@ -311,7 +331,7 @@ def test_get_dummy_variable_fails(self):
311331 # Arrange
312332 variable1 = "variable1"
313333 variable2 = "variable2"
314- ds = _xr .Dataset (data_vars = dict ( variable1 = variable1 , variable2 = variable2 ) )
334+ ds = _xr .Dataset (data_vars = { " variable1" : variable1 , " variable2" : variable2 } )
315335
316336 # Act
317337 with pytest .raises (ValueError ) as error :
@@ -329,7 +349,7 @@ def test_get_dummy_and_dependent_var_list(self):
329349 in a ugrid dataset"""
330350 # Arrange
331351 var_list = ["mesh2d" , "var2" , "var3" , "var4" , "var5" ]
332- ds = _xr .Dataset (data_vars = dict . fromkeys ( var_list ) )
352+ ds = _xr .Dataset (data_vars = { k : None for k in var_list } )
333353 ds ["mesh2d" ].attrs = {
334354 "cf_role" : "mesh_topology" ,
335355 "test_coordinates" : "var2 var3" ,
@@ -351,7 +371,7 @@ def test_get_dummy_variable(self):
351371 """Test if you receive the name of the dummy variable in a ugrid dataset"""
352372 # Arrange
353373 var_list = ["var1" , "var2" , "var3" , "var4" , "var5" ]
354- ds = _xr .Dataset (data_vars = dict . fromkeys ( var_list ) )
374+ ds = _xr .Dataset (data_vars = { k : None for k in var_list } )
355375 ds ["var1" ].attrs = {
356376 "cf_role" : "mesh_topology" ,
357377 "test_coordinates" : "var2 var3" ,
@@ -369,7 +389,7 @@ def test_get_dummy_variable_if_none(self):
369389 """Test if you receive nothing if there is no dependent variables in a ugrid dataset"""
370390 # Arrange
371391 var_list = ["var1" , "var2" , "var3" , "var4" , "var5" ]
372- ds = _xr .Dataset (data_vars = dict . fromkeys ( var_list ) )
392+ ds = _xr .Dataset (data_vars = { k : None for k in var_list } )
373393
374394 # Act
375395 dummy_variable = utilities .get_dependent_vars_by_var_name (ds , "var1" )
0 commit comments