:j`@ ddlZddlZddlZddlmZddlZddlmZddlm Z ddl m Z m Z m Z ddlmZmZddlmZmZmZmZmZmZmZmZmZmZmZmZej<j>Zej<j@Z e gZ!gdZ"ee jFGd d Z$ee jJGd d Z&ee jNGd dZ(ee jRGddZ*ee jVGddZ,ee jZGddZ.ee j^GddZ0gdZ1gdZ2ee jfGddZ4egdegdegdegdegddegddegdegd ddd! Z5ee jld"Z7ee jpGd#d$Z9ee jtGd%d&Z;ee jxGd'd(Z=ee j|Gd)d*Z?ee jGd+d,ZAee jGd-d.ZCee jGd/d0ZEee jGd1d2ZGee jGd3d4ZIee jGd5d6ZKee jGd7d8ZMid9egd:d;egd<d=egd>d?egd@dAegdBdCegdDdEegdFdGegdHdIegdJdKgdLdMgdNdOgdPdQgdRdSgdTdUgdVdWgdXdYgdZgd[gd\d]ZNee jGd^d_ZPgd`gdagdbdcgfgddgdegdfggdgfgdhgdigdjgdkgdlggdmfgdngdogdpgdqggdrfdsZQee jGdtduZSee jGdvdwZUee jGdxdyZVee jGdzd{ZXee jGd|d}ZZed~dej<jde"Dcgc]\}}ee\e |||c}}dZ]ee dZ^gdZ_ej<jde_Dcgc]}ee|d|c}ee dZ`ee jdZaycc}}wcc}w)N)array)raises)fft)windows get_windowresample)_WIN_FUNC_DATA _WIN_FUNCS) xp_assert_closexp_assert_equalarray_namespaceis_torchis_jaxis_cupyassert_array_almost_equal SCIPY_DEVICEis_numpymake_xp_test_casemake_xp_pytest_param_xp_copy_to_numpy))boxcar)triangr)parzenr)bohmanr)blackmanr)nuttallr)blackmanharrisr)flattopr)bartlettr)barthannr)hammingr)kaiser)dpss))gaussian?)general_gaussian)?r')chebwinr$)cosiner)hannr) exponentialr)taylorr)tukeyr))lanczosrceZdZdZy) TestBartHanncttjdd||jgd|jddttjd| |jgd |jddttjdd | |jgd |jddy) NTsymxp)rgA,q? 5xB$?r;gA,q?rdtypeV瞯<rtolatolr:)rHzG?\(\??rErDrF)rrDrErFrErD)r rr!asarrayfloat64selfr:s t/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/scipy/signal/tests/test_windows.py test_basiczTestBartHann.test_basic5s((< $DKM::#W" 0 ((r2 #F)+#5" 0 ((Eb9 #C2:: V" 0N__name__ __module__ __qualname__rLrrMrKr5r52s 0rMr5ceZdZdZy) TestBartlettczttjd||jgd|jttjd||jgd|jttjdd||jgd|jy) Nr7rC)r皙?皙?rVrUrr<rB)rUUUUUU?UUUUUU?rFrXrWrF)rrWrXrFrXrW)r rr rGrHrIs rKrLzTestBartlett.test_basicFs((r2 #=RZZ P R((r2 #B"** U W((Eb9 #?rzz R TrMNrNrrMrKrSrSCsTrMrSceZdZdZy) TestBlackmanc ttjdd||jgd|jdttjdd||jgd |jd ttjd| |jgd |jdttjdd | |jgd|jdy)Nr7Fr8)rp= ף?)\(?rFr]r\r<+=rArB)rFi'?b@c?Zgs?rbrar`:0yE>rC)r쳩ֲ?,,?rerdrT)rr\r]rFr]r\r)r rrrGrHrIs rKrLzTestBlackman.test_basicRs(("= #C2:: V" $ (("= $M+-**#6"  # ((r2 $O*,**#6#  $ ((DR8 #F jj#*05 7rMNrNrrMrKrZrZOs7rMrZceZdZdZy)TestBlackmanHarriscttjdd||jgd|jttjdd||jgd|jttjd||jgd |jttjdd ||jgd |jy) Nr7FrC)iUMu?0_^}?縯猨?rFrkrjr<rBr8)rigjP?g'm$M?gu?gu?g*m$M?gjP?)ri^?gU׈g?gU׈g?rlriT)rirjrkrFrkrjri)r rrrGrHrIs rKrLzTestBlackmanHarris.test_basicis..q%B? $=DFJJ#P Q ..qeC $M+-**#6 7 ..qR8 $U*,**#6 7 ..qdrB $7>@jj#J KrMNrNrrMrKrgrgfsKrMrgceZdZdZdZdZy) TestTaylorcttjddd||jdg|jttjddd||jgd|jttjd dd||jgd |jy ) zTests windows of small length that are normalized to 1. See the documentation for the Taylor window for more information on normalization. r%r'rCrFr<)%6'HA?)T ?rFrsrrr7)δX?繡1E?PD3m?rvrurtNr rr1rGrHrIs rKtest_normalizedzTestTaylor.test_normalized~s q!RB7 C5 ; = NN1a + JJLZZ  )  NN1a + JJZZ  ! rMc ttjdddd||jgd|jttjdddd||jgd |jy ) zTest windows of small length that are not normalized to 1. See the documentation for the Taylor window for more information on normalization. rqr'rpFnormr:) P?sCp?g)8ux?r}r|r<r7))S~?rFkt#?rrFr~NrwrIs rKtest_non_normalizedzTestTaylor.test_non_normalizedsx  NN1a%B 7 JJZZ  !  NN1a%B 7 JJZZ  ! rMc d}d}tj|dddd|}tt||}dt j t j |t j|z z}t jt j|dkD}t j||| }d t j|d kz|z |z} d t j|d kz|z |z} tj|d d sJtj| ddsJtj| ddsJy)aThis test ensures the correctness of the implemented Taylor Windowing function. A Taylor Window of 1024 points is created, its FFT is taken, and the Peak Sidelobe Level (PSLL) and 3dB and 18dB bandwidth are found and checked. A publication from Sandia National Laboratories was used as reference for the correctness values [1]_. References ----- .. [1] Armin Doerry, "Catalog of Window Taper Functions for Sidelobe Control", 2017. https://www.researchgate.net/profile/Armin_Doerry/publication/316281181_Catalog_of_Window_Taper_Functions_for_Sidelobe_Control/links/58f92cb2a6fdccb121c9d54d/Catalog-of-Window-Taper-Functions-for-Sidelobe-Control.pdf ii#F)nbarsllr{r9r:rr'gX$g_2gAfAr%)abs_tolgLJ?皙?gZӼ@N) rr1rrnplog10absmaxargmaxdiffmathisclose) rJr:M_winN_fftwf_npspec first_zeroPSLLBW_3dBBW_18dBs rKtest_correctnesszTestTaylor.test_correctnesss NN5qbu%B O$Q'/BHHRVVD266$<$7899YYrwwt}q01 vvd:zk23299T%8899EAEIBIId&99::UBUJ||D(A666||FFC888||GVS999rMN)rOrPrQrxrrrrMrKrnrn{s ( $":rMrnceZdZdZy) TestBohmanc|ttjd||jgd|jttjdd||jgd|jttjdd ||jgd |jy) Nr7rC)rg$I ?6D?rg I ?rr<rBTr8)rPG?4}|?rF6}|?rrF)rrrrFrr)r rrrGrHrIs rKrLzTestBohman.test_basicsqR0 $O*,**#6 7 qdr: $O*,**#6 7 q%B7 $L*,**#6 7rMNrNrrMrKrrs 7rMrceZdZdZy) TestBoxcarczttjd||jgd|jttjd||jgd|jttjdd||jgd|jy)Nr7rCrFr%r%r%r%r%r<rBrFr%r%r%r%r%r%F)r rrrGrHrIs rKrLzTestBoxcar.test_basicsqR0 #7rzz J LqR0 #:"** M Oq%B7 #7rzz J LrMNrNrrMrKrrsLrMr)5CV?CԳ ?Q&E?:*?;Bu?9#?=[?Z G?ٵݒ?Lg?(+ ?T=? zR&5?scH?u?]/M`?xC8???9?mo?[?T:?sK!q?/M.?×?)s?rFrrrrrrrrrrrrrrrrrrrrrrrrrr)6?2?w.v?f*#?灖`?d${!?? M?ut\?Ljh?xak?WY?qj?<)[?@ի?MEs?甽/?{l?[^6?:Ug$?: U\?I.!v?Clp?gF?稦$p?VLy?.?rFrFrrrrrrrrrrrrrrrrrrrrrrrrrrc*eZdZdZdZdZdZdZy) TestChebWinc tj5tjddtt t j dd||jgd|jd t t j d d||jgd |jt t j dd ||jgd |jt t j d d ||jgd|jt t j dd d||jgd|jdddy#1swYyxYw)NignoreThis window is not suitabler7drC)} ɺ?L>?rFrFrrr<rcr_rB)S?D1mPC?%5R?rFrrr )rFTEm? -o?rrrF)rFF3?NO? r]?rrrFF)rFrrrrr) warningscatch_warningsfilterwarnings UserWarningr rr-rGrHrIs rKrLzTestChebWin.test_basic s]  $ $ & P  # #7 F GOOAsr:JJ(P/1jj':"&   GOOAsr:JJ(P/1jj': ; GOOArb9JJ(U.0jj': ; GOOArb9JJ(AHJ 'T U GOOAr5R@JJ(JJ * H $ $ & 8  # #7 FqRB7H 8 "(,@!L  8 8 4A??Bc|jgd|j}tj5tjddt t jdd|}dddt|d y#1swYxYw) N)rFR?N(D!T?#R?r r r rFr<rrirrrr)rJr:cheb_even_low_at_truers rKtest_cheb_even_low_attenuationz*TestChebWin.test_cheb_even_low_attenuationBs " ,@GIzz!+!S $ $ & :  # #7 Fcb9I : ")-BAN  : :rN)rOrPrQrLrrrrrrMrKrr s P:RTMOrMr)|dy?vS ?{?rFr)1B?%?rr)́R?8,6V?rFr)\Kr? oh?rr)rrrFrr)rrrFrr) )rN皙?F)rNrT)rNrFF)rNrFT)rr'rF)rr'rT)rr'rFF)rr'rFT)rqNrT)rqNrFT)rqr'rT)rqr'rFTctjD]\\}}|#tttj g|d|i+t j |d|i}t ||j|d^y)Nr:r^r@)exponential_dataitems assert_raises ValueErrorrr0r rG)r:kvwins rKtest_exponentialr!jsf &&(<1 9 *g&9&9 EA E" E%%q0R0C CAU ;  #r?rB)r\$:?H`r?8T^j?rZrYrXTrCrX hd?Tͻ?r]r\rrrWrErFrErWr)r rr/rGrHrIs rKrLzTestHann.test_basics QEb9 #C2:: V" 0  QEb9 $L+-**#6#  0  Q4 $O*,**#6#  0  Q2. #F jj#*" 0rMNrNrrMrKrUrUs0rMrUceZdZdZy) TestKaiserc|ttjdd||jgd|jttjdd||jgd|jttjdd||jgd |jttjdd||jgd |jttjddd ||jgd |jy) Nr7r*rC)l?<3N?i>%?rdrcrbr<rB)rb $?6?rFrfrerbg@)§Aը?O F?$?rirhrg)rg*'?=a?rFrkrjrgF)rgrjrkrFrkrj)r rr#rGrHrIs rKrLzTestKaiser.test_basicsq#"5 $L+-**#6 7 q#"5 $L+-**#6 7 q#"5 $L+-**#6 7 q#"5 $L+-**#6 7 q#u< $8?Azz#K LrMNrNrrMrKr`r`sLrMr`ceZdZdZdZy)TestKaiserBesselDerivedctjdd|}t||jgd}tj|d|}tjd|d|}t |||d|dzdz|| dzddzz}t ||j |j|j tjdtjdz |dd }|j|tjdzd z }t ||t tjd tjdz |dd|jd dg|j t tjdtjdz |dd|jgd|j y)Nr*@betar:r)zkaiser bessel derivedroFfftbinsr:r'r<r%@rgU?gU3|\?r7r5)g HZ1?gKf?g"iF5?)rkaiser_bessel_derivedr rGrr onesshaper=rpi ones_likersqrtrH)rJr:rMw2actualdesireds rKrLz"TestKaiserBesselDerived.test_basic s  ) )#CB ?2::b>*   ) )!#" =    > !5R927AFq1aR1WX;!#33 FLL IJ..qruuqyRH!L,,v&15;(55abeeaiBOPRQRS NN#C2:: V X 55abeeaiBOPRQRS #S)+#5 6rMcd}d}tt|5tj|dzd|dddd}tt|5tj|dzdd| dddy#1swYBxYw#1swYyxYw) NrzHKaiser-Bessel Derived windows are only defined for even number of pointsmatchr%rorpzCKaiser-Bessel Derived windows are only defined for symmetric shapesF)rqr9r:)rrrru)rJr:r{msgs rKtest_exceptionsz'TestKaiserBesselDerived.test_exceptions,s " :S 1 A  ) )!a%bR @ A" :S 1 L  ) )!a%be K L L  A A  L LsA3 A?3A<?BN)rOrPrQrLrrrMrKrmrm s 6> LrMrmceZdZdZy) TestNuttallcttjdd||jgd|jttjdd||jgd|jttjd||jgd |jttjdd ||jgd |jy) Nr7Fr8)A"7?v;O`T?gEJ?rT)rrrrFrrr)r rrrGrHrIs rKrLzTestNuttall.test_basic>su< $/68jj#B C u< $L+-** #6 7 b1 $CJL**#V W 4B7 $ELNJJ#X YrMNrNrrMrKrr;sYrMrceZdZdZy) TestParzenc|ttjd||jgd|jttjdd||jgd|jttjdd ||jgd |jy) Nr7rC)e/?rW88?rrWrr<rBTr8)%OZQ%w?2|&?݁?rFrrrF)rrrrFrr)r rrrGrHrIs rKrLzTestParzen.test_basicTsqR0 $T*,**#6 7 qdr: $M+-**#6 7 q%B7 $8?Azz#K LrMNrNrrMrKrrQs LrMrceZdZdZy) TestTriangc|ttjdd||jgd|jttjd||jgd|jttjdd| |jgd |jy) Nr7TrC)UUUUUU?r*竪?rr*rr<rB)rWr*rEr%rEr*rWFr8)rWr*rEr%rEr*)r rrrGrHrIs rKrLzTestTriang.test_basicgsq$26 #A T VqR0 #DBJJ W Yqe; #?rzz R TrMNrNrrMrKrrdsTrMr)rr*T)rFrFr)r?T)r?rr)rrFT)rrErEr)rr*F)rrFrFrF)rrF)rrnOC?rFr)rrFF)rr*rFr*)rqrT)rFrFrFrFrF)rqrVT)rYx?rFrr)rqrFT)rr*rFr*r)r7rr)rBrr)r7rW)rr%r%r%r%r)rBrW)rr%r%r%r%r%rr7)rr]rFrFgRͻ?r)rB)rrErFrFrFrErr7rE)rgxų &?rFrFgvų &?rrBrE)rg#}`yq?)!?rFrg#}`yq?rr[r^)r7r%)rBr%ceZdZdZdZy) TestTukeyc tjD]w\}}|#tttj g|d|i+t |r |dvrddd}nddd}t j |d|i}t||j|fddi|yy) Nr:)rrrrg+i)+`>rcr?r> check_dtypeF) tukey_datarrrrr2rr rG)rJr:rr atol_rtolr s rKrLzTestTukey.test_basics$$& @DAqyj'--C!CCB@ @rMctjdd|}tjd|}t||tjdd|}tjd|}t||y)NrrrCr%)rr2rr r/)rJr:tuk0box0tuk1han1s rK test_extremeszTestTukey.test_extremessZ}}S!+~~cb)d#}}S!+||CB'd#rMN)rOrPrQrLrrrMrKrrs  @$rMr)IP?;]L?rr)g/vIef?rg>ef) #n?gLor)gm/?g96+?g雦br?)gw?V?g>J#?g;Wi?g֙}I#?gr?V?)g:7A!(?cHX?rcHXS[@!(ܿ)qh4?T1a~²?gEܿrr)rrrgS[@!(?r)gu+t?X>߿g36?rg;St?)gML?gh5"z?g\v4?gPa?gd?)dsk8Si?p&p?5Bu?OL{?08À?w-}?DC1?XB?QST`?s\NK? 3??gd?g!DKy?g+o ?gYh{b?g{^y?gSh1?gy%@?g|hzj??gP\D?gz ?gVx<҅?g?gq&?gb˱?g8|p?g_cH?g6K ?gwp΍?g"Q q?g@m8Migǐ3g7/`_qgS!g8QJg68gu\qg/Z~ܳgpQZgCg3 `g7ſg3zĿg%PldĿgMuÿgI!w.ÿ)g?g"?gy?'?g`?))rrr')r5gffffff?r5)rqr,rq)rr'rcBeZdZdZdZdZdZeddZy) TestDPSSctjD]C\}}tj|d|d\}}t ||dd|t ||ddd|Ey) NT return_ratiosr:rgHz>)rAerr_msgr%gh㈵>)r@rAr!) dpss_datarrr&r )rJr:rrr ratioss rKrLzTestDPSS.test_basicsZOO% KDAq!,,"EKC C1D! < FAaDt$ J KrMc8tddD] }tj||dz |}|dz}tt j |dj ||tj||dz d| }|dkDr?tt j |dj ||t||d tj||dz d| }|dk(rdnd }tt j |dj || y) Nr%g@rCr'rF)r! subsamplerzgQ?rr)rangerr&r rrsumr )rJr:r{r expectedwin_subwin_2s rK test_unityzTestDPSS.test_unitysq" -A,,q!c'b1C1uH BJJsB/335x$'5 +ll1a#gKBGG1u 7B 7 ; ; =x(/y3W48LLAG!;EFqH BJJub1557$)7 -! -rMcTtjdddd|d}t||j|tjdddd|d}t||j|tjdddd|d}t||j|y) Nr7rTrr%rBr )rr&r ry)rJr:lams rKrzTestDPSS.test_extremessll2q!4B?BR\\#./ll2q!4B?BR\\#./ll2q!4B?BR\\#./rMctttjdddtttjdddtttjdddtttjdddtttjdddtttjdddtttjdddy) Nrr,皙?r5rr%)rrrr& TypeErrorrIs rKtest_degeneratezTestDPSS.test_degeneratesj',,3;j',,3;iq#s;j',,3:j',,2q9j',,1a8j',,Aq9rMT)np_onlyc6tjdd}t|dgtjddd\}}t|dg|dk(sJtjdddd\}}t|dgt|tj sJt|dgt ttjddd| t ttjddd | t ttjddd | t ttjd dd | t ttjd dd | t ttjd d d | t ttjddd | y)Nr%rFT)r r)Kmaxr r,r1rCr2r3r5r) rr&r isinstancerndarrayrrr4)rJr:rratios rKtest_degenerate_signle_samplesz'TestDPSS.test_degenerate_signle_sampless, LLB B4 <<2T:5B4 {{<<2ATB5B4 %,,,t$j',,3rBj',,3rBiq#srBj',,3bAj',,2qR@j',,1aB?j',,AqR@rMN) rOrPrQrLr,rr5skip_xp_backendsr<rrMrKrrs2K-*0:d#A$ArMrceZdZdZdZy) TestLanczoscttjdd||jgd|jdttjd||jgd |jdttjd d ||jgd |jdy) Nr7Fr8rqϾv?;Fv?rFrCrBr<& .>r_rCruH%?;9?rGrFrrBT)rrBrCrFrCrBr)r rr3rGrHrIs rKrLzTestLanczos.test_basics u< $18: #D"  # b1 $5<>JJ#H"  # t; $5<>JJ#H"  #rMcdD]V}tj|d|jd|k(sJtj|d|jd|k(rVJy)N)rr Fr8rT)rr3rw)rJr:ns rKtest_array_sizezTestLanczos.test_array_sizes_ EA??1%B7==a@AE EE??1$26<dZ dZ!y) TestGetWindowz*Unit test for `scipy.signal.get_windows`. ctjD].\}}|dj}d|d}||vsJ||ddvr.Jy)a|Verify that the `_windows._WIN_FUNC_DATA` dict is consistent. The keys of _WIN_FUNC_DATA are made of tuples of strings of allowed window names. Its values are 2-tuples made up of the window function and a entry characterizing the existence of window parameters as ``True``, ``False`` or ``'OPTIONAL'``. It is verified that the correct window name (i.e., corresponding to the function in the value tuple) is included in the key tuple. It is also checked that the second entry in the value tuple is either ``True``, ``False`` or ``'OPTIONAL'``. rzFunction name in z* does not contain name of actual function!r%)TFOPTIONALN)r rrO)rJnn_v_ func_namers rKtest_WIN_FUNC_DATA_integrityz*TestGetWindow.test_WIN_FUNC_DATA_integrity$sb&++- 6GC1I%cU*TUC# (S (#a555 55  6rMctjdd|}t||j|tjdd|}t||j|y)Nr rC)r)rrr ryrJr:rs rK test_boxcarzTestGetWindow.test_boxcar8sP   x 32<<?+   {B2 62<<?+rMctj5tjddtt j ddd|}dddt |jt|jd y#1swY7xYw) Nrr)r-rrFrrr<rr) rrrrrrrrGrrHrWs rK test_cheb_oddzTestGetWindow.test_cheb_oddAsw  $ $ & O  # #7 F""#3R2NA O " rzz-rzzz:A  O Os 5BB ctj5tjddtt j ddd|}dddt |jtdy#1swY+xYw) Nrr)r-rrFrrrr) rrrrrrrrGrrWs rKtest_cheb_evenzTestGetWindow.test_cheb_evenKsk  $ $ & N  # #7 F""?B"MA N "!RZZ%?K  N Ns 5A44A=c~tjddd|}tjdd|}t||y)N)r&r5@Frrr5rC)rrr&r rJr:win1win2s rK test_dpsszTestGetWindow.test_dpssSs4!!+r5RH||Bb)d#rMctjdd|}tjddd|}t|r2t |||j |j jyt||y)Ng@r^rCFr) rrr#rr finfor=epsr r_s rKtest_kaiser_floatzTestGetWindow.test_kaiser_floatYs\!!#rb1~~b#u4 ": D$RXXdjj-A-E-E F D$ 'rMNxr1g@r5ctjtd5tjd||dddy#1swYyxYw)Nz^Parameter Nx=.*rr/rC)pytestrrrr)rJrgr:s rKtest_invalid_parameter_NXz'TestGetWindow.test_invalid_parameter_NXds:]]:-? @ 2   vrb 1 2 2 2s >Actjtd5tjddd|dddtjtd5tjdgd| dddtjtd 5tjd d| dddtjtd 5tjd d| dddtjtd5tjdd| dddtjtd5tjdd| dddtjtd5tjdd| dddtjtd5|xst }tjdddgfd| dddy#1swYxYw#1swYxYw#1swY_xYw#1swY0xYw#1swYxYw#1swYxYw#1swYxYw#1swYyxYw)z?Raise all exceptions (except those concerning parameter `Nx`). z^Parameter fftbins=.*rr/rqr%rrNz^Parameter window=.*rCz%^First tuple entry of parameter win.*)*z ^Invalid window name 'INVALID'.*INVALIDz#^'hann' does not allow parameters.*)r/r%z ^'kaiser' must have parameters.*r#z^Window dpss must have one.*)r&r%r'z#^'general_cosine' does not accept.*general cosiner')rirrrrr)rJr:xp_s rKtest_invalid_inputsz!TestGetWindow.test_invalid_inputsks ]]:-D E <   vq! ; < ]]:-C D 4   y! 3 4 ]]:-T U 0   uaB / 0 ]]:-O P 4   y! 3 4 ]]:-R S 6   {A" 5 6 ]]:-O P 3   xr 2 3 ]]:-K L 9   ~qR 8 9 ]]:-R S F(C    01a&91 E F F < < 4 4 0 0 4 4 6 6 3 3 9 9 F Fs_G2G?H HH& H3H?'I 2G<?H  HH#&H03H<?I Ictjdd|}ttddd||ttdd||ttddd||tjdd|}ttdd||ttddd||ttd d||ttd dd||y ) zIEnsure that suffixes `_periodic` and `_symmetric` work for window names. rqTr8r Frrbartlett_symmetricrCbartlett_periodicN)rr r r)rJr:w_symw_pers rKtest_symmetric_periodicz%TestGetWindow.test_symmetric_periodics  4 :q%BGO #7rBEJ #7DRPRWX  "5 :qR8%@ :q$2FN #6bA5I #65RP rMcd}|jd}tjd|dz|}d}tt|5t ||j d|z|dddy#1swYyxYw) N)r#g @r'rCz^window must|^window.shape=rr)window)arangerrrrrrw)rJr:osfactorsigr mesgs rKtest_array_as_windowz"TestGetWindow.test_array_as_windowskiin  (a-BG, :T 2 ? S#))A,1# > ? ? ?s A**A3c ttd|jgdfd|jgd|jttd|jgdfdd|jgd|jt j t 5tdgdfd| dddy#1swYyxYw) Nr>r;rr=r<F)rs)rU皙?rrUrC)r rrGrHrirrrIs rKtest_general_cosinez!TestGetWindow.test_general_cosines $4bjj6Q#RTUV #5RZZ H J $4bjj6Q#RTU+02 #9 L N]]: & F (/:A" E F F Fs (CC cttdd||jgd|jttddd||jgd|jy) N)rHrCrqrC)rUжn?男*?rrr<FrrrDr rrGrHrIs rKtest_general_hammingz"TestGetWindow.test_general_hammings` #;Q2F #T)+#5 6  #;QRTU #JJ#HNR T  9aB7"614 6rMctdd}t|tjsJtdd|}t |st|tjrJyy)Nr3r7rC)rr9rr:r)rJr:r s rKtest_xp_defaultzTestGetWindow.test_xp_defaultsSA&#rzz***A"-|!#rzz2 222rMN)"rOrPrQ__doc__rSrrrrXr-rZr\r&rbr#rfrimark parametrizerrHr/rjrpr rvr~r>rrHrr3rrrrMrKrMrM s46(w~~&,',w' ( w'L(Lw||$$%$ w~~&('( [[TBZRZZ]#;<w||$2%=2 F(w''() w~~&?'?w--.F/Fw../Q0Qw'6(63rMrMc"eZdZejj ddejj ddejj ddejj ddd Zd Zd Zd Z d Z y)TestGeneralGaussianr{)r7rBp)r%r,r|)rFrtr9TFcH|r'|j||j|dz dz z }n+|j|dz|j|dz z }|dd}|jdt||z d|zzz}t j |||||}t ||d y) Verify basic parametrizations. r<r%r'Nr1grC-q=r_)rzrHexprrr+r ) rJr{rr|r9r:tx_refxs rKrLzTestGeneralGaussian.test_basics  !2:: .!A#:A !A#RZZ 01q58A#2Atc!c'lqs334  $ $Q3 ;5u-rMctjddd|}tjdddd|}t||y)IVerify the correctness of the default value of function parameter `sym`. r7r%rCTr8N)rr+r rJr:rrs rKtest_default_func_parameterz/TestGeneralGaussian.test_default_func_parameters:  $ $Q1 4((AqdrB5!rMc~tjddd|}tjdd|}t||y)zbTesting that `general_gaussian` with p = 1 is equivalent to the normal `gaussian` window. rBrFrtrr|r:)stdr:N)rr+r(r )rJr:w1r|s rKtest_p1_equals_gaussianz+TestGeneralGaussian.test_p1_equals_gaussians8 % %a3CB ?   aSR 0BrMctjddd|}t|j|dk(sJtjddd|}t|j|dksJy) z- Testing that if M is odd, the peak is at 1. rBr5rtrrFr7r,N)rr+floatrrWs rKtest_peak_valuez#TestGeneralGaussian.test_peak_valuesf  $ $Q! <RVVAY3&&&  $ $Q#32 >RVVAY#%%%rMcftjddd|jddk(sJttjddd||j dg|j t jt5tjddd|dddy#1swYyxYw)z2Test the length edge cases are handled correctly. rr%rCrFr<r1N) rr+rwr rGrHrirrrIs rKtest_len_edge_casesz'TestGeneralGaussian.test_len_edge_casess''1aB7==a@AEEE 00AqR@ C5 ; =]]: & 6  $ $RA" 5 6 6 6s B''B0N) rOrPrQrirrrLrrrrrrMrKrrs [[S&) [[S(+ [[UH- [[UM2 .3.,* ."  & 6rMrceZdZejj ddejj dddZdZdZy) TestCosiner{)r5rr9rc4|r$|j||jdz|z }n.|j|dz|jdz|dzz }|dd}|j|j|z}t j |||}t ||dy) rr<r*r%Nr1r8rr_)rzrHsinrxrr.r )rJr{r9r:rrrs rKrLzTestCosine.test_basics 1BJJ/#5:A1Q3bjj1C7AaC@A#2Aruuqy! NN1#" -5u-rMcztjd|}tjdd|}t||y)rr7rCTr8N)rr.r rs rKrz&TestCosine.test_default_func_parameters. NN1 $qdr25!rMcZtjd|jddk(sJttjd||j dg|j t jt5tjd|dddy#1swYyxYw)z9Testing that the length edge cases are handled correctly.rrCr%rFr<r1N) rr.rwr rGrHrirrrIs rKrzTestCosine.test_len_edge_cases s~~aB'--a0A555 qR0 C5 ; =]]: & & NN2" % & & &s B!!B*N) rOrPrQrirrrLrrrrMrKrrsH [[S&) [[UM2 .3* ." &rMrz dask.arrayz(https://github.com/dask/dask/issues/2620reasonzwindow,window_name,paramsc  tt|}t|r|dvrtjd|d|dvrKt |rtjdt |rtdk7rtjdtj5tjd d t|d g|d |d }|dg|d|d }t|dd||dg|d |d }|d g|d|d }t|dd||dg|d |d jddk(sJ|dg|d|d jddk(sJ|dg|d |d jddk(sJ|dg|d|d jddk(sJtt|dg|d|itt|dg|d|it!|dg|d |d |j#g|j$t!|dg|d|d |j#g|j$t!|dg|d |d |j#dg|j$t!|dg|d|d |j#dg|j$|dg|d |d j&|j$k(sJ|dg|d|d j&|j$k(sJ|dg|d |d j&|j$k(sJ|dg|d|d j&|j$k(sJ|dg|d |d j&|j$k(sJ|dg|d|d j&|j$k(sJ|j)|dg|d |d dksJ|j)|dg|d|d dksJ|j)|dg|d |d dksJ|j)|dg|d|d dksJt+|dg|d|d }|j-|}t||j/|dt+|dg|d|d }|j-|}t||j/|ddddy#1swYyxYw)N)r1r-zwindow_name = : item assignmentr)r&z'dpss window is not implemented for cupycpuz)needs eight_tridiagonal which is CPU onlyrrr Tr8rBFr1 r7rg@r:ir<r%rFrg)\(?r^r_rI)getattrrrriskiprrrrrrrr rwrrr rGrHr=allrimag zeros_like)ry window_nameparamsr:rr|ress rKtest_windowfunc_basicsrsWk *F bzk%:: nk-->?@h 2; KKH I B>2>j&"=v="= q;6;t; 2RZZ 8 :q<6r-chebzgeneral hammingrHrBrC)rrr)r:winstrs rKtest_needs_paramsrds'@ j*faB?@rM)r!r rrrrr.rr"rrr1r0poissonr2tuktriangler3rz window,winstrct|r|dvrtj|dt|d|}|jddk(sJy)N)r1rrrBrCr)rrirrrw)r:ryrr s rKtest_not_needs_paramsrsJbzf * fX%678 VQ2 &C 99Q<1  rMc tjfD]}|d|}t|j}|j |j |||z }t ||jddd|d|}|j |j |||z }t ||jdddy)NirCrF)rcheck_0di)rr3r fliprrr rG)r:r rrerrors rKtest_symmetricrs S  q!&&rvva$q'k*+rzz#EER  rvva$q'k*+rzz#EER SrM)brrnumpyrrrirr scipy.fftr scipy.signalrrrscipy.signal.windows._windowsr r scipy._lib._array_apir r r rrrrrrrrrrr=xfail_xp_backendslazy_xp_modules window_funcsr!r5r rSrrZrrgr1rnrrrrrrr-rrr0r!rr#r(r.r>r9rHrAr"rJr/rUr#r`rurmrrrrrrrr2rr"r&rr3r?rMr+rr.rrrrr_winstrrr)rrrs000rKrs  *66D ;;//KK11) 47##$ 0 0% 0 7##$TT%T7##$77%7,7))*KK+K(7>>"J:J:#J:Z7>>"77#7"7>>"LL#L  0(7??#?O?O$?OH ( ) !JK "78!KL A B45 A B !JK3:7&&'<(<7??#88$8,7##$LL%L07))* J J+ J7**+ Y Y, Y7??#66$6(7<< 00!0.7>>"LL#L:7001-L-L2-L`7??#YY$Y*7>>"LL#L$7>>" T T# TE./E56E01  U/0  U67 U/0E34E:;E34  #  "!" %#$ B%& +'( F)* :+. $ /3 :7==!$$"$@HIADOQ\C]^:<\_GHJqrVYUWY[VXabdkl|~J(L(D<F<`PaPcPWQXQ  7<< DADA!DAN7??#EE$EB7%%&_3_3'_3D7++,1616-16h7>>"&&#&B,'QR$0 K WWk:KP?=S?=D:@@ ,  Z/2F;:7??# S$ SsVs 5VV