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

@@ -227,4 +227,4 @@ def test_gh109():
     r2 = mkl_fft.fftn(b_int, shape=None, axes=(0,), overwrite_x=False, forward_scale=1/3)
 
     rtol, atol = _get_rtol_atol(b)
-    assert_allclose(r1, r2, rtol=rtol, atol=atol)
+    assert_allclose(r1, r2, rtol=rtol, atol=atol)