Coverage for /usr/lib/python3/dist-packages/numpy/dual.py : 80%

""" 

Aliases for functions which may be accelerated by Scipy. 

Scipy_ can be built to use accelerated or otherwise improved libraries 

for FFTs, linear algebra, and special functions. This module allows 

developers to transparently support these accelerated functions when 

scipy is available but still support users who have only installed 

NumPy. 

.. _Scipy : https://www.scipy.org 

""" 

from __future__ import division, absolute_import, print_function 

# 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 --- from numpy.dual import fft, inv 

__all__ = ['fft', 'ifft', 'fftn', 'ifftn', 'fft2', 'ifft2', 

'norm', 'inv', 'svd', 'solve', 'det', 'eig', 'eigvals', 

'eigh', 'eigvalsh', 'lstsq', 'pinv', 'cholesky', 'i0'] 

import numpy.linalg as linpkg 

import numpy.fft as fftpkg 

from numpy.lib import i0 

import sys 

fft = fftpkg.fft 

ifft = fftpkg.ifft 

fftn = fftpkg.fftn 

ifftn = fftpkg.ifftn 

fft2 = fftpkg.fft2 

ifft2 = fftpkg.ifft2 

norm = linpkg.norm 

inv = linpkg.inv 

svd = linpkg.svd 

solve = linpkg.solve 

det = linpkg.det 

eig = linpkg.eig 

eigvals = linpkg.eigvals 

eigh = linpkg.eigh 

eigvalsh = linpkg.eigvalsh 

lstsq = linpkg.lstsq 

pinv = linpkg.pinv 

cholesky = linpkg.cholesky 

_restore_dict = {} 

def register_func(name, func): 

if name not in __all__: 

raise ValueError("%s not a dual function." % name) 

f = sys._getframe(0).f_globals 

_restore_dict[name] = f[name] 

f[name] = func 

def restore_func(name): 

if name not in __all__: 

raise ValueError("%s not a dual function." % name) 

try: 

val = _restore_dict[name] 

except KeyError: 

return 

else: 

sys._getframe(0).f_globals[name] = val 

def restore_all(): 

for name in _restore_dict.keys(): 

restore_func(name)