some lint fixes on compressible/ (#125)

Author: zingale

pyro/compressible/BC.py

@@ -14,7 +14,7 @@
 
 import numpy as np
 
-import pyro.compressible.eos as eos
+from pyro.compressible import eos
 from pyro.util import msg
 
 
@@ -211,7 +211,7 @@ def user(bc_name, bc_edge, variable, ccdata):
                 v[:, :] = 0.0   # no source term
 
     else:
-        msg.fail("error: bc type %s not supported" % (bc_name))
+        msg.fail(f"error: bc type {bc_name} not supported")
 
 
 def inflow_post_bc(var, g):

pyro/compressible/derives.py

@@ -1,6 +1,6 @@
 import numpy as np
 
-import pyro.compressible.eos as eos
+from pyro.compressible import eos
 
 
 def derive_primitives(myd, varnames):
@@ -52,5 +52,5 @@ def derive_primitives(myd, varnames):
 
     if len(derived_vars) > 1:
         return derived_vars
-    else:
-        return derived_vars[0]
+
+    return derived_vars[0]

pyro/compressible/eos.py

@@ -4,7 +4,7 @@
 """
 
 
-def pres(gamma, dens, eint):
+def pres(gamma, rho, eint):
     """
     Given the density and the specific internal energy, return the
     pressure
@@ -13,7 +13,7 @@ def pres(gamma, dens, eint):
     ----------
     gamma : float
         The ratio of specific heats
-    dens : float
+    rho : float
         The density
     eint : float
         The specific internal energy
@@ -24,11 +24,11 @@ def pres(gamma, dens, eint):
        The pressure
 
      """
-    p = dens*eint*(gamma - 1.0)
+    p = rho * eint * (gamma - 1.0)
     return p
 
 
-def dens(gamma, pres, eint):
+def dens(gamma, p, eint):
     """
     Given the pressure and the specific internal energy, return
     the density
@@ -37,7 +37,7 @@ def dens(gamma, pres, eint):
     ----------
     gamma : float
         The ratio of specific heats
-    pres : float
+    p : float
         The pressure
     eint : float
         The specific internal energy
@@ -48,19 +48,19 @@ def dens(gamma, pres, eint):
        The density
 
     """
-    dens = pres/(eint*(gamma - 1.0))
-    return dens
+    rho = p / (eint * (gamma - 1.0))
+    return rho
 
 
-def rhoe(gamma, pres):
+def rhoe(gamma, p):
     """
     Given the pressure, return (rho * e)
 
     Parameters
     ----------
     gamma : float
         The ratio of specific heats
-    pres : float
+    p : float
         The pressure
 
     Returns
@@ -69,5 +69,4 @@ def rhoe(gamma, pres):
        The internal energy density, rho e
 
     """
-    rhoe = pres/(gamma - 1.0)
-    return rhoe
+    return p / (gamma - 1.0)