fix flake 8 error

Author: SUTIENG Tam

skypy/halos/_colossus.py

@@ -204,6 +204,7 @@ def colossus_mf(redshift, model, mdef, m_min, m_max, sky_area, cosmology,
 
     return z, m
 
+
 def concentration(mass, mdef, redshift, model, cosmology, sigma8, ns):
     r'''Halo concentration calculator.
 
@@ -228,7 +229,7 @@ def concentration(mass, mdef, redshift, model, cosmology, sigma8, ns):
         scale of 8 Mpc/h.
     ns : float
         Cosmology parameter, spectral index of scalar perturbation power spectrum.
-    
+
     Returns
     -------
     concentration : float or array_like
@@ -237,13 +238,14 @@ def concentration(mass, mdef, redshift, model, cosmology, sigma8, ns):
     '''
     from colossus.cosmology.cosmology import fromAstropy
     from colossus.halo import concentration
-    
+
     fromAstropy(cosmology, sigma8=sigma8, ns=ns)
-    
+
     c = concentration.concentration(mass, mdef, redshift, model)
-    
+
     return c
 
+
 def radius(mass, concentration, redshift, mdef, Delta, cosmology, sigma8, ns):
     r'''Calculate the scale radius and the spherical overdensity radius of halo by assuming
     the NFW model.
@@ -271,7 +273,7 @@ def radius(mass, concentration, redshift, mdef, Delta, cosmology, sigma8, ns):
         scale of 8 Mpc/h.
     ns : float
         Cosmology parameter, spectral index of scalar perturbation power spectrum.
-    
+
     Returns
     -------
     rs : float
@@ -282,16 +284,16 @@ def radius(mass, concentration, redshift, mdef, Delta, cosmology, sigma8, ns):
     '''
     from colossus.cosmology.cosmology import fromAstropy
     from colossus.halo import profile_nfw
-        
+
     fromAstropy(cosmology, sigma8=sigma8, ns=ns)
 
-    prof = profile_nfw.NFWProfile(M = mass, c = concentration, z = redshift, mdef = mdef)
+    prof = profile_nfw.NFWProfile(M=mass, c=concentration, z=redshift, mdef=mdef)
     rs = prof.par['rs']
-    RDelta =  prof.RDelta(redshift, Delta)
+    RDelta = prof.RDelta(redshift, Delta)
 
     return rs, RDelta
-    
-    
+
+
 def Delta_Sigma(mass, concentration, redshift, mdef, radius, cosmology, sigma8, ns):
     r'''The excess surface density at given radius by assuming the NFW model.
 
@@ -318,20 +320,20 @@ def Delta_Sigma(mass, concentration, redshift, mdef, radius, cosmology, sigma8,
         scale of 8 Mpc/h.
     ns : float
         Cosmology parameter, spectral index of scalar perturbation power spectrum.
-    
+
     Returns
     -------
     DeltaSigma: float or array_like
         The excess surface density at the given radius, in units of h physical Msun/kpc^2;
         has the same dimensions as radius.
-    
+
     '''
     from colossus.cosmology.cosmology import fromAstropy
     from colossus.halo import profile_nfw
-        
+
     fromAstropy(cosmology, sigma8=sigma8, ns=ns)
 
-    prof = profile_nfw.NFWProfile(M = mass, c = concentration, z = redshift, mdef = mdef)
+    prof = profile_nfw.NFWProfile(M=mass, c=concentration, z=redshift, mdef=mdef)
     deltaSigma = prof.deltaSigma(radius)
 
     return deltaSigma

skypy/halos/tests/test_colossus.py

@@ -106,28 +106,28 @@ def test_colossus_mf():
     assert np.all(m_halo >= m_min)
     assert np.all(m_halo <= m_max)
 
+
 @pytest.mark.skipif(not HAS_COLOSSUS, reason='test requires colossus')
 @pytest.mark.flaky
 def test_colossus_concentration():
     from skypy.halos._colossus import concentration
     from astropy.cosmology import Planck15
-    from colossus.cosmology.cosmology import fromAstropy
-    
-    mass=np.logspace(np.log10(1e+14), np.log10(1e+15), 10)
+
+    mass = np.logspace(np.log10(1e+14), np.log10(1e+15), 10)
     model, mdef = 'diemer19', '200c'
     redshift = 0.5
     cosmo = Planck15
     sigma8, ns = 0.82, 0.96
     c = concentration(mass, mdef, redshift, model, cosmo, sigma8, ns)
-    
+
     assert len(c) == len(mass)
-    
+
+
 @pytest.mark.skipif(not HAS_COLOSSUS, reason='test requires colossus')
 @pytest.mark.flaky
 def test_colossus_radius():
     from skypy.halos._colossus import radius
     from astropy.cosmology import Planck15
-    from colossus.cosmology.cosmology import fromAstropy
 
     mass, c, redshift = 1e+15, 3, 0.5
     Delta, mdef = '500c', '200c'
@@ -138,20 +138,20 @@ def test_colossus_radius():
 
     assert rs > 0
     assert RDelta > 0
-    
+
+
 @pytest.mark.skipif(not HAS_COLOSSUS, reason='test requires colossus')
 @pytest.mark.flaky
 def test_colossus_Delta_Sigma():
     from skypy.halos._colossus import Delta_Sigma
     from astropy.cosmology import Planck15
-    from colossus.cosmology.cosmology import fromAstropy
 
     mass, c, redshift = 1e+15, 3, 0.5
     mdef = '200c'
     cosmo = Planck15
     sigma8, ns = 0.82, 0.96
     radius = np.logspace(np.log10(300), np.log10(3000), 10)
-    
-    DeltaSigma=Delta_Sigma(mass, c, redshift, mdef, radius, cosmo, sigma8, ns)
+
+    DeltaSigma = Delta_Sigma(mass, c, redshift, mdef, radius, cosmo, sigma8, ns)
 
     assert len(DeltaSigma) == len(radius)