Black :-)

bench/profiling/simul_profile2d_vmprof.py

@@ -5,7 +5,6 @@
 
 """
 
-
 from simul_profile2d import sim
 
 sim.time_stepping.start()

bench/spectraldns/bench_2d.py

@@ -19,6 +19,7 @@
   mpirun -np 2 fluidsim-bench 1152 -d 2 -s ns2d -it 10
 
 """
+
 import time
 import numpy as np
 from spectralDNS import config, get_solver, solve
@@ -68,7 +69,7 @@ def update(context):
             "T": 1e-11,  # Should run 10 iterations
             "write_result": 100,
             "L": [L, L],
-            "M": [10, 10]  # Mesh size is pow(2, M[i]) in direction i
+            "M": [10, 10],  # Mesh size is pow(2, M[i]) in direction i
             # 2**9 == 512
         },
         "doublyperiodic",

bench/spectraldns/bench_3d.py

@@ -16,6 +16,7 @@
   mpirun -np 2 fluidsim-bench 128 -d 3 -s ns3d -it 10
 
 """
+
 import time
 import numpy as np
 from numpy import zeros, sum, float64, sin, cos, prod, asscalar

bench/spectraldns/profile_2d.py

@@ -16,6 +16,7 @@
 
 
 """
+
 import time
 import numpy as np
 
@@ -71,7 +72,7 @@ def update(context):
             "T": 11 * dt,  # Should run 10 iterations
             "write_result": 100,
             "L": [L, L],
-            "M": [10, 10]  # Mesh size is pow(2, M[i]) in direction i
+            "M": [10, 10],  # Mesh size is pow(2, M[i]) in direction i
             # 2**9 == 512
         },
         "doublyperiodic",

bench/spectraldns/profile_3d.py

@@ -13,6 +13,7 @@
   fluidsim-bench 128 -d 3 -s ns3d -it 10
 
 """
+
 import time
 import numpy as np
 from numpy import zeros, sum, float64, sin, cos, prod, asscalar

doc/examples/simul_ns2dbouss_initfields_in_script.py

@@ -3,6 +3,7 @@
 The field initialization is done in the script.
 
 """
+
 import os
 
 import numpy as np

doc/examples/simul_ns2dstrat_forcing_const_energy_rate.py

@@ -1,6 +1,7 @@
 """Script for a short simulation with the solver ns2d.strat
 
 """
+
 import numpy as np
 
 from fluiddyn.util import mpi

doc/examples/simul_ns2dstrat_forcing_in_script.py

@@ -3,6 +3,7 @@
 Note how is used the user-defined forcing maker...
 
 """
+
 from math import pi
 import os
 

doc/examples/simul_ns2dstrat_initfields_in_script.py

@@ -3,6 +3,7 @@
 The field initialization is done in the script.
 
 """
+
 import os
 import numpy as np
 

doc/examples/simul_ns3dbouss_initfields_in_script.py

@@ -7,6 +7,7 @@
   mpirun -np 4 python simul_ns3dbouss_initfields_in_script.py
 
 """
+
 import os
 
 import numpy as np

doc/examples/simul_ns3dstrat_waves.py

@@ -6,6 +6,7 @@
   mpirun -np 4 python simul_ns3dstrat_waves.py
 
 """
+
 import os
 from math import pi
 

doc/examples/time_stepping/3d/run_profile.py

