Apply black style checks

Author: ndgrigorian

mkl_fft/_numpy_fft.py

@@ -512,7 +512,9 @@ def irfft(a, n=None, axis=-1, norm=None):
     else:
         fsc = ortho_sc_1d(nn, nn)
 
-    return trycall(mkl_fft.irfft, (x,), {"n": n, "axis": axis, "fwd_scale": fsc})
+    return trycall(
+        mkl_fft.irfft, (x,), {"n": n, "axis": axis, "fwd_scale": fsc}
+    )
 
 
 def hfft(a, n=None, axis=-1, norm=None):
@@ -604,7 +606,9 @@ def hfft(a, n=None, axis=-1, norm=None):
     else:
         fsc = ortho_sc_1d(nn, nn)
 
-    return trycall(mkl_fft.irfft, (x,), {"n": n, "axis": axis, "fwd_scale": fsc})
+    return trycall(
+        mkl_fft.irfft, (x,), {"n": n, "axis": axis, "fwd_scale": fsc}
+    )
 
 
 def ihfft(a, n=None, axis=-1, norm=None):
@@ -676,7 +680,9 @@ def ihfft(a, n=None, axis=-1, norm=None):
     else:
         fsc = ortho_sc_1d(n, x.shape[axis])
 
-    output = trycall(mkl_fft.rfft, (x,), {"n": n, "axis": axis, "fwd_scale": fsc})
+    output = trycall(
+        mkl_fft.rfft, (x,), {"n": n, "axis": axis, "fwd_scale": fsc}
+    )
 
     conjugate(output, out=output)
     return output
@@ -922,7 +928,9 @@ def ifftn(a, s=None, axes=None, norm=None):
     else:
         fsc = sqrt(frwd_sc_nd(s, x.shape))
 
-    return trycall(mkl_fft.ifftn, (x,), {"s": s, "axes": axes, "fwd_scale": fsc})
+    return trycall(
+        mkl_fft.ifftn, (x,), {"s": s, "axes": axes, "fwd_scale": fsc}
+    )
 
 
 def fft2(a, s=None, axes=(-2, -1), norm=None):
@@ -1219,7 +1227,9 @@ def rfftn(a, s=None, axes=None, norm=None):
         s, axes = _cook_nd_args(x, s, axes)
         fsc = sqrt(frwd_sc_nd(s, x.shape))
 
-    return trycall(mkl_fft.rfftn, (x,), {"s": s, "axes": axes, "fwd_scale": fsc})
+    return trycall(
+        mkl_fft.rfftn, (x,), {"s": s, "axes": axes, "fwd_scale": fsc}
+    )
 
 
 def rfft2(a, s=None, axes=(-2, -1), norm=None):
@@ -1371,7 +1381,9 @@ def irfftn(a, s=None, axes=None, norm=None):
         s, axes = _cook_nd_args(x, s, axes, invreal=1)
         fsc = sqrt(frwd_sc_nd(s, x.shape))
 
-    return trycall(mkl_fft.irfftn, (x,), {"s": s, "axes": axes, "fwd_scale": fsc})
+    return trycall(
+        mkl_fft.irfftn, (x,), {"s": s, "axes": axes, "fwd_scale": fsc}
+    )
 
 
 def irfft2(a, s=None, axes=(-2, -1), norm=None):

mkl_fft/_scipy_fft.py

@@ -267,7 +267,9 @@ def _compute_nd_fwd_scale(norm, s, axes, x_shape):
     return fsc
 
 
-def fft(a, n=None, axis=-1, norm=None, overwrite_x=False, workers=None, plan=None):
+def fft(
+    a, n=None, axis=-1, norm=None, overwrite_x=False, workers=None, plan=None
+):
     try:
         x = _float_utils.__supported_array_or_not_implemented(a)
     except ValueError:
