summaryrefslogtreecommitdiff
path: root/numpy/fft/fftpack.py
diff options
context:
space:
mode:
authorJarrod Millman <millman@berkeley.edu>2007-11-26 05:08:55 +0000
committerJarrod Millman <millman@berkeley.edu>2007-11-26 05:08:55 +0000
commit5eabfa765d5aa1d33fd2e8704dfb4bfbdb25761b (patch)
tree4e8075b20ffc4342589c3e61fd1e19013f7ff58f /numpy/fft/fftpack.py
parent6daa0bbd1cea9c8134e6d67464843e44f35b30df (diff)
downloadnumpy-5eabfa765d5aa1d33fd2e8704dfb4bfbdb25761b.tar.gz
using identity testing for None
Diffstat (limited to 'numpy/fft/fftpack.py')
-rw-r--r--numpy/fft/fftpack.py18
1 files changed, 10 insertions, 8 deletions
diff --git a/numpy/fft/fftpack.py b/numpy/fft/fftpack.py
index 7918ea3e5..6a0d3ab05 100644
--- a/numpy/fft/fftpack.py
+++ b/numpy/fft/fftpack.py
@@ -35,9 +35,11 @@ def _raw_fft(a, n=None, axis=-1, init_function=fftpack.cffti,
work_function=fftpack.cfftf, fft_cache = _fft_cache ):
a = asarray(a)
- if n == None: n = a.shape[axis]
+ if n is None:
+ n = a.shape[axis]
- if n < 1: raise ValueError("Invalid number of FFT data points (%d) specified." % n)
+ if n < 1:
+ raise ValueError("Invalid number of FFT data points (%d) specified." % n)
try:
wsave = fft_cache[n]
@@ -110,7 +112,7 @@ def ifft(a, n=None, axis=-1):
different n's."""
a = asarray(a).astype(complex)
- if n == None:
+ if n is None:
n = shape(a)[axis]
return _raw_fft(a, n, axis, fftpack.cffti, fftpack.cfftb, _fft_cache) / n
@@ -155,7 +157,7 @@ def irfft(a, n=None, axis=-1):
within numerical accuracy."""
a = asarray(a).astype(complex)
- if n == None:
+ if n is None:
n = (shape(a)[axis] - 1) * 2
return _raw_fft(a, n, axis, fftpack.rffti, fftpack.rfftb,
_real_fft_cache) / n
@@ -174,7 +176,7 @@ def hfft(a, n=None, axis=-1):
within numerical accuracy."""
a = asarray(a).astype(complex)
- if n == None:
+ if n is None:
n = (shape(a)[axis] - 1) * 2
return irfft(conjugate(a), n, axis) * n
@@ -192,7 +194,7 @@ def ihfft(a, n=None, axis=-1):
within numerical accuracy."""
a = asarray(a).astype(float)
- if n == None:
+ if n is None:
n = shape(a)[axis]
return conjugate(rfft(a, n, axis))/n
@@ -200,14 +202,14 @@ def ihfft(a, n=None, axis=-1):
def _cook_nd_args(a, s=None, axes=None, invreal=0):
if s is None:
shapeless = 1
- if axes == None:
+ if axes is None:
s = list(a.shape)
else:
s = take(a.shape, axes)
else:
shapeless = 0
s = list(s)
- if axes == None:
+ if axes is None:
axes = range(-len(s), 0)
if len(s) != len(axes):
raise ValueError, "Shape and axes have different lengths."