Apply suggestions from code review

Co-authored-by: ndgrigorian <[email protected]>

Author: vtavana

mkl_fft/_pydfti.pyx

@@ -1208,7 +1208,7 @@ def rfftn(x, s=None, axes=None, fwd_scale=1.0):
                 if a_res is not a_inp:
                     a[tind] = a_res # copy in place
         else:
-            for ii in range(len(axes)-2, -1, -1):
+            for ii in range(len(axes) - 2, -1, -1):
                 a = fft(a, s[ii], axes[ii], overwrite_x=True)
     return a
 

mkl_fft/tests/test_interfaces.py

@@ -154,4 +154,4 @@ def test_axes(func):
     res = func(x, axes=(1, 2))
     exp = np.fft.rfft2(x, axes=(1, 2))
     tol = 64 * np.finfo(np.float64).eps
-    assert np.allclose(res, exp, atol=tol, rtol=tol)
+    assert np.allclose(res, exp, atol=tol, rtol=tol)

mkl_fft/tests/test_pocketfft.py

@@ -513,7 +513,7 @@ def test_fft_with_order(dtype, order, fft):
         for ax in axes:
             X_res = fft(X, axes=ax)
             Y_res = fft(Y, axes=ax)
-            assert_allclose(X_res, Y_res, atol=_tol, rtol=10*_tol)
+            assert_allclose(X_res, Y_res, atol=_tol, rtol=10 * _tol)
     else:
         raise ValueError