summaryrefslogtreecommitdiff
path: root/numpy/dual.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/dual.py')
-rw-r--r--numpy/dual.py37
1 files changed, 37 insertions, 0 deletions
diff --git a/numpy/dual.py b/numpy/dual.py
new file mode 100644
index 000000000..44b7b51c4
--- /dev/null
+++ b/numpy/dual.py
@@ -0,0 +1,37 @@
+# This module should be used for functions both in numpy and scipy if
+# you want to use the numpy version if available but the scipy version
+# otherwise.
+# Usage --- import numpy.dual
+
+__all__ = ['fft','ifft','fftn','ifftn','fft2','ifft2',
+ 'inv','svd','solve','det','eig','eigvals','lstsq',
+ 'pinv','cholesky']
+
+try:
+ import scipy.linalg as linpkg
+except ImportError:
+ import numpy.linalg as linpkg
+
+try:
+ import scipy.fftpack as fftpkg
+except ImportError:
+ import numpy.dft as fftpkg
+
+
+fft = fftpkg.fft
+ifft = fftpkg.ifft
+fftn = fftpkg.fftn
+ifftn = fftpkg.ifftn
+fft2 = fftpkg.fft2
+ifft2 = fftpkg.ifft2
+
+inv = linpkg.inv
+svd = linpkg.svd
+solve = linpkg.solve
+det = linpkg.det
+eig = linpkg.eig
+eigvals = linpkg.eigvals
+lstsq = linpkg.lstsq
+pinv = linpkg.pinv
+cholesky = linpkg.cholesky
+