f d dl Zd dlZd dlmZ d dlmZmZmZm Z d dl
Z
d dl Z d Z G d d Z
G d d Zej d ej ej ej ej g ej d
d d g ej d
ej j ej j ej j ej j ej j ej j g d Zej e d G d d Z dS ) N)random)assert_array_equal
assert_raisesassert_allcloseIS_WASMc t | }dt j z t j | |z z }t j | dd |z }t j | t j | z d S )Ny axis)lennppiarangereshapesumexp)xLphases r/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/fft/tests/test_pocketfft.pyfft1r sm AA "%K29Q<r+ backwardnormortho >@forward)r r r r r sqrtr! r s r test_fftzTestFFT1D.test_fft$ s 2JJF2JJ &QA T::::QAJ!?!?dKKKKQ"'"++-
17
33$ @ @ @ @Q#
19
55D B B B B B Br r: Nr8 r; r= c t d dt d z z }t |t j t j || | d t j t d 5 t j g | d d d d S # 1 swxY w Y d S )Nr6 r* r9 r7 r+ z!Invalid number of FFT data points)match)r r r r r. pytestraisesr )r! r: r s r test_ifftzTestFFT1D.test_ifft- s 2JJF2JJ &
rv{{26::ad:33${?? ]:!DF F F ' ' FKKK & & & ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's "B==CCc ( t d dt d z z }t t j t j |d d t j | d t t j | t j |d d t t j | t j d
z t j |d d t t j | d z t j |d
d d S Nr6 r* r
r r r7 r+ r8 r9 X r; @r= )r r r r fft2r> r? s r test_fft2zTestFFT1D.test_fft28 sO 8 r&"2"2 2 2
26::aa:#8#8q
AA AT 3 3 3 3 A AJ 77d D D D D A)9)99 AG 444 A A A A A)4 AI 66T C C C C C Cr c ( t d dt d z z }t t j t j |d d t j | d t t j | t j |d d t t j | t j d
z t j |d d t t j | d z t j |d
d d S rH )r r r r r. ifft2r> r? s r
test_ifft2zTestFFT1D.test_ifft2C sO 8 r&"2"2 2 2 BFKKK$:$: CC Qd 4 4 4 4 Q QZ 88t E E E E Q"''*:*:: QW 55D B B B B Q95 QY 77d D D D D D Dr c
f t d dt d z z }t t j t j t j |d d d t j | d t t j | t j |d
d t t j | t j d z t j |d
d t t j | d
z t j |d
d d S Nr6 rJ
r* r r r
r r7 r+ r8 r9 p r; p@r= )r r r r fftnr> r? s r test_fftnzTestFFT1D.test_fftnN sd < 2f\&:&:#: : FJJrvzz"&**QQ*"7"7az@@qJ I I FKKNN ' ' ' ' A AJ 77d D D D D A )>)>> AG 444 A A A A A/: AI 66T C C C C C Cr c
f t d dt d z z }t t j t j t j |d d d t j | d t t j | t j |d
d t t j | t j d z t j |d
d t t j | d
z t j |d
d d S rS )r r r r r. ifftnr> r? s r
test_ifftnzTestFFT1D.test_ifftnZ sd < 2f\&:&:#: : FKK BFKKK$:$: CC!K L L FLLOO$ ( ( ( ( Q QZ 88t E E E E Q"',*?*?? QW 55D B B B B Q?; QY 77d D D D D D Dr c
0 t d }|j d|j z fD ]s}dD ]b}t t j ||| d |dz dz t j ||| d ct t j || t j ||d d t t j || t j | z t j ||d
d t t j || |z t j ||d d ud S ) Nr6 r rA nr: r
r7 r+ )r_ r8 r; r= )r sizer r r r0 r> )r! r r_ r: s r test_rfftzTestFFT1D.test_rfftf s 2JJ&!AF(#
@
@A>
?
? FJJqADJ11+AqD1H+>FKKQTK22? ? ? ? ?
A ## A 444
A
A
A
A
A ##bgajj0 A 11
>
>
>
>
A ##a' A 33$
@
@
@
@
@
@
@r c t d }t |t j t j | d t |t j t j |d d d t |t j t j |d d d t |t j t j |d d d d S )Nr6 r7 r+ r8 r9 r; r= )r r r r r/ r0 r? s r
test_irfftzTestFFT1D.test_irfftw s$ 2JJ26<< A77dCCCC26<< AJ (G(G ' ( ) )/3 5 5 5 526<< AG (D(D $ ( & &,0 2 2 2 226<< AI (F(F & ( ( (.2 4 4 4 4 4 4r c t d }t t j | d d d df t j | d t t j | t j |d d t t j | t j d z t j |d d t t j | d z t j |d
d d S ) NrI r7 r+ r8 r9 rK r; rL r= )r r r r rM rfft2r> r? s r
test_rfft2zTestFFT1D.test_rfft2 s- 8 Aqqq#2#v. QdKKKK Q QZ 88t E E E E Q"''*:*:: QW 55D B B B B Q95 QY 77d D D D D D Dr c t d }t |t j t j | d t |t j t j |d d d t |t j t j |d d d t |t j t j |d d d d S )NrI r7 r+ r8 r9 r; r= )r r r r irfft2rf r? s r test_irfft2zTestFFT1D.test_irfft2 s& 8 26== a 99EEEE26== aj )I)I ' ) ) )/3 5 5 5 526== ag )F)F $ ) & &,0 2 2 2 226== ai )H)H & ) ( (.2 4 4 4 4 4 4r c t d }t t j | d d d d d df t j | d t t j | t j |d d t t j | t j d z t j |d d t t j | d z t j |d
d d S ) NrT r7 r+ r8 r9 rV r; rW r= )r r r r rX rfftnr> r? s r
test_rfftnzTestFFT1D.test_rfftn s3 < Aqqq!!!RaRx0"&,,q//MMMM Q QZ 88t E E E E Q"',*?*?? QW 55D B B B B Q?; QY 77d D D D D D Dr c t d }t |t j t j | d t |t j t j |d d d t |t j t j |d d d t |t j t j |d d d d S )NrT r7 r+ r8 r9 r; r= )r r r r irfftnrm r? s r test_irfftnzTestFFT1D.test_irfftn s& < 26== a 99EEEE26== aj )I)I ' ) ) )/3 5 5 5 526== ag )F)F $ ) & &,0 2 2 2 226== ai )H)H & ) ( (.2 4 4 4 4 4 4r c t d dt d z z }t j t d |t d f }t j ||d d d f }t t j | t j | d t t j | t j |d d t t j | t j d z t j |d
d t t j | d z t j |d d d S )
N r* r
r r7 r+ r8 r9 r6 r; r< r= )r r concatenateconjr r hfftr> r! r x_herms r test_hfftzTestFFT1D.test_hfft sw 2JJF2JJ & Avayy 9:: NFAdddGLLNN3 4 4
1
rv{{6':':FFFF F++ F <<4 I I I I F++bgbkk9 F 99 F F F F F++c1 F ;;$ H H H H H Hr c t d dt d z z }t j t d |t d f }t j ||d d d f }t |t j t j | d t |t j t j |d d d t |t j t j |d d d t |t j t j |d
d
d d S ) Nrs r* r
r r7 r+ r8 r9 r; r= )r r rt ru r r ihfftrv rw s r
test_ihfftzTestFFT1D.test_ihfft s 2JJF2JJ & Avayy 9:: NFAdddGLLNN3 4 4 RV[[-@-@ A AMMMM RV[[ ' .9 .) .)/9 !- !; !;AE G G G G RV[[ $ .9 .& .&,3 !- !5 !5;? A A A A RV[[ & .9 .( .(.7 !- !9 !9?C E E E E E Er opc t d }g d}|D ]R} |t j || }t j ||| | }t ||d Sd S )NrT ))r r
r )r r r
)r
r r )r
r r )r r r
)r r
r axesr7 r+ )r r transposer )r! r} r r aop_trtr_ops r test_axeszTestFFT1D.test_axes s
< QQQ 5 5ABr|Aq))**ELAA 22E E5t 4 4 4 4 4 5 5r c t d }t j | }|j dz }t j j t j j ft j j t j j ft j j t j j
fg}|D ]i\ }}|j d|j z fD ]R}dD ]M} |||| } |||| }t |t j | d NSjd S )Nr6 r rA r^ r7 r+ ) r r linalgr: r` r r. r0 r/ r{ rv r ) r! r x_normr_
func_pairsforwbackr: tmps r test_all_1d_norm_preservingz%TestFFT1D.test_all_1d_norm_preserving s$ 2JJ ""
FQJvz26;/v{BFL1 v|RV[1
% D DJD$fah'
D
D B D DD $qAD111C $sad333C#F$&INN3$7$7dD D D D DD
D D Dr dtypec z t d | }t t j t j | |d t t j t j | |d d S )Nr6 r7 r+ )r astyper r r r. r/ r0 )r! r r s r test_dtypeszTestFFT1D.test_dtypes s
2JJ e $ $ BFJJqMM22ADAAAA RV[[^^44adCCCCCCr N)"r# r$ r% r4 r@ rD mark parametrizerF rN rQ rY r\ ra rc rg rj rn rq ry r| r r rX r[ rm rp r r halfsingledouble
longdoubler r& r r r( r( s 1 1 1B B B [ V%KLL' ' ML' C C C D D D
C
C
C
D
D
D@ @ @"4 4 4D D D4 4 4D D D4 4 4
H
H
H
E
E
E [ TBFK $&FL"&-$A B B5 5B B5D D D& [ Wrw 29')}'6 7 7D D7 7D D Dr r( r orderFznon-contiguousr c t j d }| ddd | d }dt j t j |j z t j |j j
z }|dk rt j | }n(|d d d }t j |d d d }|j
d
rAt d D ]/} ||| } ||| } t! || ||
0d S |j
d rig d}
|j
d r|
g d |
D ]/} ||| } ||| } t! || ||
0d S t% )N*
F)copyg @r r r r r )r, rtol)rM rX ))r r
)r
r )r r rX ))r )r
)r Nr )r r RandomStaterandr r> log2r` finfor epsasfortranarrayascontiguousarrayr# endswithr- r extendr ) r r r rngX_tolYr X_resY_resr axs r test_fft_with_orderr s
)
#
#C Ar!!%e!44A )) )BHQW,=,=,A AD || a
dddG 44R4 ) )
| U## !HH @ @DC%%%EC%%%E E5t$ ? ? ? ? ? @ @
/ 0 0 ''' < ( ( 2 KK000 1 1 1 @ @BCOOOECOOOE E5t$ ? ? ? ? ? @ @
llr zCannot start thread)reasonc 4 e Zd ZdZdZd Zd Zd Zd Zd Z dS ) TestFFTThreadSafe )i c 4 fdt j }fdt | j D }d |D d |D t | j D ]'}t d |d (d S )Nc 8 | | d S N)put)argsqfuncs r workerz.TestFFTThreadSafe._test_mtsame..worker s!
EE$$+ r c @ g | ]}t j f S ))targetr ) threadingThread).0r3 r r r s r
z2TestFFTThreadSafe._test_mtsame.. s=
+
+
+
V4)
<
<
<
+
+
+r c 6 g | ]}| S r& )startr r s r r z2TestFFTThreadSafe._test_mtsame.. s q r c 6 g | ]}| S r& )joinr s r r z2TestFFTThreadSafe._test_mtsame.. s a r )timeoutz6Function returned wrong value in multithreaded context)queueQueuer- threadsr get)r! r r expectedtr3 r r s `` @@r _test_mtsamez TestFFTThreadSafe._test_mtsame s
KMM4;
+
+
+
+
+
+DL))
+
+
+ A 1 t|$$ J JA quuQu / /H
J
J
J
J J Jr c t j | j dz dz }| t j j | d S Nr
y )r ones input_shaper r r! r s r r@ zTestFFTThreadSafe.test_fft& s> GD$ % % )" , "&*a(((((r c t j | j dz dz }| t j j | d S r )r r r r r r. r s r rF zTestFFTThreadSafe.test_ifft* s> GD$ % % )" , "&+q)))))r c t j | j }| t j j | d S r )r r r r r r0 r s r ra zTestFFTThreadSafe.test_rfft. s4 GD$ % % "&+q)))))r c t j | j dz dz }| t j j | d S r )r r r r r r/ r s r rc z TestFFTThreadSafe.test_irfft2 s> GD$ % % )" , "&,*****r N)
r# r$ r% r r r r@ rF ra rc r& r r r r sl G KJ J J$) ) )* * ** * *+ + + + +r r ) numpyr rD numpy.randomr
numpy.testingr r r r r r r r r( r r float32float64 complex64
complex128r rM rX r. rP r[ r skipifr r& r r r s
+ + +< < < < < < < < MD MD MD MD MD MD MD MD` RZ r}=? ? 3(8"9::
RV["&+ bflBFL 23 3 3 3 ;:? ? > G$9::$+ $+ $+ $+ $+ $+ $+ ;:$+ $+ $+r