@@ -277,11 +279,15 @@ def fft(a, n=None, axis=-1, norm=None, overwrite_x=False, workers=None, plan=Non
     fsc = _compute_1d_fwd_scale(norm, n, x.shape[axis])
     _check_plan(plan)
     with Workers(workers):
-        output = mkl_fft.fft(x, n=n, axis=axis, overwrite_x=overwrite_x, fwd_scale=fsc)
+        output = mkl_fft.fft(
+            x, n=n, axis=axis, overwrite_x=overwrite_x, fwd_scale=fsc
+        )
     return output
 
 
-def ifft(a, n=None, axis=-1, norm=None, overwrite_x=False, workers=None, plan=None):
+def ifft(
+    a, n=None, axis=-1, norm=None, overwrite_x=False, workers=None, plan=None
+):
     try:
         x = _float_utils.__supported_array_or_not_implemented(a)
     except ValueError:
@@ -291,12 +297,20 @@ def ifft(a, n=None, axis=-1, norm=None, overwrite_x=False, workers=None, plan=No
     fsc = _compute_1d_fwd_scale(norm, n, x.shape[axis])
     _check_plan(plan)
     with Workers(workers):
-        output = mkl_fft.ifft(x, n=n, axis=axis, overwrite_x=overwrite_x, fwd_scale=fsc)
+        output = mkl_fft.ifft(
+            x, n=n, axis=axis, overwrite_x=overwrite_x, fwd_scale=fsc
+        )
     return output
 
 
 def fft2(
-    a, s=None, axes=(-2, -1), norm=None, overwrite_x=False, workers=None, plan=None
+    a,
+    s=None,
+    axes=(-2, -1),
+    norm=None,
+    overwrite_x=False,
+    workers=None,
+    plan=None,
 ):
     try:
         x = _float_utils.__supported_array_or_not_implemented(a)
@@ -307,12 +321,20 @@ def fft2(
     fsc = _compute_nd_fwd_scale(norm, s, axes, x.shape)
     _check_plan(plan)
     with Workers(workers):
-        output = mkl_fft.fftn(x, s=s, axes=axes, overwrite_x=overwrite_x, fwd_scale=fsc)
+        output = mkl_fft.fftn(
+            x, s=s, axes=axes, overwrite_x=overwrite_x, fwd_scale=fsc
+        )
     return output
 
 
 def ifft2(
-    a, s=None, axes=(-2, -1), norm=None, overwrite_x=False, workers=None, plan=None
+    a,
+    s=None,
+    axes=(-2, -1),
+    norm=None,
+    overwrite_x=False,
+    workers=None,
+    plan=None,
 ):
     try:
         x = _float_utils.__supported_array_or_not_implemented(a)
@@ -329,7 +351,9 @@ def ifft2(
     return output
 
 
-def fftn(a, s=None, axes=None, norm=None, overwrite_x=False, workers=None, plan=None):
+def fftn(
+    a, s=None, axes=None, norm=None, overwrite_x=False, workers=None, plan=None
+):
     try:
         x = _float_utils.__supported_array_or_not_implemented(a)
     except ValueError:
@@ -339,11 +363,15 @@ def fftn(a, s=None, axes=None, norm=None, overwrite_x=False, workers=None, plan=
     fsc = _compute_nd_fwd_scale(norm, s, axes, x.shape)
     _check_plan(plan)
     with Workers(workers):
-        output = mkl_fft.fftn(x, s=s, axes=axes, overwrite_x=overwrite_x, fwd_scale=fsc)
+        output = mkl_fft.fftn(
+            x, s=s, axes=axes, overwrite_x=overwrite_x, fwd_scale=fsc
+        )
     return output
 
 
-def ifftn(a, s=None, axes=None, norm=None, overwrite_x=False, workers=None, plan=None):
+def ifftn(
+    a, s=None, axes=None, norm=None, overwrite_x=False, workers=None, plan=None
+):
     try:
         x = _float_utils.__supported_array_or_not_implemented(a)
     except ValueError:
@@ -423,7 +451,9 @@ def irfft2(a, s=None, axes=(-2, -1), norm=None, workers=None, plan=None):
         return NotImplemented
     if x is NotImplemented:
         return x
-    s, axes, fsc = _compute_nd_fwd_scale_for_rfft(norm, s, axes, x, invreal=True)
+    s, axes, fsc = _compute_nd_fwd_scale_for_rfft(
+        norm, s, axes, x, invreal=True
+    )
     _check_plan(plan)
     with Workers(workers):
         output = mkl_fft.irfftn(x, s, axes, fwd_scale=fsc)
@@ -451,7 +481,9 @@ def irfftn(a, s=None, axes=None, norm=None, workers=None, plan=None):
         return NotImplemented
     if x is NotImplemented:
         return x
-    s, axes, fsc = _compute_nd_fwd_scale_for_rfft(norm, s, axes, x, invreal=True)
+    s, axes, fsc = _compute_nd_fwd_scale_for_rfft(
+        norm, s, axes, x, invreal=True
+    )
     _check_plan(plan)
     with Workers(workers):
         output = mkl_fft.irfftn(x, s, axes, fwd_scale=fsc)

mkl_fft/tests/test_fft1d.py

@@ -55,7 +55,9 @@ def setUp(self):
         rnd.seed(1234567)
         self.xd1 = rnd.standard_normal(128)
         self.xf1 = self.xd1.astype(np.float32)
-        self.xz1 = rnd.standard_normal((128, 2)).view(dtype=np.complex128).squeeze()
+        self.xz1 = (
+            rnd.standard_normal((128, 2)).view(dtype=np.complex128).squeeze()
+        )
         self.xc1 = self.xz1.astype(np.complex64)
 
     def test_vector1(self):

mkl_fft/tests/test_fftnd.py

@@ -192,7 +192,14 @@ def test_cf_contig(self):
 
     def test_rfftn(self):
         """Test that rfftn works as expected"""
-        axes = [(0, 1, 2), (0, 2, 1), (1, 0, 2), (1, 2, 0), (2, 0, 1), (2, 1, 0)]
+        axes = [
+            (0, 1, 2),
+            (0, 2, 1),
+            (1, 0, 2),
+            (1, 2, 0),
+            (2, 0, 1),
+            (2, 1, 0),
+        ]
         for x in [self.ad, self.af]:
             for a in axes:
                 r_tol, a_tol = _get_rtol_atol(x)
@@ -274,7 +281,9 @@ def test_gh109():
     b = np.asarray(b_int, dtype=np.float32)
 
     r1 = mkl_fft.fftn(b, s=None, axes=(0,), overwrite_x=False, fwd_scale=1 / 3)
-    r2 = mkl_fft.fftn(b_int, s=None, axes=(0,), overwrite_x=False, fwd_scale=1 / 3)
+    r2 = mkl_fft.fftn(
+        b_int, s=None, axes=(0,), overwrite_x=False, fwd_scale=1 / 3
+    )
 
     rtol, atol = _get_rtol_atol(b)
     assert_allclose(r1, r2, rtol=rtol, atol=atol)

mkl_fft/tests/test_interfaces.py

@@ -30,7 +30,9 @@
 
 
 @pytest.mark.parametrize("norm", [None, "forward", "backward", "ortho"])
-@pytest.mark.parametrize("dtype", [np.float32, np.float64, np.complex64, np.complex128])
+@pytest.mark.parametrize(
+    "dtype", [np.float32, np.float64, np.complex64, np.complex128]
+)
 def test_scipy_fft(norm, dtype):
     x = np.ones(511, dtype=dtype)
     w = mfi.scipy_fft.fft(x, norm=norm, workers=None, plan=None)
@@ -40,7 +42,9 @@ def test_scipy_fft(norm, dtype):
 
 
 @pytest.mark.parametrize("norm", [None, "forward", "backward", "ortho"])
-@pytest.mark.parametrize("dtype", [np.float32, np.float64, np.complex64, np.complex128])
+@pytest.mark.parametrize(
+    "dtype", [np.float32, np.float64, np.complex64, np.complex128]
+)
 def test_numpy_fft(norm, dtype):
     x = np.ones(511, dtype=dtype)
     w = mfi.numpy_fft.fft(x, norm=norm)
@@ -54,7 +58,9 @@ def test_numpy_fft(norm, dtype):
 def test_scipy_rfft(norm, dtype):
     x = np.ones(511, dtype=dtype)
     w = mfi.scipy_fft.rfft(x, norm=norm, workers=None, plan=None)
-    xx = mfi.scipy_fft.irfft(w, n=x.shape[0], norm=norm, workers=None, plan=None)
+    xx = mfi.scipy_fft.irfft(
+        w, n=x.shape[0], norm=norm, workers=None, plan=None
+    )
     tol = 64 * np.finfo(np.dtype(dtype)).eps
     assert np.allclose(x, xx, atol=tol, rtol=tol)
 
@@ -76,7 +82,9 @@ def test_numpy_rfft(norm, dtype):
 
 
 @pytest.mark.parametrize("norm", [None, "forward", "backward", "ortho"])
-@pytest.mark.parametrize("dtype", [np.float32, np.float64, np.complex64, np.complex128])
+@pytest.mark.parametrize(
+    "dtype", [np.float32, np.float64, np.complex64, np.complex128]
+)
 def test_scipy_fftn(norm, dtype):
     x = np.ones((37, 83), dtype=dtype)
     w = mfi.scipy_fft.fftn(x, norm=norm, workers=None, plan=None)
@@ -86,7 +94,9 @@ def test_scipy_fftn(norm, dtype):
 
 
 @pytest.mark.parametrize("norm", [None, "forward", "backward", "ortho"])
-@pytest.mark.parametrize("dtype", [np.float32, np.float64, np.complex64, np.complex128])
+@pytest.mark.parametrize(
+    "dtype", [np.float32, np.float64, np.complex64, np.complex128]
+)
 def test_numpy_fftn(norm, dtype):
     x = np.ones((37, 83), dtype=dtype)
     w = mfi.numpy_fft.fftn(x, norm=norm)