aufraeumen

fresnel/propagate.py

-from pyfftw.interfaces.numpy_fft import fftn, ifftn, ifftshift, fftshift
 import numpy as np
 from functools import reduce
+from pyfftw.interfaces.numpy_fft import fftn, ifftn, ifftshift, fftshift
 
 from . import finite_differences as fd
 from .misc import fftfreqn
@@ -9,8 +9,7 @@ _lambda0 = 1.  # all lengths are in units of the wavelength
 _k0 = 2 * np.pi / _lambda0
 
 
-def _compute_potential(n_, k_=_k0):
-    return k_*k_ * (1 - n_*n_)
+def _compute_potential(n_, k_=_k0): return k_*k_ * (1 - n_*n_)
 
 
 def squaresum(a):

fresnel/vacuum.py

-from . import hankel
-from .misc import fftfreqn, gridn
-
 import numpy as np
 from pyfftw.interfaces.numpy_fft import fftn, ifftn
 from scipy.signal import fftconvolve
 
-_lambda0 = 1.  # all lengths are in units of the wavelength
-_k0 = 2 * np.pi / _lambda0
+from . import hankel
+from .misc import fftfreqn, gridn
 
 
 def fresnelKernelCS(N, fresnelNumber):