@@ -14,6 +14,7 @@
 ```
 
 """
+
 from pathlib import Path
 import sys
 

fluidsim/base/forcing/kolmogorov.py

@@ -10,6 +10,7 @@
    :private-members:
 
 """
+
 from warnings import warn
 
 import numpy as np

fluidsim/base/output/base.py

@@ -492,9 +492,9 @@ def _compute_mean_values(self, tmin, tmax):
         lengths = self.spectra.compute_length_scales(data=data_spectra)
         result.update(lengths)
 
-        result[
-            "I_dissipation"
-        ] = self.spect_energy_budg.compute_isotropy_dissipation(tmin)
+        result["I_dissipation"] = (
+            self.spect_energy_budg.compute_isotropy_dissipation(tmin)
+        )
 
         return result
 

fluidsim/base/output/cross_corr3d.py

@@ -11,7 +11,6 @@
 
 """
 
-
 import itertools
 
 import numpy as np

fluidsim/base/output/horiz_means.py

@@ -10,6 +10,7 @@
    :undoc-members:
 
 """
+
 import numpy as np
 import h5py
 import matplotlib.pyplot as plt

fluidsim/base/output/spatiotemporal_spectra.py

@@ -1164,9 +1164,7 @@ def plot_kzkhomega(
             omega_disp = ikh / np.sqrt(ikh**2 + dkz_over_dkh**2 * xaxis**2)
             ax.plot(xaxis, omega_disp, "k-", linewidth=2)
         elif equation.startswith(r"$k_z"):
-            omega_disp = xaxis / np.sqrt(
-                xaxis**2 + dkz_over_dkh**2 * ikz**2
-            )
+            omega_disp = xaxis / np.sqrt(xaxis**2 + dkz_over_dkh**2 * ikz**2)
             ax.plot(xaxis, omega_disp, "k-", linewidth=2)
         else:
             raise ValueError("wrong equation for dispersion relation")
@@ -1375,9 +1373,7 @@ def plot_temporal_spectra(
                     aspect_ratio = self.sim.oper.Lx / self.sim.oper.Ly
 
                 def modes(nx, nz):
-                    return np.sqrt(
-                        nx**2 / (nx**2 + aspect_ratio**2 * nz**2)
-                    )
+                    return np.sqrt(nx**2 / (nx**2 + aspect_ratio**2 * nz**2))
 
                 nxs = np.arange(1, 11)
                 modes_nz1 = modes(nxs, 1)

fluidsim/base/output/spectra.py

@@ -17,7 +17,6 @@
 
 """
 
-
 import os
 
 import numpy as np

fluidsim/base/output/spectra3d.py

@@ -23,7 +23,6 @@
 
 """
 
-
 import os
 from textwrap import dedent
 from math import sqrt, tau

fluidsim/base/output/time_signals_fft.py

@@ -108,12 +108,12 @@ def _init_save(self, sim):
         for ishell, kh_s in enumerate(self.kh_shell):
             angle = -np.pi / 2
             for ikps in range(self.nb_k_per_shell):
-                kx_array_ik_approx[
-                    ishell * self.nb_shells + ikps
-                ] = kh_s * np.cos(angle)
-                ky_array_ik_approx[
-                    ishell * self.nb_shells + ikps
-                ] = kh_s * np.sin(angle)
+                kx_array_ik_approx[ishell * self.nb_shells + ikps] = (
+                    kh_s * np.cos(angle)
+                )
+                ky_array_ik_approx[ishell * self.nb_shells + ikps] = (
+                    kh_s * np.sin(angle)
+                )
                 angle += delta_angle
 
         self.ik0_array_ik = np.empty([self.nb_k_tot], dtype=np.int32)
@@ -161,9 +161,7 @@ def _init_save(self, sim):
                 self.ky_array_ik[ik] = ky_1k
 
         if mpi.rank == 0:
-            self.kh_array_ik = np.sqrt(
-                self.kx_array_ik**2 + self.ky_array_ik**2
-            )
+            self.kh_array_ik = np.sqrt(self.kx_array_ik**2 + self.ky_array_ik**2)
 
             self.omega_array_ik = self.output.omega_from_wavenumber(
                 self.kh_array_ik

fluidsim/base/params.py

@@ -10,6 +10,7 @@
 
 
 """
+
 from fluidsim_core.params import Parameters
 from fluidsim.base.solvers.info_base import InfoSolverBase
 

fluidsim/base/preprocess/base.py

@@ -8,6 +8,7 @@
    :private-members:
 
 """
+
 from fluidsim_core.params import iter_complete_params
 
 

fluidsim/base/preprocess/pseudo_spect.py

@@ -348,9 +348,7 @@ def calcul_viscosity(
                 nu = length_scale**order / time_scale
                 v.append(nu)
                 if verbose:
-                    kolmo_len.append(
-                        (nu**3 / epsilon) ** (1.0 / (3 * order - 2))
-                    )
+                    kolmo_len.append((nu**3 / epsilon) ** (1.0 / (3 * order - 2)))
             else:
                 v.append(0.0)
 

fluidsim/base/solvers/base.py

@@ -12,6 +12,7 @@
    :private-members:
 
 """
+
 from time import time
 import atexit
 from pathlib import Path

fluidsim/base/solvers/info_base.py

@@ -6,6 +6,7 @@
    :private-members:
 
 """
+
 from copy import deepcopy
 
 from fluidsim_core.info import InfoSolverCore, create_info_simul  # noqa

fluidsim/base/time_stepping/pseudo_spect.py

@@ -21,6 +21,7 @@
   https://ntrs.nasa.gov/archive/nasa/casi.ntrs.nasa.gov/19810022965.pdf).
 
 """
+
 from random import randint
 
 import numpy as np

fluidsim/base/turb_model/smagorinsky.py

@@ -36,6 +36,7 @@ class SmagorinskyModel(SpecificTurbModelSpectral):
     with :math:`C = 0.18` and :math:`\Delta = L_x / n_x`.
 
     """
+
     _module_name = "fluidsim.base.turb_model.smagorinsky"
     tag = "smagorinsky"
 

fluidsim/base/turb_model/stress_tensor.py

@@ -40,6 +40,4 @@ def compute_stress_tensor(self, ux_fft, uy_fft, uz_fft):
         return Sxx, Syy, Szz, Syx, Szx, Szy
 
     def compute_norm(self, Sxx, Syy, Szz, Syx, Szx, Szy):
-        return np.sqrt(
-            Sxx**2 + Syy**2 + Szz**2 + 2 * (Syx**2 + Szx**2 + Szy**2)
-        )
+        return np.sqrt(Sxx**2 + Syy**2 + Szz**2 + 2 * (Syx**2 + Szx**2 + Szy**2))

fluidsim/magic.py

@@ -6,6 +6,7 @@
    :private-members:
 
 """
+
 from fluiddyn.io.query import query_yes_no
 from fluidsim_core.magic import MagicsCore
 from IPython.core.magic import line_magic, magics_class

fluidsim/operators/sphericalharmo.py

@@ -3,6 +3,7 @@
 
 
 """
+
 import numpy as np
 
 from pyshtools.constants import Earth

fluidsim/solvers/__init__.py

@@ -28,6 +28,7 @@
    pkgload
 
 """
+
 import numpy as _np
 from warnings import warn as _warn
 

fluidsim/solvers/ad1d/init_fields.py

@@ -11,6 +11,7 @@
    :members:
 
 """
+
 import numpy as np
 
 from fluidsim.base.init_fields import InitFieldsBase, SpecificInitFields

fluidsim/solvers/ad1d/pseudo_spect/solver.py

@@ -10,6 +10,7 @@
 
 
 """
+
 from fluidsim.base.setofvariables import SetOfVariables
 from fluidsim.base.solvers.pseudo_spect import (
     SimulBasePseudoSpectral,

fluidsim/solvers/ad1d/solver.py

@@ -52,6 +52,7 @@ class Simul(SimulBase):
     constant velocity.
 
     """
+
     InfoSolver = InfoSolverAD1D
 
     @staticmethod

fluidsim/solvers/burgers1d/skew_sym/solver.py

@@ -9,6 +9,7 @@
 
 
 """
+
 from fluidsim.base.setofvariables import SetOfVariables
 from ..solver import Simul as SimulBurgers, InfoSolver as InfoSolverBurgers
 

fluidsim/solvers/burgers1d/solver.py

@@ -9,6 +9,7 @@
 
 
 """
+
 from fluidsim.base.setofvariables import SetOfVariables
 from fluidsim.base.solvers.pseudo_spect import (
     SimulBasePseudoSpectral,

fluidsim/solvers/nl1d/solver.py

@@ -8,6 +8,7 @@
    :private-members:
 
 """
+
 import numpy as np
 
 from fluidsim.base.setofvariables import SetOfVariables

fluidsim/solvers/ns2d/bouss/solver.py

@@ -6,6 +6,7 @@
    :private-members:
 
 """
+
 import numpy as np
 
 from transonic import boost, Array

fluidsim/solvers/ns2d/strat/forcing.py

@@ -10,7 +10,6 @@
 
 
 class ForcingNS2DStrat(ForcingNS2D):
-
     """Forcing class for the ns2d strat solver.
 
     .. inheritance-diagram:: ForcingNS2DStrat

fluidsim/solvers/ns2d/strat/output/_old_miguel_frequency_spectra.py

@@ -230,12 +230,12 @@ def _online_save(self):
                         :: self.spatial_decimate, :: self.spatial_decimate
                     ]
 
-                    self.temp_array_new[
-                        0, self.nb_times_in_temp_array, :, :
-                    ] = field_ap_decimate
-                    self.temp_array_new[
-                        1, self.nb_times_in_temp_array, :, :
-                    ] = field_am_decimate
+                    self.temp_array_new[0, self.nb_times_in_temp_array, :, :] = (
+                        field_ap_decimate
+                    )
+                    self.temp_array_new[1, self.nb_times_in_temp_array, :, :] = (
+                        field_am_decimate
+                    )
 
                 # Save the time to self.times_arr
                 self.times_arr[self.nb_times_in_temp_array] = (