#71: Sort imports

demos/lineal_spring.py

@@ -44,16 +44,15 @@
 #
 # We begin by importing from the namespaces of Firedrake and Movement. ::
 
-from firedrake import *
-from movement import *
-
 # Let's start with a uniform mesh of the unit square. It has four boundary segments,
 # which are tagged with the integers 1, 2, 3, and 4. Note that segment 4 corresponds to
 # the top boundary. ::
-
 import matplotlib.pyplot as plt
+from firedrake import *
 from firedrake.pyplot import triplot
 
+from movement import *
+
 n = 10
 mesh = UnitSquareMesh(n, n)
 fig, axes = plt.subplots()

demos/monge_ampere1.py

@@ -40,6 +40,7 @@
 # We begin the example by importing from the namespaces of Firedrake and Movement.
 
 from firedrake import *
+
 from movement import *
 
 # To start with a simple example, consider a uniform mesh of the unit square.

demos/monge_ampere_3d.py

@@ -5,9 +5,10 @@
 # can also be applied to 3D meshes. We employ the `sinatan3` function
 # from :cite:`park2019` to introduce an interesting pattern.
 
-import movement
 from firedrake import *
 
+import movement
+
 
 def sinatan3(mesh):
     x, y, z = SpatialCoordinate(mesh)

demos/monge_ampere_helmholtz.py

@@ -42,6 +42,7 @@
 # we instead apply Dirichlet boundary conditions based on the chosen analytical solution.
 
 from firedrake import *
+
 from movement import MongeAmpereMover
 
 mesh = UnitSquareMesh(20, 20)  # initial mesh

movement/laplacian.py

@@ -1,10 +1,10 @@
 import firedrake
-from firedrake.petsc import PETSc
 import ufl
+from firedrake.petsc import PETSc
+
 import movement.solver_parameters as solver_parameters
 from movement.mover import PrimeMover
 
-
 __all__ = ["LaplacianSmoother"]
 
 

movement/monge_ampere.py

@@ -1,12 +1,12 @@
 import firedrake
+import numpy as np
+import ufl
 from firedrake.petsc import PETSc
 from pyadjoint import no_annotations
-import ufl
-import numpy as np
+
 import movement.solver_parameters as solver_parameters
 from movement.mover import PrimeMover
 
-
 __all__ = [
     "MongeAmpereMover_Relaxation",
     "MongeAmpereMover_QuasiNewton",
@@ -194,7 +194,7 @@ def l2_projector(self):
             nzero = sum(iszero)
             if nzero == self.dim:
                 raise ValueError(f"Invalid normal vector {_n}")
-            elif nzero == self.dim-1:
+            elif nzero == self.dim - 1:
                 idx = iszero.index(False)
                 bcs.append(firedrake.DirichletBC(self.P1_vec.sub(idx), 0, i))
                 continue
@@ -206,7 +206,10 @@ def l2_projector(self):
 
             # Allow tangential movement, but only up until the end of boundary segments
             a_bc = ufl.dot(tangential(v_cts, n), tangential(u_cts, n)) * self.ds
-            L_bc = ufl.dot(tangential(v_cts, n), tangential(ufl.grad(self.phi_old), n)) * self.ds
+            L_bc = (
+                ufl.dot(tangential(v_cts, n), tangential(ufl.grad(self.phi_old), n))
+                * self.ds
+            )
             edges = set(self.mesh.exterior_facets.unique_markers)
             if len(edges) == 0:
                 bbc = None  # Periodic case
@@ -332,8 +335,7 @@ def equidistributor(self):
         a = ufl.inner(tau, sigma) * self.dx
         L = (
             -ufl.dot(ufl.div(tau), ufl.grad(self.phi)) * self.dx
-            + ufl.dot(ufl.dot(tangential(ufl.grad(self.phi), n), tau), n)
-            * self.ds
+            + ufl.dot(ufl.dot(tangential(ufl.grad(self.phi), n), tau), n) * self.ds
         )
         problem = firedrake.LinearVariationalProblem(a, L, self.sigma)
         sp = {

movement/mover.py

@@ -1,7 +1,6 @@
 import firedrake
-from firedrake.cython.dmcommon import create_section
 import numpy as np
-
+from firedrake.cython.dmcommon import create_section
 
 __all__ = ["PrimeMover"]
 

movement/spring.py

@@ -1,11 +1,11 @@
 import firedrake
-from firedrake.petsc import PETSc
-import ufl
 import numpy as np
+import ufl
+from firedrake.petsc import PETSc
+
 import movement.solver_parameters as solver_parameters
 from movement.mover import PrimeMover
 
-
 __all__ = ["SpringMover_Lineal", "SpringMover_Torsional", "SpringMover"]
 
 

movement/tangling.py

@@ -1,7 +1,7 @@
-import firedrake
-import ufl
 import warnings
 
+import firedrake
+import ufl
 
 __all__ = ["MeshTanglingChecker"]
 

test/test_demos.py

@@ -1,14 +1,15 @@
 """
 Checks that all demo scripts run.
 """
+
 import glob
 import os
-from os.path import splitext
-import pytest
 import shutil
 import subprocess
 import sys
+from os.path import splitext
 
+import pytest
 
 # Set environment variable to specify that tests are being run so that we can
 # cut down the length of the demos

test/test_monge_ampere.py

@@ -1,7 +1,8 @@
-from movement import *
-from monitors import *
-import pytest
 import numpy as np
+import pytest
+from monitors import *
+
+from movement import *
 
 
 @pytest.fixture(params=["relaxation", "quasi_newton"])

test/test_smoothing.py

@@ -1,8 +1,10 @@
-from firedrake import *
-from movement import *
-import numpy as np
 import os
+
+import numpy as np
 import pytest
+from firedrake import *
+
+from movement import *
 
 
 @pytest.fixture(params=["laplacian"])

test/test_spring.py

@@ -1,9 +1,11 @@
+import os
+
 import firedrake
-from movement import SpringMover
 import numpy as np
-import os
 import pytest
 
+from movement import SpringMover
+
 
 @pytest.fixture(params=["lineal"])
 def method(request):

test/test_tangling.py

@@ -1,6 +1,8 @@
+import unittest
+
 from firedrake import *
+
 from movement import *
-import unittest
 
 
 class TestTangling(unittest.TestCase):