@@ -165,22 +165,22 @@ def check_grb_disv1d(fpth):
165165 assert grb .xorigin == xorigin , "xorigin in grb file is not correct"
166166 assert grb .yorigin == yorigin , "yorigin in grb file is not correct"
167167 assert grb .angrot == angrot , "angrot in grb file is not correct"
168- assert np .allclose (
169- grb . bot . reshape (( nodes ,)), np . zeros (( nodes ,))
170- ), "grb botm not correct"
168+ assert np .allclose (grb . bot . reshape (( nodes ,)), np . zeros (( nodes ,))), (
169+ " grb botm not correct"
170+ )
171171 cellx = np .linspace (dx / 2 , nreach * dx - dx / 2 , nreach )
172172 celly = np .zeros (nreach )
173173 assert np .allclose (grb ._datadict ["CELLX" ], cellx .flatten ()), "cellx is not right"
174174 assert np .allclose (grb ._datadict ["CELLY" ], celly .flatten ()), "celly is not right"
175175 assert grb ._datadict ["IAVERT" ].shape [0 ] == nodes + 1 , "iavert size not right"
176- assert (
177- grb . _datadict [ "IAVERT" ][ - 1 ] - 1 == grb . _datadict [ "JAVERT" ]. shape [ 0 ]
178- ), "javert size not right"
176+ assert grb . _datadict [ "IAVERT" ][ - 1 ] - 1 == grb . _datadict [ "JAVERT" ]. shape [ 0 ], (
177+ "javert size not right"
178+ )
179179 assert grb .ia .shape [0 ] == grb .ncells + 1 , "ia in grb file is not correct size"
180180 assert grb .ja .shape [0 ] == grb .nja , "ja in grb file is not correct size"
181- assert np .allclose (
182- grb . idomain . reshape (( nodes ,)), idomain . reshape (( nodes ,))
183- ), "grb idomain not correct"
181+ assert np .allclose (grb . idomain . reshape (( nodes ,)), idomain . reshape (( nodes ,))), (
182+ " grb idomain not correct"
183+ )
184184
185185
186186def check_output (idx , test ):
@@ -198,9 +198,9 @@ def check_output(idx, test):
198198 fpth = test .workspace / f"{ modelname } .stage"
199199 sobj = flopy .utils .HeadFile (fpth , precision = "double" , text = "STAGE" )
200200 stage = sobj .get_data ().reshape ((nodes ,))
201- assert np .allclose (
202- stage [ idomain == 0 ], 3.0e30
203- ), "stage should have nodata values where idomain is zero"
201+ assert np .allclose (stage [ idomain == 0 ], 3.0e30 ), (
202+ " stage should have nodata values where idomain is zero"
203+ )
204204 assert stage [idomain == 1 ].max () == 1.0 , "maximum stage should be 1.0"
205205 assert stage [idomain == 1 ].min () == 0.5 , "minimum stage should be 0.5"
206206
0 commit comments