U
|
P ±d3 ã @ sô d dl Zd dlZd dlmZ d dlmZmZmZm Z d dl
|
Z
|
d dl Z dd Z G dd dZ G dd d Zej d
|
ejejejejg¡ej d d d g¡ej dejjejjejjejjejjejjg¡dd Zejj e ddG dd dZ dS )é N)Úrandom)Úassert_array_equalÚ assert_raisesÚassert_allcloseÚIS_WASMc C sN t | }dtj t |¡| }t |¡ dd¡| }tj| t |¡ ddS )Ny Àéÿÿÿÿé ©Úaxis)ÚlenÚnpÚpiZarangeZreshapeÚsumÚexp)ÚxÚLÚphase© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/fft/tests/test_pocketfft.pyÚfft1 s r c @ s e Zd Zdd ZdS )Ú TestFFTShiftc C s t ttjjdddgd d S )Nr é é r )r Ú
|
ValueErrorr Úfft)Úselfr r r Ú
|
test_fft_n s zTestFFTShift.test_fft_nN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r r s r c @ sî e Zd Zdd Zdd Zej dd¡dd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zd d Zej d!ejjejjejjejjg¡d"d# Zd$d% Z ej d&ej ej ej ej g¡d'd( Z!d)S )*Ú TestFFT1Dc C s d}t |dt | }t |}td|D ]b}ttj tj |d|
¡¡|d|
dd ttj tj |d|
¡|¡|d|
dd q*d S )Ni ù ð?r r gê-q=©Úatol)r Úranger r r ÚifftÚirfftÚrfft)r Úmaxlenr ZxrÚir r r Ú test_identity s &ÿ
|
ÿzTestFFT1D.test_identityc C s t ddt d }tt|tj |¡dd tt|tjj|dddd tt|t d¡ tjj|dddd tt|d tjj|d ddd d S )
|
Né r! çíµ ÷Æ°>r" Úbackward©ÚnormÚorthoç >@Úforward)r r r r r Úsqrt©r r r r r Útest_fft$ s ÿ ÿzTestFFT1D.test_fftr/ ©Nr- r0 r2 c C sh t ddt d }t|tjjtjj||d|ddd tjtdd tjjg |d W 5 Q R X d S )Nr+ r! r. r, r" z!Invalid number of FFT data points)Úmatch)r r r r r% ÚpytestZraisesr )r r/ r r r r Ú test_ifft- s þÿzTestFFT1D.test_ifftc C sº t ddt d }ttjjtjj|ddddtj |¡dd ttj |¡tjj|dd dd ttj |¡t d
|
¡ tjj|d d dd ttj |¡d tjj|d d dd d S ©N©r+ é r! r r r r, r" r- r. éX r0 ç À@r2 )r r r r Úfft2r3 r4 r r r Ú test_fft28 s"
|
ÿ ÿ ÿ ÿzTestFFT1D.test_fft2c C sº t ddt d }ttjjtjj|ddddtj |¡dd ttj |¡tjj|dd dd ttj |¡t d
|
¡ tjj|d d dd ttj |¡d tjj|d d dd d S r: )r r r r r% Úifft2r3 r4 r r r Ú
|
test_ifft2C s"
|
ÿ ÿ ÿ ÿzTestFFT1D.test_ifft2c C sÆ t ddt d }ttjjtjjtjj|ddddddtj |¡dd ttj |¡tjj|d d
|
dd ttj |¡t d ¡ tjj|d d
|
dd ttj |¡d tjj|dd
|
dd d S ©N©r+ r< é
|
r! r r r r r, r" r- r. ép r0 ç p·@r2 )r r r r Úfftnr3 r4 r r r Ú test_fftnN s$ &
|
þ ÿ ÿ ÿzTestFFT1D.test_fftnc C sÆ t ddt d }ttjjtjjtjj|ddddddtj |¡dd ttj |¡tjj|d d
|
dd ttj |¡t d ¡ tjj|d d
|
dd ttj |¡d tjj|dd
|
dd d S rC )r r r r r% Úifftnr3 r4 r r r Ú
|
test_ifftnZ s$ &
|
þ ÿ ÿ ÿzTestFFT1D.test_ifftnc C sì t d}|jd|j fD ]Î}dD ]>}ttjj|||dd |d d
tjj|||ddd q ttjj||dtjj||d ddd ttjj||dt |¡ tjj||d
|
ddd ttjj||d| tjj||d ddd qd S ) Nr+ r r6 ©Únr/ r r, r" )rM r- r0 r2 )r Úsizer r r r' r3 )r r rM r/ r r r Ú test_rfftf s. þ þ þ þzTestFFT1D.test_rfftc C s t d}t|tj tj |¡¡dd t|tjjtjj|dddddd t|tjjtjj|dddddd t|tjjtjj|dddddd d S )Nr+ r, r" r- r. r0 r2 )r r r r r& r' r4 r r r Ú
|
test_irfftw s" ÿÿÿÿÿÿzTestFFT1D.test_irfftc C s® t d}ttj |¡d d
d d
f tj |¡dd ttj |¡tjj|dddd ttj |¡t d¡ tjj|dddd ttj |¡d tjj|d
|
ddd d S ) Nr; é r, r" r- r. r= r0 r> r2 )r r r r r? Úrfft2r3 r4 r r r Ú
|
test_rfft2 s . ÿ ÿ ÿzTestFFT1D.test_rfft2c C s t d}t|tj tj |¡¡dd t|tjjtjj|dddddd t|tjjtjj|dddddd t|tjjtjj|dddddd d S )Nr; r, r" r- r. r0 r2 )r r r r Zirfft2rR r4 r r r Ú test_irfft2 s" ÿÿÿÿÿÿzTestFFT1D.test_irfft2c C s´ t d}ttj |¡d d
d d
d d
f tj |¡dd ttj |¡tjj|dddd ttj |¡t d¡ tjj|dddd ttj |¡d tjj|d
|
ddd d S ) NrD é r, r" r- r. rF r0 rG r2 )r r r r rH Úrfftnr3 r4 r r r Ú
|
test_rfftn s 4 ÿ ÿ ÿzTestFFT1D.test_rfftnc C s t d}t|tj tj |¡¡dd t|tjjtjj|dddddd t|tjjtjj|dddddd t|tjjtjj|dddddd d S )NrD r, r" r- r. r0 r2 )r r r r ÚirfftnrV r4 r r r Ú test_irfftn s" ÿÿÿÿÿÿzTestFFT1D.test_irfftnc C sÞ t ddt d }t t d|t df¡}t ||d d d
¡ f¡}ttj |¡tj |¡dd ttj |¡tjj|dddd ttj |¡t d ¡ tjj|d
|
ddd ttj |¡d tjj|d ddd d S ) Né r! r r r, r" r- r. r+ r0 r1 r2 )r r Ú concatenateÚconjr r Úhfftr3 ©r r Zx_hermr r r Ú test_hfft© s ÿ ÿ ÿzTestFFT1D.test_hfftc C sÜ t ddt d }t t d|t df¡}t ||d d d
¡ f¡}t|tj tj |¡¡dd t|tjjtjj|dddddd t|tjjtjj|d dd ddd t|tjjtjj|d
|
dd
|
ddd d S ) NrZ r! r r r, r" r- r. r0 r2 )r r r[ r\ r r Úihfftr] r^ r r r Ú
|
test_ihfftµ s2 ÿÿÿÿÿÿÿÿÿzTestFFT1D.test_ihfftÚopc C sX t d}ddddddg}|D ]6}|t ||¡}t |||d|¡}t||d d
|
q d S ) NrD )r r r )r r r )r r r )r r r )r r r )r r r ©Úaxesr, r" )r r Z transposer )r rb r rd ÚaZop_trZtr_opr r r Ú test_axesÁ s zTestFFT1D.test_axesc C s° t d}tj |¡}|jd }tjjtjjftjjtjjftjj tjj
|
fg}|D ]Z\}}|jd|j fD ]@}dD ]6}||||d}||||d}t |tj |¡dd qpqhqPd S )Nr+ r r6 rL r, r" ) r r Zlinalgr/ rN r r% r' r& r` r] r ) r r Zx_normrM Z
|
func_pairsZforwÚbackr/ Útmpr r r Útest_all_1d_norm_preservingË s
|
ü
|
ÿz%TestFFT1D.test_all_1d_norm_preservingÚdtypec C sN t d |¡}ttj tj |¡¡|dd ttj tj |¡¡|dd d S )Nr+ r, r" )r Úastyper r r r% r& r' )r rj r r r r Ú test_dtypesÞ s zTestFFT1D.test_dtypesN)"r r r r* r5 r8 ÚmarkÚ parametrizer9 r@ rB rI rK rO rP rS rT rW rY r_ ra r r rH rJ rV rX rf ri ZhalfZsingleÚdoubleZ
|
longdoublerl r r r r r s4
|
|
|
|
|
|
|
|
ÿ
|
ÿr rj ÚorderÚFznon-contiguousr c C s6 t j d¡}| ddd¡j| dd}dt t |j¡¡ t |j ¡j
|
}|dkr\t |¡}n"|d d d
}t |d d d
¡}|j d
|
¡rÂtd D ],}|||d }|||d } t|| ||d qnp|j d¡r,dddg}
|
|j d¡rø|
|
dddd g¡ |
|
D ],} ||| d}||| d} t|| ||d qünt d S )Né* é é é F)Úcopyg @rq r r r r )r# Zrtol)r? rH )r r )r r )r r rH )r )r )r rc )r r Z RandomStateZrandrk r3 Úlog2rN Zfinforj ZepsZasfortranarrayZascontiguousarrayr Úendswithr$ r Úextendr ) rj rp r ÚrngÚXZ_tolÚYr
|
ZX_resZY_resrd Zaxr r r Útest_fft_with_orderè s* $
|
r} zCannot start thread)Úreasonc @ s< e Zd ZdZdZdd Zdd Zdd Zd d
|
Zd d Z d S )ÚTestFFTThreadSafeé )i éÈ c s| fddt ¡ } fddt| jD }dd |D dd |D t| jD ]}tjdd|d q^d S )
|
Nc s | | ¡ d S ©N)Úput)ÚargsÚq)Úfuncr r Úworker s z.TestFFTThreadSafe._test_mtsame.<locals>.workerc s g | ]}t j fd qS ))Útargetr )Ú threadingÚThread)Ú.0r) )r r
r r r Ú
|
<listcomp> s ÿz2TestFFTThreadSafe._test_mtsame.<locals>.<listcomp>c S s g | ] }| ¡ qS r )Ústart©r r r r r r s c S s g | ] }| ¡ qS r )Újoinr r r r r s é )Útimeoutz6Function returned wrong value in multithreaded context)ÚqueueÚQueuer$ Úthreadsr Úget)r r r ÚexpectedÚtr) r )r r r
r r Ú _test_mtsame s ÿÿz TestFFTThreadSafe._test_mtsamec C s( t | j¡d d }| t jj|¡ d S ©Nr y )r ÚonesÚ input_shaper r ©r re r r r r5 & s zTestFFTThreadSafe.test_fftc C s( t | j¡d d }| t jj|¡ d S r )r r r r r r% r r r r r9 * s zTestFFTThreadSafe.test_ifftc C s t | j¡}| t jj|¡ d S r )r r r r r r' r r r r rO . s zTestFFTThreadSafe.test_rfftc C s( t | j¡d d }| t jj|¡ d S r )r r r r r r& r r r r rP 2 s z TestFFTThreadSafe.test_irfftN)
|
r r r r r r r5 r9 rO rP r r r r r s r ) Únumpyr r8 Z numpy.randomr Z numpy.testingr r r r r r r r r rm rn Úfloat32Úfloat64Z complex64Z
|
complex128r r? rH r% rA rJ r} Zskipifr r r r r Ú<module> s2 Qþ ÿþ
|
