Lint modifications from issue 130

spyro/examples/elastic_local_abc.py

@@ -1,4 +1,3 @@
-import firedrake as fire
 import numpy as np
 import spyro
 
@@ -21,6 +20,7 @@
 final_time = 2.0  # [s]
 out_freq = int(0.01/time_step)
 
+
 def build_solver(local_abc, dt_scheme):
     d = {}
 
@@ -91,4 +91,4 @@ def build_solver(local_abc, dt_scheme):
     wave = spyro.IsotropicWave(d)
     wave.set_mesh(mesh_parameters={'dx': h})
 
-    return wave
+    return wave

spyro/io/field_logger.py

@@ -107,4 +107,4 @@ def log(self, t):
                 functional.sample()
 
     def get(self, key):
-        return self.__enabled_functionals[key].list[-1]
+        return self.__enabled_functionals[key].list[-1]

spyro/solvers/elastic_wave/functionals.py

@@ -18,4 +18,4 @@ def mechanical_energy_form(wave):
     eps = lambda v: 0.5*(grad(v) + grad(v).T)
     U = (lmbda*div(u_n)*div(u_n) + 2*mu*inner(eps(u_n), eps(u_n)))*dx
 
-    return K + U
+    return K + U

spyro/solvers/elastic_wave/local_abc.py

@@ -86,6 +86,7 @@ def local_abc_form(wave):
                     uz_dx, ux_dx, uy_dx,
                     uz_dy, ux_dy, uy_dy)
 
+
 def clayton_engquist_A1_terms(ndim, rho, c_p, c_s,
                               v, iz, ix, iy, qr_s,
                               uz_dt, ux_dt, uy_dt,
@@ -150,6 +151,7 @@ def clayton_engquist_A1_terms(ndim, rho, c_p, c_s,
 
     return F_t
 
+
 def stacey_terms(ndim, rho, c_p, c_s,
                  v, iz, ix, iy, qr_s,
                  uz_dt, ux_dt, uy_dt,

test/test_elastic_local_abc.py

@@ -22,12 +22,12 @@ def test_clayton_engquist_abc():
 
 def test_with_central():
     wave = build_solver("Stacey", "central")
-    with pytest.raises(AssertionError) as e:
+    with pytest.raises(AssertionError):
         wave.forward_solve()
 
 
 def test_with_backward_2nd():
     wave = build_solver("Stacey", "backward_2nd")
     wave.forward_solve()
     last_mechanical_energy = wave.field_logger.get("mechanical_energy")
-    assert last_mechanical_energy < expected_mechanical_energy
+    assert last_mechanical_energy < expected_mechanical_energy