| Current File : //usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyc |
�
E�`Qc @ s� d Z d d l m Z d d l Z d d l m Z m Z m Z d d l m Z d d l m
Z
d e f d � � YZ d e f d
� � YZ d e f d � � YZ
e d
k r� e � n d S( s* Test functions for fftpack.helper module
i����( t divisionN( t TestCaset run_module_suitet assert_array_almost_equal( t fft( t pit TestFFTShiftc B s# e Z d � Z d � Z d � Z RS( c
C s� d d d d d d d d d g } d d d d d d d d d g } t t j | � | � t t j | � | � d d d d d d
d d d d g
} d
d d d d d d d d d g
} t t j | � | � t t j | � | � d S( Ni i i i i i����i����i����i����i����( R R t fftshiftt ifftshift( t selft xt y( ( sA /usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyt test_definition s !!$$c C sX xQ d d d d d g D]: } t j j | f � } t t j t j | � � | � q Wd S( Ni i i id i� ( t npt randomR R R R ( R t nR
( ( sA /usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyt test_inverse s c C s� d d d g d d d g d d d g g } d d d g d d d g d d d g g } t t j | d
d �| � t t j | d
d �t j | d
d �� t t j | d
d
�| � t t j | d
d �t j | d
d �� d S( Ni i i i i i����i����i����i����t axes( i i ( i ( i i ( i ( R R R R ( R t freqst shifted( ( sA /usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyt test_axes_keyword s ** ( t __name__t
__module__R R R ( ( ( sA /usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyR
s
t TestFFTFreqc B s e Z d � Z RS( c
C s� d d d d d d d d d g } t d
t j d
� | � t d
t t j d
t � | � d d d d d d d d d d g
} t d t j d � | � t d t t j d t � | � d S(
Ni i i i i i����i����i����i����i i����i
( R R t fftfreqR ( R R
( ( sA /usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyR + s !!$( R R R ( ( ( sA /usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyR ) s t
TestIRFFTNc B s e Z d � Z RS( c C sC t j j d � \ } } | d | } d } t j | d | �d S(
Ni i i i y �?i����R ( i i i i ( i����( R
R R t irfftn( R t art ait aR ( ( sA /usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyt test_not_last_axis_success6 s ( R R R ( ( ( sA /usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyR 4 s t __main__( t __doc__t
__future__R t numpyR
t
numpy.testingR R R R R R R R R ( ( ( sA /usr/lib64/python2.7/site-packages/numpy/fft/tests/test_helper.pyt <module> s