def test_roundtrip_outofplace(self, **kwargs):
shift = kwargs.get('shift', 0)
grid = np.arange(128)
mu = 43.
stdv = 3.
g = (np.exp(-(grid - mu)**2 / (2 * stdv**2)) /
(2 * np.pi * stdv**2)).astype(self.dt)
g2 = g.copy()
g_bkp = g.copy()
## gw = np.empty_like(g)
## grt = np.empty_like(g)
gw = _fftn(g, inplace=False, shift=shift)
gw_np = reference_fftn(g2, axes=(0, ), shift=shift)
grt = _ifftn(gw, inplace=False, shift=shift)
grt_np = reference_ifftn(gw, axes=(0, ), shift=shift)
# strong assertion that g is not altered at all
assert (g == g_bkp).all(), 'out of place xform error, shift=%d' % shift
self.comp(gw,
gw_np,
err_msg='differs from numpy fft ref, shift=%d' % shift)
self.nrg_comp(gw, gw_np)
self.comp(g,
grt,
err_msg='roundtrip transforms diverge, shift=%d' % shift)
self.nrg_comp(g, grt)
def test_roundtrip_outofplace(self, **kwargs):
shift = kwargs.get('shift', 0)
grid = np.arange(128)
mu = 43.
stdv = 3.
g = (np.exp(-(grid-mu)**2 / (2*stdv**2)) / (2*np.pi*stdv**2)).astype(self.dt)
g2 = g.copy()
g_bkp = g.copy()
## gw = np.empty_like(g)
## grt = np.empty_like(g)
gw = _fftn(g, inplace=False, shift=shift)
gw_np = reference_fftn(g2, axes=(0,), shift=shift)
grt = _ifftn(gw, inplace=False, shift=shift)
grt_np = reference_ifftn(gw, axes=(0,), shift=shift)
# strong assertion that g is not altered at all
assert (g==g_bkp).all(), 'out of place xform error, shift=%d'%shift
self.comp(gw, gw_np,
err_msg='differs from numpy fft ref, shift=%d'%shift)
self.nrg_comp(gw, gw_np)
self.comp(g, grt,
err_msg='roundtrip transforms diverge, shift=%d'%shift)
self.nrg_comp(g, grt)
def test_roundtrip_inplace(self, **kwargs):
shift = kwargs.get('shift', 0)
grid = np.arange(128)
mu = 43.
stdv = 3.
g = (np.exp(-(grid-mu)**2 / (2*stdv**2)) / (2*np.pi*stdv**2)).astype(self.dt)
g2 = g.copy()
g_bkp = g.copy()
_fftn(g, inplace=True, shift=shift)
gw_np = reference_fftn(g2, shift=shift, axes=(0,))
self.comp(g, gw_np,
err_msg='differs from numpy fft ref, shift=%d'%shift)
self.nrg_comp(g, gw_np)
_ifftn(g, inplace=True, shift=shift)
self.comp(g_bkp, g,
err_msg='roundtrip transforms diverge, shift=%d'%shift)
self.nrg_comp(g_bkp, g)
def test_roundtrip_inplace(self, **kwargs):
shift = kwargs.get('shift', 0)
grid = np.arange(128)
mu = 43.
stdv = 3.
g = (np.exp(-(grid - mu)**2 / (2 * stdv**2)) /
(2 * np.pi * stdv**2)).astype(self.dt)
g2 = g.copy()
g_bkp = g.copy()
_fftn(g, inplace=True, shift=shift)
gw_np = reference_fftn(g2, shift=shift, axes=(0, ))
self.comp(g,
gw_np,
err_msg='differs from numpy fft ref, shift=%d' % shift)
self.nrg_comp(g, gw_np)
_ifftn(g, inplace=True, shift=shift)
self.comp(g_bkp,
g,
err_msg='roundtrip transforms diverge, shift=%d' % shift)
self.nrg_comp(g_bkp, g)
