qKg] <ddlZddlZddlmZddlmZmZmZm Z ddl Z ddl Z dZ GddZ GddZejj!d ej"ej$ej&ej(gejj!d d d gejj!d ej*j*ej*j,ej*j.ej*j0ej*j2ej*j4gdZejj!d d dgejj!dgddZejj;e dGddZdZdZ ejj!d ej*j*ej*j0ej*jBej*jDgejj!dejFgdejHdejJejHdejLgdZ'y)N)random)assert_array_equal assert_raisesassert_allcloseIS_WASMct|}dtjztj||z z}tj|j dd|z}tj |tj |zdS)Nyaxis)lennppiarangereshapesumexp)xLphases b/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numpy/fft/tests/test_pocketfft.pyfft1r sf AA "%%K299QK$$/ // KAat! DAQ AAFF1I1uz qst BMM!$4 56B&&++bffll1l&:a+@K$$/ // KAat! D$rctddtdzz}tt|tjj |dtt|tjj |ddtt|tj dz tjj |ddtt|dz tjj |d dy) Nr*ư>r+backwardnormortho>@forward)rrrrr sqrtr!rs rtest_fftzTestFFT1D.test_fftNs 2JF2J &QAT:QAJ!?dKQ"''"+- 17 3$ @Q#  19 5D Brr rr transpose)TFc fd}|turOtddtdzz}tjjtjj}}n?td}tjj tjj }}|||}||} ||| |} | | usJt| ||||} |tur| n|| } || | |} | | usJt| | y)Ncr)tj|jjStj|SNrrDTrr`s rrDz3TestFFT1D.test_fft_out_argument..zeros_like[.}}QSS)+++}}Q''r) r*r )outr )complexrrr r.r0r/r)r!r5r`r rDrOr r.expectedrjresult expected2out2result2s ` rtest_fft_out_argumentzTestFFT1D.test_fft_out_argumentWs ( G x 2fX&6#66A BFFKKCx A RVV\\Cqt$"QCd+}}68,- w&sJy,As40$7I.rrr c<tddtdzz}|j}tjj||}tjj|||}||usJt |||j}|dk(r|ddn |ddddf}tjj|d|}tjj|d||} | |usJt | ||dk(rt |dd|ddnt |ddddf|ddddf|j} |dk(r| d dn | ddd df} |dk(r| d d n | ddd d f} tjj| d|} tjj| d|| }|| usJt || |dk(r%t | dd |dd t | d d|d dn ? FFJJqbmmA.66q"=J >@ ]]9M : FFJJqbmmAU;J <; : D C ? ? : :s$"D"6AD.;D:"D+.D7:ErXNrVrYr[c rtddtdzz}t|tjj tjj|||dt j td5tjj g|dddy#1swYyxYw)NrTr*rWrUr+z!Invalid number of FFT data pointsr)rrrr r.rrr)r!rXrs r test_ifftzTestFFT1D.test_iffts 2JF2J & rvv{{266::ad:3${? ]]:!DF FFKKK &FFFs "B--B6ctddtdzz}ttjjtjj|ddtjj |dttjj |tjj |d dttjj |tj d z tjj |d dttjj |d z tjj |d dyNrTrir*r r rrUr+rVrWXrY@r[)rrrr fft2r\r]s r test_fft2zTestFFT1D.test_fft2s 8 r&"22 2 266::aa:#8q A AT 3 A AJ 7d D A)99 AG 44 A A)4 AI 6T Crctddtdzz}ttjj tjj |ddtjj |dttjj |tjj |d dttjj |tj d ztjj |d dttjj |d ztjj |d dyr)rrrr r.ifft2r\r]s r test_ifft2zTestFFT1D.test_ifft2s 8 r&"22 2 BFFKKK$: C Qd 4 Q QZ 8t E Q"'''*:: QW 5D B Q95 QY 7d Drc 6tddtdzz}ttjjtjjtjj|dddtjj |dttjj |tjj |d dttjj |tj d z tjj |d dttjj |d z tjj |d dyNrTrirhr*rr r rrUr+rVrWprYp@r[)rrrr fftnr\r]s r test_fftnzTestFFT1D.test_fftns < 2f\&:#: : FFJJrvvzz"&&**QQ*"7az@qJ I FFKKN '  A AJ 7d D A)>> AG 44 A A/: AI 6T Crc 6tddtdzz}ttjj tjj tjj |dddtjj |dttjj |tjj |d dttjj |tj d ztjj |d dttjj |d ztjj |d dyr)rrrr r.ifftnr\r]s r test_ifftnzTestFFT1D.test_ifftns < 2f\&:#: : FFKK BFFKKK$: C!K L FFLLO$ (  Q QZ 8t E Q"'',*?? QW 5D B Q?; QY 7d Drc td}|jd|jzfD]a}dD]Y}ttjj |||d|dzdztjj |||d[ttjj ||tjj ||d dttjj ||tj |z tjj ||d dttjj |||z tjj ||d ddy) NrTrrr=rXr rUr+r<rVrYr[)rsizerrr r0r\)r!rr=rXs r test_rfftzTestFFT1D.test_rffts+ 2J&&!AFF(#A>FFJJqADJ1+AqD1H>FFKKQTK2??  A # A 44 A  A #bggaj0 A 1 >  A #a' A 3$ @$rctjd}d}tjj||}t |tjj|d|d|dzdzdy)Nrr +=r?)rrr r0r)r!rr=rOs rtest_rfft_evenzTestFFT1D.test_rfft_evensV IIaL  FFKK1 266::ae,YadQh7eDrctjgd}tjj|}t |tjj|dddy)N)r rrrrrr)rrAr r0r)r!rrOs r test_rfft_oddzTestFFT1D.test_rfft_odd sA HH% & FFKKN266::a=!,59rc rtd}t|tjj tjj |dt|tjj tjj |dddt|tjj tjj |dddt|tjj tjj |dddy)NrTrUr+rVrWrYr[)rrrr r/r0r]s r test_irfftzTestFFT1D.test_irffts 2J266<< A7dC266<< AJ (G'()/3 5266<< AG (D$(&,0 2266<< AI (F&((.2 4rctd}ttjj |ddddftjj |dttjj |tjj |ddttjj |tj dz tjj |ddttjj |d z tjj |d dy) Nr rUr+rVrWrrYrr[)rrrr rrfft2r\r]s r test_rfft2zTestFFT1D.test_rfft2s 8  Aq#2#v. QdK Q QZ 8t E Q"'''*:: QW 5D B Q95 QY 7d Drc rtd}t|tjj tjj |dt|tjj tjj |dddt|tjj tjj |dddt|tjj tjj |dddy)NrrUr+rVrWrYr[)rrrr irfft2rr]s r test_irfft2zTestFFT1D.test_irfft2#s 8 266==a9E266==aj)I'))/3 5266==ag)F$)&,0 2266==ai)H&)(.2 4rctd}ttjj |ddddddftjj |dttjj |tjj |ddttjj |tj dz tjj |ddttjj |d z tjj |d dy) NrrUr+rVrWrrYrr[)rrrr rrfftnr\r]s r test_rfftnzTestFFT1D.test_rfftn-s <  Aq!RaRx0"&&,,q/M Q QZ 8t E Q"'',*?? QW 5D B Q?; QY 7d Drc rtd}t|tjj tjj |dt|tjj tjj |dddt|tjj tjj |dddt|tjj tjj |dddy)NrrUr+rVrWrYr[)rrrr irfftnrr]s r test_irfftnzTestFFT1D.test_irfftn7s < 266==a9E266==aj)I'))/3 5266==ag)F$)&,0 2266==ai)H&)(.2 4rc`tddtdzz}tjtd|tdf}tj||dddjf}t tj j |tj j |dt tj j |tj j |ddt tj j |tjd z tj j |d dt tj j |d z tj j |d dy) Nr*r r rUr+rVrWrTrYrZr[)rrrCconjrr hfftr\r!rx_herms r test_hfftzTestFFT1D.test_hfftAs 2JF2J &Avay 9: NNFAddGLLN3 4 1 rvv{{6':F F+ F <4 I F+bggbk9 F 9 F F+c1 F ;$ Hrc :tddtdzz}tjtd|tdf}tj||dddjf}t |tj j tj j|dt |tj j tj j|dddt |tj j tj j|d d dt |tj j tj j|d d dy) Nrr*r r rUr+rVrWrYr[)rrrCrrr ihfftrrs r test_ihfftzTestFFT1D.test_ihfftMs6 2JF2J &Avay 9: NNFAddGLLN3 4 RVV[[-@ AM RVV[['.9.)/9!-!;AE G RVV[[$.9.&,3!-!5;? A RVV[[&.9.(.7!-!9?C Eropctd}gd}|D]J}|tj||}tj||||}t||dLy)Nr)rr r)rrr )r rr)r rr)rrr )rr raxesrUr+)rrr`r)r!rrraop_trtr_ops r test_axeszTestFFT1D.test_axesYsT < QAr||Aq)*ELLAA2E E5t 4rctjdjdd}||ddjdk(sJy)Ndrhr r9r_sr)rhr9)rrrshaper!rrs rtest_s_negative_1zTestFFT1D.test_s_negative_1cs; IIcN " "2r *!wV,22g===rctjdjdd}tjd5||ddddy#1swYyxYw)Nrrh"`axes` should not be `None` if `s`rr)rrrrrwarnsrs rtest_s_axes_nonezTestFFT1D.test_s_axes_nonejsB IIcN " "2r * \\ D E qG F E Es AActjdjdd}tjd5||dddddy#1swYyxYw)Nrrhrrrrrrs rtest_s_axes_none_2DzTestFFT1D.test_s_axes_none_2DqsB IIcN " "2r * \\ D E qG$ 'F E Es AActd}tjd5||dddddy#1swYyxYw)Nrz%array containing `None` values to `s`r)rhNrhrr)rrrrs rtest_s_contains_nonezTestFFT1D.test_s_contains_nonews4 <  \\ G H qN 3I H Hs 7Ac td}tjj|}|jdz}tj j tj j ftj jtj jftj jtj jfg}|D]m\}}|jd|jzfD]J}dD]C}||||}||||}t|tjj|dELoy)NrTrrrrUr+) rrlinalgrXrr r.r0r/rrr) r!rx_normr= func_pairsforwbackrXtmps rtest_all_1d_norm_preservingz%TestFFT1D.test_all_1d_norm_preservings 2J" FFQJvvzz266;;/vv{{BFFLL1vv||RVV[[1  %JD$ffah'BDqAD1Csad3C#F$&IINN3$7dDC(%rr)r_rrNc fd}|turOtddtdzz}tjjtjj }}n?td}tjj tjj}}|||}||} ||| |} | | usJt| ||||} |tur| n|| } || | |} | | usJt| | y)Ncr)tj|jjStj|Srcrdrfs rrDz4TestFFT1D.test_fftn_out_argument..zeros_likergrrhr9rr*r)rjr) rkrrr rrrrr)r!r5r`rrDrr r.rlrjrmrnrorps ` rtest_fftn_out_argumentz TestFFT1D.test_fftn_out_arguments ( G z"Rz(:%::A RVV\\Cz"A bffmmCqt$"QCd+}}68,- w&sJy,As40$7I.rr c|tjjur td}ntddtdzz}t j t d5||tj|dddddd}|||d }tj|}|||d| }||usJt||y#1swYJxYw) Nrr*rrrrrr)rjrr)rhr9r9rrrrj) rr rrrrrrDr)r!r rrrlrjrms rtest_fftn_out_and_s_interactionz)TestFFT1D.test_fftn_out_and_s_interactions "&&,, z"Az"Rz(:%::A ]]:-> ? r}}Q'99 E@ qAI.mmH%Q!)5}}68,@ ?s  CCr) r9r9rctddtdzz}tjj||d}tj|}tjj||d|}||usJt ||y)N)rr9rrr*rrr)rrr rrDr)r!rrrlrjrms r!test_irfftn_out_and_s_interactionz+TestFFT1D.test_irfftn_out_and_s_interactionss < 2f\&:#: :66==ai=8mmH%qAI3?}}68,rN)2r#r$r%r4rmark parametrizersingledouble longdoublerJrRr^rkrrqrrrrrrrrrrrrrrrrrr rrrrrrrrrrrrrrrr&rrr(r(s1 [[Wryy"))R]]&KLBMB" [[Wryy"))R]]&KLEME.B [[VV, [[Ww&67 [[[-8/98-/6 [[VaV,9/-9/v= [[V%KL'M' C D C D@"E : 4D4D4 H E [[TBFFKK$&FFLL"&&--$AB5B5 [[TBFFKK$&FFKK$?@>@>  [[TBFFKK$&FFLL"&&--$ABB  [[TBFFKK#>?(@(  [[TBFFKK$&FFLL"&&--$&FFKK$?@4@4 D& [[V%;< [[Ww&67 [[[-8/98=/6 [[URVV[["&&,, $MN-O-  [[S9i"89-:-rr(r5orderFznon-contiguousr cdtjjd}|jdddj |d}dtj tj |jztj|jjz}|dk(rtj|}n#|ddd }tj|ddd }|jjd r4td D]%}||| }||| } t!|| || 'y|jjdr]gd} |jjdr| j#gd| D]%} ||| }||| } t!|| || 'yt%)N*r F)rLg @rr r rr r>)rr)r_)r rrr))r)r )rNr)rr RandomStaterandrBr\log2rfinfor5epsasfortranarrayascontiguousarrayr#endswithr-rextendr) r5rr rngX_tolYr X_resY_resraxs rtest_fft_with_orderrsf ))   #C Ar!!%e!4A ) )BHHQWW,=,A,A AD |   a  ddG  4R4 ) ||U#!HD%E%E E5t$ ?   / 0' <<  ( KK0 1BOEOE E5t$ ? lrCr=)Nr ctjjd}|jd}tj|tj |}tj j ||}|jj|jjk(sJ|jj|jjk(sJy)Nrrh)r5rr<) rrr r asarray complex64r flags c_contiguous f_contiguous)rr=rrress rtest_fft_output_orderr$s ))   #C  A 1BLL6A &&**Q!* C 99 ! !QWW%9%9 99 9 99 ! !QWW%9%9 99 9rzCannot start thread)reasonc2eZdZdZdZdZdZdZdZdZ y) TestFFTThreadSafer8)i cfd}tj}|}t|jDcgc]}t j |||f}}|Dcgc]}|j c}|Dcgc]}|jc}t|jD]}t|jd|d!ycc}wcc}wcc}w)Nc.|j|yrc)put)argsqfuncs rworkerz.TestFFTThreadSafe._test_mtsame..workers EE$+ r)targetr,r9)timeoutz6Function returned wrong value in multithreaded context) queueQueuer-threads threadingThreadstartjoinrget) r!r.r,r/r-rlr3trs ` r _test_mtsamezTestFFTThreadSafe._test_mtsames  KKM;DLL) +)  V4) <)  +AqA1a1t||$A quuQu/H J%  +s C C;Cctj|jdzdz}|jtjj|yNr y)rones input_shaper;r r!rs rr^zTestFFTThreadSafe.test_ffts8 GGD$$ % )" , "&&**a(rctj|jdzdz}|jtjj |yr=)rr>r?r;r r.r@s rrzTestFFTThreadSafe.test_iffts8 GGD$$ % )" , "&&++q)rctj|j}|jtjj |yrc)rr>r?r;r r0r@s rrzTestFFTThreadSafe.test_rffts. GGD$$ % "&&++q)rctj|jdzdz}|jtjj |yr=)rr>r?r;r r/r@s rrzTestFFTThreadSafe.test_irfft!s8 GGD$$ % )" , "&&,,*rN) r#r$r%r4r?r;r^rrrr&rrr'r's&GKJ$)**+rr'c&tjd}tjj|dtjj |dtjjtj dgt dy)Nrhr r<r)rrr r/rrArk)rs rtest_irfft_with_n_1_regressionrE&sX " AFFLLaLFFKKQKFFLL1#w'2L.rctjddz}tjj|d}tjgd}t ||y)Nr9y??rhr<) g4@gk"#@g'g$ pn]k@grg@gG& gdƍĿgt=M?g@L>M4)rrr rrAr)rrmrls r"test_irfft_with_n_large_regressionrG.sI ! A VV[[b[ !Fxx23HFH%rdata)FTFrhr;r9c||}|jtj|d}||}t||y)Nr:)rBr result_typer)rHr rm float_datarls r#test_fft_with_integer_or_bool_inputrL8s9YFR^^D"56J:Hvx(r)(numpyrr numpy.randomr numpy.testingrrrrr5r2rrr(rrfloat32float64r complex128r rrr.rrrr$skipifr'rErGr0r/rAruint8int16rLr&rrrVs!    +<< r-r-j  RZZr}}=?3(8"9: RVV[["&&++ bffllBFFLL 23 3;?>3*-m,:-.:G$9:$+$+;$+N/&FFJJ RVV[["&&,,! BHH !" BIIb! BIIarxx " ) )r