Current File : //opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyo
�
� 9Zc @` s d d l m Z m Z m Z d d l Z d d l m Z d d l m Z m
Z
m Z d d l m Z d d l
Z
d d l Z e j d d k r� d d l Z n d d l Z d � Z d e f d � � YZ d
e f d � � YZ d e f d
� � YZ e d k r e
� n d S( i ( t divisiont absolute_importt print_functionN( t random( t TestCaset run_module_suitet assert_array_almost_equal( t assert_array_equali c C` so t | � } d t j t j | � t | � } t j | � j d d � | } t j | t j | � d d �S( Ny �i����i t axis( t lent npt pit aranget floatt reshapet sumt exp( t xt Lt phase( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt fft1 s $ t TestFFTShiftc B` s e Z d � Z RS( c C` s) | j t t j j d d d g d � d S( Ni i i i ( t assertRaisest
ValueErrorR
t fft( t self( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt
test_fft_n s ( t __name__t
__module__R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyR s t TestFFT1Dc B` s� e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z d
� Z d � Z
d � Z d
� Z RS( c C` so t d � d t d � } t t | � t j j | � � t t | � t j d � t j j | d d �� d S( Ni y �?t normt ortho( R R R R
R t sqrt( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt test_fft s c C` st t d � d t d � } t | t j j t j j | � � � t | t j j t j j | d d �d d �� d S( Ni y �?R R ( R R R
R t ifft( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt test_ifft$ s %c C` s� t d � d t d
� } t t j j t j j | d d �d d �t j j | � � t t j j | � t j d � t j j | d d �� d S( Ni i y �?R i i R R ( i i ( i i iX ( R R R
R t fft2R ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt test_fft2* s
* c C` s� t d � d t d
� } t t j j t j j | d d �d d �t j j | � � t t j j | � t j d � t j j | d d �� d S( Ni i y �?R i i R R ( i i ( i i iX ( R R R
R R" t ifft2R ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt
test_ifft21 s
* c C` s� t d � d t d � } t t j j t j j t j j | d d �d d �d d �t j j | � � t t j j | � t j d � t j j | d d
�� d S( Ni i i
y �?R i i i R R ( i i i
( i i i
iX ip ( R R R
R t fftnR ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt test_fftn8 s 9 c C` s� t d � d t d � } t t j j t j j t j j | d d �d d �d d �t j j | � � t t j j | � t j d � t j j | d d
�� d S( Ni i i
y �?R i i i R R ( i i i
( i i i
iX ip ( R R R
R R" t ifftnR ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt
test_ifftn@ s 9 c C` sq t d � } t t j j | � d t j j | � � t t j j | � t j d � t j j | d d �� d S( Ni i R R ( R R R
R t rfftR ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt test_rfftH s ) c C` sf t d � } t | t j j t j j | � � � t | t j j t j j | d d �d d �� d S( Ni R R ( R R R
R t irfftR, ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt
test_irfftN s %c C` s� t d � } t t j j | � d d � d d � f t j j | � � t t j j | � t j d � t j j | d d �� d S( Ni i i R R ( i i iX ( R R R
R R$ t rfft2R ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt
test_rfft2T s ; c C` sf t d � } t | t j j t j j | � � � t | t j j t j j | d d �d d �� d S( Ni i R R ( i i ( R R R
R t irfft2R0 ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt test_irfft2Z s %c C` s� t d � } t t j j | � d d � d d � d d � f t j j | � � t t j j | � t j d � t j j | d d �� d S(
Ni i i
i R R ( i i i
iX ip ( R R R
R R( t rfftnR ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt
test_rfftn` s D c C` sf t d � } t | t j j t j j | � � � t | t j j t j j | d d �d d �� d S( Ni i i
R R ( i i i
( R R R
R t irfftnR4 ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt test_irfftnf s %c C` s� t d � d t d � } t j t d � | t d � f � } t j | | d d d � j � f � } t t j j | � t j j | � � t t j j | � t j d � t j j | d d �� d S( Ni y �?i i����i R R ( R R
t concatenatet conjR R t hfftR ( R R t x_herm( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt test_hfftl s $(% c C` s� t d � d t d � } t j t d � | t d � f � } t j | | d d d � j � f � } t | t j j t j j | � � � t | t j j t j j | d d �d d �� d S( Ni y �?i i����R R ( R R
R8 R9 R R t ihfftR: ( R R R; ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt
test_ihttft s $(%$( R R R! R# R% R' R) R+ R- R/ R1 R3 R5 R7 R< R> ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyR s t TestFFTThreadSafec B` sA e Z d Z d Z d � Z d � Z d � Z d � Z d � Z RS( i i i� c ` s� � f d � } t j � } � | � } g t | j � D]$ } t j d | d | | f � ^ q7 } g | D] } | j � ^ qh g | D] } | j � ^ q� x3 t | j � D]" } t | j d d � | d � q� Wd S( Nc ` s | j � | � � d S( N( t put( t argst q( t func( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt worker� s t targetRA t timeouti s6 Function returned wrong value in multithreaded context(
t queuet Queuet ranget threadst threadingt Threadt startt joinR t get( R RC RA RD RB t expectedt it tR ( ( RC sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt _test_mtsame� s 7 c C` s4 t j | j � d d } | j t j j | � d S( Ni y ( R
t onest input_shapeRS R ( R t a( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyR! � s c C` s4 t j | j � d d } | j t j j | � d S( Ni y ( R
RT RU RS R R" ( R RV ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyR# � s c C` s, t j | j � } | j t j j | � d S( N( R
RT RU RS R R, ( R RV ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyR- � s c C` s4 t j | j � d d } | j t j j | � d S( Ni y ( R
RT RU RS R R. ( R RV ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyR/ � s ( i i� ( R R RJ RU RS R! R# R- R/ ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyR? ~ s t __main__( t
__future__R R R t numpyR
t numpy.randomR t
numpy.testingR R R R RK t syst version_infoRG RH R R R R? R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/fft/tests/test_fftpack.pyt <module> s b'
Mini Shell