@@ -204,12 +204,12 @@ def test_compound_without_units(model, fitter):
204204 res_fit = fitter (model , x , y * u .Hz )
205205 for param_name in res_fit .param_names :
206206 print (getattr (res_fit , param_name ))
207- assert all ([ res_fit [i ]._has_units for i in range (3 )] )
207+ assert all (res_fit [i ]._has_units for i in range (3 ))
208208 z = res_fit (x )
209209 assert isinstance (z , u .Quantity )
210210
211211 res_fit = fitter (model , np .arange (10 ) * u .Unit ("Angstrom" ), y )
212- assert all ([ res_fit [i ]._has_units for i in range (3 )] )
212+ assert all (res_fit [i ]._has_units for i in range (3 ))
213213 z = res_fit (x )
214214 assert isinstance (z , np .ndarray )
215215
@@ -234,12 +234,12 @@ def test_compound_fitting_with_units(fitter):
234234 z = model (x , y )
235235 res = fitter (model , x , y , z )
236236 assert isinstance (res (x , y ), np .ndarray )
237- assert all ([ res [i ]._has_units for i in range (2 )] )
237+ assert all (res [i ]._has_units for i in range (2 ))
238238
239239 model = models .Gaussian2D () + models .Planar2D ()
240240 res = fitter (model , x , y , z )
241241 assert isinstance (res (x , y ), np .ndarray )
242- assert all ([ res [i ]._has_units for i in range (2 )] )
242+ assert all (res [i ]._has_units for i in range (2 ))
243243
244244 # A case of a mixture of models with and without units
245245 model = models .BlackBody (temperature = 3000 * u .K ) * models .Const1D (amplitude = 1.0 )
0 commit comments