expose mkl_fft.rfft2_numpy and mkl_fft.irfft2_numpy

Author: Vahid Tavanashad

mkl_fft/__init__.py

@@ -25,10 +25,12 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 from ._pydfti import (fft, ifft, fft2, ifft2, fftn, ifftn, rfft, irfft,
-                      rfft_numpy, irfft_numpy, rfftn_numpy, irfftn_numpy)
+                      rfft_numpy, irfft_numpy, rfft2_numpy, irfft2_numpy,
+                      rfftn_numpy, irfftn_numpy)
 
 from ._version import __version__
 import mkl_fft.interfaces
 
 __all__ = ['fft', 'ifft', 'fft2', 'ifft2', 'fftn', 'ifftn', 'rfft', 'irfft',
-           'rfft_numpy', 'irfft_numpy', 'rfftn_numpy', 'irfftn_numpy', 'interfaces']
+           'rfft_numpy', 'irfft_numpy', 'rfft2_numpy', 'irfft2_numpy',
+           'rfftn_numpy', 'irfftn_numpy', 'interfaces']

mkl_fft/_pydfti.pyx

@@ -1122,11 +1122,11 @@ def ifftn(x, shape=None, axes=None, overwrite_x=False, forward_scale=1.0):
 
 
 def rfft2_numpy(x, s=None, axes=(-2,-1), forward_scale=1.0):
-    return rfftn_numpy(x, s=s, axes=axes, fsc=forward_scale)
+    return rfftn_numpy(x, s=s, axes=axes, forward_scale=forward_scale)
 
 
 def irfft2_numpy(x, s=None, axes=(-2,-1), forward_scale=1.0):
-    return irfftn_numpy(x, s=s, axes=axes, fsc=forward_scale)
+    return irfftn_numpy(x, s=s, axes=axes, forward_scale=forward_scale)
 
 
 def _remove_axis(s, axes, axis_to_remove):

mkl_fft/tests/test_fftnd.py

@@ -228,3 +228,18 @@ def test_gh109():
 
     rtol, atol = _get_rtol_atol(b)
     assert_allclose(r1, r2, rtol=rtol, atol=atol)
+
+
+def test_rfftn_numpy():
+    x = np.ones((37, 83))
+
+    w = mkl_fft.rfftn_numpy(x)
+    xx = mkl_fft.irfftn_numpy(w, s=x.shape)
+    tol = 64 * np.finfo(np.dtype(x.dtype)).eps
+    assert np.allclose(x, xx, atol=tol, rtol=tol)
+
+    w = mkl_fft.rfft2_numpy(x)
+    xx = mkl_fft.irfft2_numpy(w, s=x.shape)
+    tol = 64 * np.finfo(np.dtype(x.dtype)).eps
+    assert np.allclose(x, xx, atol=tol, rtol=tol)  
+