def test_twice_odd_length2():
# functions with dimension lengths such that N_i/2 is odd need
# special treatment in the output modulation
a = np.random.randn(30, 40).astype('D')
A_ref = reference_fftn(a, axes=(0, 1), shift=True)
A_test = fft2(a, shift=True)
assert np.allclose(A_ref, A_test), 'twice-odd length FT fails in 2D'
def test_twice_odd_length2():
# functions with dimension lengths such that N_i/2 is odd need
# special treatment in the output modulation
a = np.random.randn(30,40).astype('D')
A_ref = reference_fftn(a, axes=(0,1), shift=True)
A_test = fft2(a, shift=True)
assert np.allclose(A_ref, A_test), 'twice-odd length FT fails in 2D'
def bench_fft2_time():
from numpy.fft import fftn as numpy_fftn
from scipy.fftpack import fftn as scipy_fftn
print
print ' 2D Double Precision Fast Fourier Transform'
print '==================================================='
print ' | complex input '
print '---------------------------------------------------'
print ' size | fftw | numpy | scipy |'
print '---------------------------------------------------'
for size, repeat in [
((100, 100), 100),
((1000, 100), 7),
((256, 256), 10),
((512, 512), 3),
]:
print '%9s' % ('%sx%s' % size),
sys.stdout.flush()
x = random(repeat, *size).astype('D') + \
random(repeat, *size).astype('D')*1j
tr0 = time.time()
y = fft2(x, shift=False)
trf = time.time()
print '|%8.2f' % (trf - tr0),
sys.stdout.flush()
tn0 = time.time()
ny = numpy_fftn(x, axes=(-2, -1))
tnf = time.time()
## assert_array_almost_equal(ny,y)
print '|%8.2f' % (tnf - tn0),
sys.stdout.flush()
ts0 = time.time()
sy = scipy_fftn(x, axes=(-2, -1))
tsf = time.time()
## assert_array_almost_equal(sy,y)
print '|%8.2f' % (tsf - ts0),
sys.stdout.flush()
print ' (secs for %s calls)' % (repeat)
sys.stdout.flush()
print
print ' 2D Double Precision Fast Fourier Transform'
print '==================================================='
print ' | complex input shifted '
print '---------------------------------------------------'
print ' size | fftw | numpy | scipy |'
print '---------------------------------------------------'
for size, repeat in [
((100, 100), 100),
((1000, 100), 7),
((256, 256), 10),
((512, 512), 3),
]:
chk = checkerboard(*size).astype('d')
print '%9s' % ('%sx%s' % size),
sys.stdout.flush()
x = random(repeat, *size).astype('D') + \
random(repeat, *size).astype('D')*1j
tr0 = time.time()
y = fft2(x, shift=True)
trf = time.time()
print '|%8.2f' % (trf - tr0),
sys.stdout.flush()
tn0 = time.time()
ny = numpy_fftn(x, axes=(-2, -1))
tnf = time.time()
## assert_array_almost_equal(ny,y)
print '|%8.2f' % (tnf - tn0),
sys.stdout.flush()
ts0 = time.time()
sy = scipy_fftn(x, axes=(-2, -1))
tsf = time.time()
## assert_array_almost_equal(sy,y)
print '|%8.2f' % (tsf - ts0),
sys.stdout.flush()
print ' (secs for %s calls)' % (repeat)
sys.stdout.flush()
print
print ' 2D Single Precision Fast Fourier Transform'
print '==================================================='
print ' | complex input '
print '---------------------------------------------------'
print ' size | fftw | numpy | scipy* |'
print '---------------------------------------------------'
for size, repeat in [
((100, 100), 100),
((1000, 100), 7),
((256, 256), 10),
((512, 512), 3),
]:
print '%9s' % ('%sx%s' % size),
sys.stdout.flush()
x = random(repeat, *size).astype('F') + \
random(repeat, *size).astype('F')*1j
tr0 = time.time()
y = fft2(x, shift=False)
trf = time.time()
print '|%8.2f' % (trf - tr0),
sys.stdout.flush()
tn0 = time.time()
ny = numpy_fftn(x, axes=(-2, -1))
tnf = time.time()
## assert_array_almost_equal(ny,y, decimal=2)
print '|%8.2f' % (tnf - tn0),
sys.stdout.flush()
ts0 = time.time()
sy = scipy_fftn(x.astype('D'), axes=(-2, -1)).astype('F')
tsf = time.time()
## assert_array_almost_equal(sy,y, decimal=2)
print '|%8.2f' % (tsf - ts0),
sys.stdout.flush()
print ' (secs for %s calls)' % (repeat)
print "(* casted float->FT{double}->float)"
sys.stdout.flush()
print
print ' 2D Single Precision Fast Fourier Transform'
print '==================================================='
print ' | complex input shifted '
print '---------------------------------------------------'
print ' size | fftw | numpy | scipy* |'
print '---------------------------------------------------'
for size, repeat in [
((100, 100), 100),
((1000, 100), 7),
((256, 256), 10),
((512, 512), 3),
]:
chk = checkerboard(*size).astype('f')
print '%9s' % ('%sx%s' % size),
sys.stdout.flush()
x = random(repeat, *size).astype('F') + \
random(repeat, *size).astype('F')*1j
tr0 = time.time()
y = fft2(x, shift=True)
trf = time.time()
print '|%8.2f' % (trf - tr0),
sys.stdout.flush()
tn0 = time.time()
ny = chk * numpy_fftn(chk * x, axes=(-2, -1))
tnf = time.time()
## assert_array_almost_equal(ny,y, decimal=2)
print '|%8.2f' % (tnf - tn0),
sys.stdout.flush()
ts0 = time.time()
sy = chk * (scipy_fftn(
(chk * x).astype('D'), axes=(-2, -1)).astype('F'))
tsf = time.time()
## assert_array_almost_equal(sy,y, decimal=2)
print '|%8.2f' % (tsf - ts0),
sys.stdout.flush()
print ' (secs for %s calls)' % (repeat)
print "(* casted float->FT{double}->float)"
sys.stdout.flush()
assert True, 'asdf'
def bench_fft2_time():
from numpy.fft import fftn as numpy_fftn
from scipy.fftpack import fftn as scipy_fftn
print
print ' 2D Double Precision Fast Fourier Transform'
print '==================================================='
print ' | complex input '
print '---------------------------------------------------'
print ' size | fftw | numpy | scipy |'
print '---------------------------------------------------'
for size,repeat in [((100,100),100),((1000,100),7),
((256,256),10),
((512,512),3),
]:
print '%9s' % ('%sx%s'%size),
sys.stdout.flush()
x = random(repeat, *size).astype('D') + \
random(repeat, *size).astype('D')*1j
tr0 = time.time()
y = fft2(x, shift=False)
trf = time.time()
print '|%8.2f' % (trf-tr0),
sys.stdout.flush()
tn0 = time.time()
ny = numpy_fftn(x, axes=(-2,-1))
tnf = time.time()
## assert_array_almost_equal(ny,y)
print '|%8.2f' % (tnf-tn0),
sys.stdout.flush()
ts0 = time.time()
sy = scipy_fftn(x, axes=(-2,-1))
tsf = time.time()
## assert_array_almost_equal(sy,y)
print '|%8.2f' % (tsf-ts0),
sys.stdout.flush()
print ' (secs for %s calls)' % (repeat)
sys.stdout.flush()
print
print ' 2D Double Precision Fast Fourier Transform'
print '==================================================='
print ' | complex input shifted '
print '---------------------------------------------------'
print ' size | fftw | numpy | scipy |'
print '---------------------------------------------------'
for size,repeat in [((100,100),100),((1000,100),7),
((256,256),10),
((512,512),3),
]:
chk = checkerboard(*size).astype('d')
print '%9s' % ('%sx%s'%size),
sys.stdout.flush()
x = random(repeat, *size).astype('D') + \
random(repeat, *size).astype('D')*1j
tr0 = time.time()
y = fft2(x, shift=True)
trf = time.time()
print '|%8.2f' % (trf-tr0),
sys.stdout.flush()
tn0 = time.time()
ny = numpy_fftn(x, axes=(-2,-1))
tnf = time.time()
## assert_array_almost_equal(ny,y)
print '|%8.2f' % (tnf-tn0),
sys.stdout.flush()
ts0 = time.time()
sy = scipy_fftn(x, axes=(-2,-1))
tsf = time.time()
## assert_array_almost_equal(sy,y)
print '|%8.2f' % (tsf-ts0),
sys.stdout.flush()
print ' (secs for %s calls)' % (repeat)
sys.stdout.flush()
print
print ' 2D Single Precision Fast Fourier Transform'
print '==================================================='
print ' | complex input '
print '---------------------------------------------------'
print ' size | fftw | numpy | scipy* |'
print '---------------------------------------------------'
for size,repeat in [((100,100),100),((1000,100),7),
((256,256),10),
((512,512),3),
]:
print '%9s' % ('%sx%s'%size),
sys.stdout.flush()
x = random(repeat, *size).astype('F') + \
random(repeat, *size).astype('F')*1j
tr0 = time.time()
y = fft2(x, shift=False)
trf = time.time()
print '|%8.2f' % (trf-tr0),
sys.stdout.flush()
tn0 = time.time()
ny = numpy_fftn(x, axes=(-2,-1))
tnf = time.time()
## assert_array_almost_equal(ny,y, decimal=2)
print '|%8.2f' % (tnf-tn0),
sys.stdout.flush()
ts0 = time.time()
sy = scipy_fftn(x.astype('D'), axes=(-2,-1)).astype('F')
tsf = time.time()
## assert_array_almost_equal(sy,y, decimal=2)
print '|%8.2f' % (tsf-ts0),
sys.stdout.flush()
print ' (secs for %s calls)' % (repeat)
print "(* casted float->FT{double}->float)"
sys.stdout.flush()
print
print ' 2D Single Precision Fast Fourier Transform'
print '==================================================='
print ' | complex input shifted '
print '---------------------------------------------------'
print ' size | fftw | numpy | scipy* |'
print '---------------------------------------------------'
for size,repeat in [((100,100),100),((1000,100),7),
((256,256),10),
((512,512),3),
]:
chk = checkerboard(*size).astype('f')
print '%9s' % ('%sx%s'%size),
sys.stdout.flush()
x = random(repeat, *size).astype('F') + \
random(repeat, *size).astype('F')*1j
tr0 = time.time()
y = fft2(x, shift=True)
trf = time.time()
print '|%8.2f' % (trf-tr0),
sys.stdout.flush()
tn0 = time.time()
ny = chk*numpy_fftn(chk*x, axes=(-2,-1))
tnf = time.time()
## assert_array_almost_equal(ny,y, decimal=2)
print '|%8.2f' % (tnf-tn0),
sys.stdout.flush()
ts0 = time.time()
sy = chk*(scipy_fftn((chk*x).astype('D'), axes=(-2,-1)).astype('F'))
tsf = time.time()
## assert_array_almost_equal(sy,y, decimal=2)
print '|%8.2f' % (tsf-ts0),
sys.stdout.flush()
print ' (secs for %s calls)' % (repeat)
print "(* casted float->FT{double}->float)"
sys.stdout.flush()
assert True, 'asdf'
def test_fails_for_odd_1():
a = np.empty((3, 4, 5)).astype('D')
fft2(a, axes=(0, 1))
def test_fails_for_odd_1():
a = np.empty((3,4,5)).astype('D')
fft2(a, axes=(0,1))
