python cleanups

Author: tgastine

python/magic/radial.py

@@ -234,7 +234,7 @@ def plot(self):
             if hasattr(self, 'con_radratio'):
                 if self.nVarCond == 2:
                     ax.axvline(self.con_radratio*self.radius[0], color='k',
-                              linestyle='--')
+                               linestyle='--')
             fig.tight_layout()
 
             fig = plt.figure()
@@ -298,7 +298,7 @@ def plot(self):
             ax.semilogy(x_axis, self.dynVisc, label='dyn. visc')
             ax.semilogy(x_axis, self.kinVisc, label='kin. visc')
             ax.semilogy(x_axis, self.prandtl, label='Prandtl')
-            if self.mode not in (1,5):
+            if self.mode not in (1, 5):
                 ax.semilogy(x_axis, self.prandtlmag, label='Pm')
             ax.semilogy(x_axis, self.ekman, label='Ekman')
             ax.set_ylabel('Thermal properties')
@@ -562,7 +562,7 @@ def plot(self):
         if hasattr(self, 'con_radratio'):
             if self.nVarCond == 2:
                 ax.axvline(self.con_radratio*self.radius[0], color='k',
-                          linestyle='--')
+                           linestyle='--')
 
 
 class RadLookUpTable:

python/magic/series.py

@@ -578,7 +578,7 @@ def plot(self):
             fig = plt.figure()
             ax = fig.add_subplot(111)
             if self.buoPower.max() != 0.:
-               ax.semilogy(self.time, self.buoPower, label='Thermal buoyancy')
+                ax.semilogy(self.time, self.buoPower, label='Thermal buoyancy')
             if self.buoPower_chem.max() != 0.:
                 ax.semilogy(self.time, self.buoPower_chem,
                             label='Chemical buoyancy')
@@ -824,9 +824,9 @@ def __init__(self, data, field):
             self.corr_vortz_otc = data[:, 6]
             self.corr_hel_otc = data[:, 7]
             if data.shape[-1] == 8:
-                self.geosA= np.zeros_like(self.time)
-                self.geosZ= np.zeros_like(self.time)
-                self.geosM= np.zeros_like(self.time)
+                self.geosA = np.zeros_like(self.time)
+                self.geosZ = np.zeros_like(self.time)
+                self.geosM = np.zeros_like(self.time)
                 self.geosNAP = np.zeros_like(self.time)
             elif data.shape[-1] == 12:
                 self.geosA = data[:, 8]