def test_fft_freq_execution(setup):
t = fftfreq(10, .1, chunk_size=6)
res = t.execute().fetch()
np.testing.assert_allclose(res, np.fft.fftfreq(10, .1))
t = fftfreq(11, .01, chunk_size=6)
res = t.execute().fetch()
np.testing.assert_allclose(res, np.fft.fftfreq(11, .01))
def testFFTFreqExecution(self):
t = fftfreq(10, .1, chunk_size=3)
res = self.executor.execute_tensor(t, concat=True)[0]
np.testing.assert_allclose(res, np.fft.fftfreq(10, .1))
t = fftfreq(11, .01, chunk_size=3)
res = self.executor.execute_tensor(t, concat=True)[0]
np.testing.assert_allclose(res, np.fft.fftfreq(11, .01))
def testFFTShiftExecution(self):
t = fftfreq(10, .1, chunk_size=3)
r = fftshift(t)
res = self.executor.execute_tensor(r, concat=True)[0]
np.testing.assert_allclose(res, np.fft.fftshift(np.fft.fftfreq(10, .1)))
freqs = fftfreq(9, d=1./9, chunk_size=2).reshape(3, 3)
r = fftshift(freqs, axes=(1,))
res = self.executor.execute_tensor(r, concat=True)[0]
expected = np.fft.fftshift(np.fft.fftfreq(9, d=1./9).reshape(3, 3), axes=(1,))
np.testing.assert_allclose(res, expected)
def test_fft_shift_execution(setup):
t = fftfreq(10, .1, chunk_size=6)
r = fftshift(t)
res = r.execute().fetch()
np.testing.assert_allclose(res, np.fft.fftshift(np.fft.fftfreq(10, .1)))
freqs = fftfreq(9, d=1. / 9, chunk_size=4).reshape(3, 3)
r = fftshift(freqs, axes=(1, ))
res = r.execute().fetch()
expected = np.fft.fftshift(np.fft.fftfreq(9, d=1. / 9).reshape(3, 3),
axes=(1, ))
np.testing.assert_allclose(res, expected)
def test_ifft_shift_execution(setup):
t = fftfreq(9, d=1. / 9, chunk_size=4).reshape(3, 3)
r = ifftshift(t)
res = r.execute().fetch()
expected = np.fft.ifftshift(np.fft.fftfreq(9, d=1. / 9).reshape(3, 3))
np.testing.assert_allclose(res, expected)
def testIFFTShiftExecution(self):
t = fftfreq(9, d=1. / 9, chunk_size=2).reshape(3, 3)
r = ifftshift(t)
res = self.executor.execute_tensor(r, concat=True)[0]
expected = np.fft.ifftshift(np.fft.fftfreq(9, d=1. / 9).reshape(3, 3))
np.testing.assert_allclose(res, expected)
def test_fft_shift():
freqs = fftfreq(9, d=1. / 9).reshape(3, 3)
t = ifftshift(fftshift(freqs))
assert t.dtype is not None
expect_dtype = np.fft.ifftshift(np.fft.fftshift(np.fft.fftfreq(9, d=1. / 9).reshape(3, 3))).dtype
assert t.dtype == expect_dtype
def testFFTShift(self):
freqs = fftfreq(9, d=1. / 9).reshape(3, 3)
t = ifftshift(fftshift(freqs))
self.assertIsNotNone(t.dtype)
expect_dtype = np.fft.ifftshift(
np.fft.fftshift(np.fft.fftfreq(9, d=1. / 9).reshape(3, 3))).dtype
self.assertEqual(t.dtype, expect_dtype)
def testFFTFreq(self):
t = fftfreq(10, .1, chunk_size=3)
self.assertEqual(t.shape, np.fft.fftfreq(10, .1).shape)
t.tiles()
self.assertEqual(t.shape, tuple(sum(ns) for ns in t.nsplits))
t = rfftfreq(10, .1, chunk_size=3)
self.assertEqual(t.shape, np.fft.rfftfreq(10, .1).shape)
t.tiles()
self.assertEqual(t.shape, tuple(sum(ns) for ns in t.nsplits))
def test_fft_freq():
t = fftfreq(10, .1, chunk_size=3)
assert t.shape == np.fft.fftfreq(10, .1).shape
t = tile(t)
assert t.shape == tuple(sum(ns) for ns in t.nsplits)
t = rfftfreq(10, .1, chunk_size=3)
assert t.shape == np.fft.rfftfreq(10, .1).shape
t = tile(t)
assert t.shape == tuple(sum(ns) for ns in t.nsplits)
