Sort imports for safe MPI initialization

python/.isort.cfg

@@ -0,0 +1,8 @@
+[settings]
+src_paths = benchmarks,demos,dolfinx_mpc
+known_first_party = dolfinx_mpc
+known_third_party = basix,dolfinx,ffcx,ufl,gmsh,numba,numpy,pytest,pyvista
+known_mpi = mpi4py,petsc4py
+sections=FUTURE,STDLIB,MPI,THIRDPARTY,FIRSTPARTY,LOCALFOLDER
+
+sections=FUTURE,STDLIB,MPI,THIRDPARTY,FIRSTPARTY,LOCALFOLDER

python/benchmarks/bench_contact_3D.py

@@ -8,12 +8,16 @@
 # between two cubes.
 from __future__ import annotations
 
+import warnings
 from argparse import ArgumentDefaultsHelpFormatter, ArgumentParser
 from pathlib import Path
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import basix.ufl
 import numpy as np
-from dolfinx import default_scalar_type, default_real_type
+from dolfinx import default_real_type, default_scalar_type
 from dolfinx.common import Timer, TimingType, list_timings, timing
 from dolfinx.cpp.mesh import entities_to_geometry
 from dolfinx.fem import (Constant, Function, dirichletbc, form, functionspace,
@@ -22,16 +26,14 @@
 from dolfinx.mesh import (CellType, compute_midpoints, create_mesh,
                           create_unit_cube, locate_entities_boundary, meshtags,
                           refine)
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import (Cell, Identity, Mesh, TestFunction, TrialFunction,
                  VectorElement, dx, grad, inner, sym, tr)
 
 from dolfinx_mpc import (MultiPointConstraint, apply_lifting, assemble_matrix,
                          assemble_vector)
 from dolfinx_mpc.utils import (create_normal_approximation, log_info,
                                rigid_motions_nullspace, rotation_matrix)
-import warnings
+
 comm = MPI.COMM_WORLD
 
 if default_real_type == np.float32:

python/benchmarks/bench_elasticity.py

@@ -10,17 +10,18 @@
 from pathlib import Path
 from typing import Optional
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import basix.ufl
 import h5py
 import numpy as np
 from dolfinx.common import Timer, TimingType, list_timings
-from dolfinx.fem import (Constant, Function, functionspace, dirichletbc, form,
+from dolfinx.fem import (Constant, Function, dirichletbc, form, functionspace,
                          locate_dofs_topological, set_bc)
 from dolfinx.io import XDMFFile
 from dolfinx.mesh import (create_unit_cube, locate_entities_boundary, meshtags,
                           refine)
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import (Identity, TestFunction, TrialFunction, ds, dx, grad, inner,
                  sym, tr)
 

python/benchmarks/bench_elasticity_edge.py

@@ -9,18 +9,19 @@
 from argparse import ArgumentDefaultsHelpFormatter, ArgumentParser
 from pathlib import Path
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import basix.ufl
 import h5py
 import numpy as np
 from dolfinx import default_scalar_type
 from dolfinx.common import Timer, TimingType, list_timings
-from dolfinx.fem import (Constant, Function, functionspace, dirichletbc, form,
+from dolfinx.fem import (Constant, Function, dirichletbc, form, functionspace,
                          locate_dofs_topological, set_bc)
 from dolfinx.io import XDMFFile
 from dolfinx.mesh import (CellType, create_unit_cube, locate_entities_boundary,
                           meshtags)
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import (Identity, SpatialCoordinate, TestFunction, TrialFunction,
                  as_vector, ds, dx, grad, inner, sym, tr)
 

python/benchmarks/bench_periodic.py

@@ -15,6 +15,9 @@
 from argparse import ArgumentDefaultsHelpFormatter, ArgumentParser
 from pathlib import Path
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import h5py
 import numpy as np
 from dolfinx import default_scalar_type
@@ -24,8 +27,6 @@
 from dolfinx.io import XDMFFile
 from dolfinx.mesh import (CellType, create_unit_cube, locate_entities_boundary,
                           meshtags)
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import (SpatialCoordinate, TestFunction, TrialFunction, dx, exp, grad,
                  inner, pi, sin)
 

python/benchmarks/post_proc.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
-import numpy as np
 import matplotlib.pyplot as plt
+import numpy as np
 
 # Res 0.1 31776, 0.05 234546, 0.025 1801086, 0.02 3488856, 0.0175 5147961,0.015 7960200
 dofs = [31776, 234546, 1801086, 3488856, 5147961, 7960200]

python/benchmarks/ref_elasticity.py

@@ -11,6 +11,9 @@
 from time import perf_counter
 from typing import Optional
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import basix.ufl
 import h5py
 import numpy as np
@@ -24,8 +27,6 @@
 from dolfinx.log import LogLevel, log, set_log_level
 from dolfinx.mesh import (CellType, create_unit_cube, locate_entities_boundary,
                           meshtags, refine)
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import (Identity, SpatialCoordinate, TestFunction, TrialFunction,
                  as_vector, ds, dx, grad, inner, sym, tr)
 

python/benchmarks/ref_periodic.py

@@ -21,19 +21,20 @@
 from time import perf_counter
 from typing import Optional
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import h5py
 import numpy as np
 from dolfinx import default_scalar_type
 from dolfinx.common import Timer, TimingType, list_timings
-from dolfinx.fem import (Function, functionspace, dirichletbc, form,
+from dolfinx.fem import (Function, dirichletbc, form, functionspace,
                          locate_dofs_geometrical)
 from dolfinx.fem.petsc import (apply_lifting, assemble_matrix, assemble_vector,
                                set_bc)
 from dolfinx.io import XDMFFile
 from dolfinx.log import LogLevel, log, set_log_level
 from dolfinx.mesh import CellType, create_unit_cube, refine
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import (SpatialCoordinate, TestFunction, TrialFunction, dx, exp, grad,
                  inner, pi, sin)
 

python/benchmarks/visualize_iterations.py

@@ -1,13 +1,15 @@
 from __future__ import annotations
 
-import matplotlib.pyplot as plt
-from matplotlib.ticker import MaxNLocator, LogLocator, NullFormatter
-import matplotlib.transforms as mtransforms
-import h5py
-import mpi4py
 import argparse
 import sys
+
+import mpi4py
+
+import h5py
+import matplotlib.pyplot as plt
+import matplotlib.transforms as mtransforms
 import numpy as np
+from matplotlib.ticker import LogLocator, MaxNLocator, NullFormatter
 
 
 def visualize_elasticity():

python/demos/create_and_export_mesh.py

@@ -3,6 +3,8 @@
 from pathlib import Path
 from typing import Dict, List, Sequence, Tuple, Union
 
+from mpi4py import MPI
+
 import dolfinx.common as _common
 import dolfinx.cpp as _cpp
 import dolfinx.io as _io
@@ -11,7 +13,6 @@
 import numpy as np
 import ufl
 from dolfinx.io import gmshio
-from mpi4py import MPI
 
 import dolfinx_mpc.utils as _utils
 

python/demos/demo_contact_2D.py

@@ -17,10 +17,12 @@
 from argparse import ArgumentDefaultsHelpFormatter, ArgumentParser
 from pathlib import Path
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import numpy as np
 import scipy.sparse.linalg
-from create_and_export_mesh import gmsh_2D_stacked, mesh_2D_dolfin
-from dolfinx import default_scalar_type, default_real_type
+from dolfinx import default_real_type, default_scalar_type
 from dolfinx.common import Timer, TimingType, list_timings
 from dolfinx.fem import (Constant, dirichletbc, form, functionspace,
                          locate_dofs_geometrical)
@@ -29,11 +31,10 @@
 from dolfinx.io import XDMFFile
 from dolfinx.log import LogLevel, set_log_level
 from dolfinx.mesh import locate_entities_boundary, meshtags
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import (Identity, Measure, TestFunction, TrialFunction, dx, grad,
                  inner, sym, tr)
 
+from create_and_export_mesh import gmsh_2D_stacked, mesh_2D_dolfin
 from dolfinx_mpc import LinearProblem, MultiPointConstraint
 from dolfinx_mpc.utils import (compare_mpc_lhs, compare_mpc_rhs,
                                create_normal_approximation,

python/demos/demo_contact_3D.py

@@ -12,18 +12,19 @@
 from argparse import ArgumentDefaultsHelpFormatter, ArgumentParser
 from pathlib import Path
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import dolfinx.fem as fem
 import numpy as np
 import scipy.sparse.linalg
-from create_and_export_mesh import gmsh_3D_stacked, mesh_3D_dolfin
 from dolfinx import default_real_type, default_scalar_type
 from dolfinx.common import Timer, TimingType, list_timings
 from dolfinx.io import XDMFFile
 from dolfinx.mesh import CellType
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import Identity, TestFunction, TrialFunction, dx, grad, inner, sym, tr
 
+from create_and_export_mesh import gmsh_3D_stacked, mesh_3D_dolfin
 from dolfinx_mpc import (MultiPointConstraint, apply_lifting, assemble_matrix,
                          assemble_vector)
 from dolfinx_mpc.utils import (compare_mpc_lhs, compare_mpc_rhs,

python/demos/demo_elasticity.py

@@ -7,15 +7,16 @@
 
 from pathlib import Path
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import dolfinx.fem as fem
 import numpy as np
 import scipy.sparse.linalg
 from dolfinx import default_scalar_type
 from dolfinx.common import Timer
 from dolfinx.io import XDMFFile
 from dolfinx.mesh import create_unit_square, locate_entities_boundary
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import (Identity, SpatialCoordinate, TestFunction, TrialFunction,
                  as_vector, dx, grad, inner, sym, tr)
 

python/demos/demo_elasticity_disconnect.py

@@ -9,14 +9,15 @@
 
 from pathlib import Path
 
+from mpi4py import MPI
+
 import basix.ufl
 import gmsh
 import numpy as np
 from dolfinx import default_scalar_type
 from dolfinx.fem import (Constant, Function, dirichletbc, functionspace,
                          locate_dofs_topological)
 from dolfinx.io import XDMFFile, gmshio
-from mpi4py import MPI
 from ufl import (Identity, Measure, SpatialCoordinate, TestFunction,
                  TrialFunction, as_vector, grad, inner, sym, tr)
 

python/demos/demo_elasticity_disconnect_2D.py

@@ -9,6 +9,8 @@
 
 from pathlib import Path
 
+from mpi4py import MPI
+
 import gmsh
 import numpy as np
 from dolfinx import default_scalar_type
@@ -17,7 +19,6 @@
                          locate_dofs_topological)
 from dolfinx.io import XDMFFile, gmshio
 from dolfinx.mesh import locate_entities_boundary
-from mpi4py import MPI
 from ufl import (Identity, Measure, SpatialCoordinate, TestFunction,
                  TrialFunction, grad, inner, sym, tr)
 

python/demos/demo_periodic3d_topological.py

@@ -19,6 +19,8 @@
 from pathlib import Path
 from typing import Dict, Union
 
+from mpi4py import MPI
+
 import dolfinx.fem as fem
 import numpy as np
 import scipy.sparse.linalg
@@ -27,7 +29,6 @@
 from dolfinx.io import VTXWriter
 from dolfinx.mesh import (CellType, create_unit_cube, locate_entities_boundary,
                           meshtags)
-from mpi4py import MPI
 from numpy.typing import NDArray
 from ufl import (SpatialCoordinate, TestFunction, TrialFunction, as_vector, dx,
                  exp, grad, inner, pi, sin)

python/demos/demo_periodic_geometrical.py

@@ -15,15 +15,16 @@
 from pathlib import Path
 from typing import Union
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import dolfinx.fem as fem
 import numpy as np
 import scipy.sparse.linalg
 from dolfinx import default_scalar_type
 from dolfinx.common import Timer, TimingType, list_timings
 from dolfinx.io import XDMFFile
 from dolfinx.mesh import create_unit_square, locate_entities_boundary
-from mpi4py import MPI
-from petsc4py import PETSc
 from ufl import (SpatialCoordinate, TestFunction, TrialFunction, as_vector, dx,
                  exp, grad, inner, pi, sin)
 

python/demos/demo_periodic_gep.py

@@ -31,13 +31,14 @@
 from pathlib import Path
 from typing import List, Tuple
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import dolfinx.fem as fem
 import numpy as np
 from dolfinx import default_scalar_type
 from dolfinx.io import XDMFFile
 from dolfinx.mesh import create_unit_square, locate_entities_boundary, meshtags
-from mpi4py import MPI
-from petsc4py import PETSc
 from slepc4py import SLEPc
 from ufl import TestFunction, TrialFunction, dx, grad, inner
 

python/demos/demo_stokes.py

@@ -14,21 +14,23 @@
 from pathlib import Path
 from typing import Union
 
+from mpi4py import MPI
+from petsc4py import PETSc
+
 import basix.ufl
-import dolfinx_mpc.utils
 import gmsh
 import numpy as np
 import scipy.sparse.linalg
 from dolfinx import common, default_scalar_type, fem, io
 from dolfinx.io import gmshio
-from mpi4py import MPI
 from numpy.typing import NDArray
-from petsc4py import PETSc
 from ufl import (FacetNormal, Identity, Measure, TestFunctions, TrialFunctions,
                  div, dot, dx, grad, inner, outer, sym)
 from ufl.core.expr import Expr
 
+import dolfinx_mpc.utils
 from dolfinx_mpc import LinearProblem, MultiPointConstraint
+
 # -
 # ## Mesh generation
 #