Python fftshiftの例

コード例 #1

ファイル: test_fft_execute.py プロジェクト: Haxine/mars-1

    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)

コード例 #2

ファイル: test_fft_execute.py プロジェクト: haijohn/mars

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)

コード例 #3

ファイル: test_fft.py プロジェクト: fyrestone/mars

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

コード例 #4

    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)