mirror/kissfft
1
0
Fork 0
mirror of https://github.com/mborgerding/kissfft.git synced 2025-05-27 21:20:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

more modern python

Browse Source
This commit is contained in:
Mark Borgerding 2019-02-01 07:38:42 -05:00
parent e931ae3d4b
commit a7ea65edb1
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
test/testkiss.py
View File

@ -4,7 +4,7 @@
#
# SPDX-License-Identifier: BSD-3-Clause
# See COPYING file for more information.
from __future__ import division,print_function
import math
import sys
import os
@ -50,7 +50,7 @@ def dopack(x,cpx=1):
return s
def dounpack(x,cpx):
uf = fmt * ( len(x) / struct.calcsize(fmt) )
uf = fmt * ( len(x) // struct.calcsize(fmt) )
s = struct.unpack(uf,x)
if cpx:
return numpy.array(s[::2]) + numpy.array( s[1::2] )*1j
@ -99,12 +99,12 @@ def test_fft(ndims):
errpow = numpy.vdot(errf,errf)+1e-10
sigpow = numpy.vdot(xverf,xverf)+1e-10
snr = 10*math.log10(abs(sigpow/errpow) )
print 'SNR (compared to NumPy) : %.1fdB' % float(snr)
print( 'SNR (compared to NumPy) : {0:.1f}dB'.format( float(snr) ) )
if snr<minsnr:
print 'xver=',xver
print 'x2=',x2
print 'err',err
print( 'xver=',xver )
print( 'x2=',x2)
print( 'err',err)
sys.exit(1)
def dofft(x,isreal):
@ -124,7 +124,7 @@ def dofft(x,isreal):
if doreal:
cmd += ' -R '
print cmd
print( cmd)
from subprocess import Popen,PIPE
p = Popen(cmd,shell=True,stdin=PIPE,stdout=PIPE )
@ -149,9 +149,9 @@ def main():
doreal = opts.has_key('-r')
if doreal:
print 'Testing multi-dimensional real FFTs'
print( 'Testing multi-dimensional real FFTs')
else:
print 'Testing multi-dimensional FFTs'
print( 'Testing multi-dimensional FFTs')
for dim in range(1,4):
test_fft( dim )