bi 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 jGd1d2ZGGd3d4ZHee jGd5d6ZJee jGd7d8ZLid9egd:d;egd<d=egd>d?egd@dAegdBdCegdDdEegdFdGegdHdIegdJdKgdLdMgdNdOgdPdQgdRdSgdTdUgdVdWgdXdYgdZgd[gd\d]ZMee jGd^d_ZOgd`gdagdbdcgfgddgdegdfggdgfgdhgdigdjgdkgdlggdmfgdngdogdpgdqggdrfdsZPee jGdtduZRee jGdvdwZTee jGdxdyZUedzd{|ej<jd}e"Dcgc]\}}eeWe |||c}}d~ZXee dZYgdZZej<jdeZDcgc]}ee|d|c}ee dZ[ee jdZ\ycc}}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 Z/home/cdr/jupyterlab/.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%g@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:)rrrrt)rJr:rzmsgs 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_basicSsqR0 $T*,**#6 7 qdr: $M+-**#6 7 q%B7 $8?Azz#K LrMNrNrrMrKrrPs 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_basicfsq$26 #A T VqR0 #DBJJ W Yqe; #?rzz R TrMNrNrrMrKrrcsTrMr)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:rzr 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 rx)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)rr)Kmaxrr,r0rCr1r2r5r) rr&r isinstancerndarrayrrr3)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+rr4skip_xp_backendsr;rrMrKrrs2K-*0:d#A$ArMrceZdZdZdZy) TestLanczoscttjdd||jgd|jdttjd||jgd |jdttjd d ||jgd |jdy) Nr7Fr8rqϾv?;Fv?rFrBrAr<& .>r_rCruH%?;9?rFrErrBT)rrArBrFrBrAr)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)rr3rv)rJr:ns rKtest_array_sizezTestLanczos.test_array_sizes_ EA??1%B7==a@AE EE??1$26<r>s #2ErMr>ceZdZdZdZeejdZeejdZ eejdZ eejdZ eejdZej"j%dd d ej(d geej*d Zd Zeej0dZeejdZeej6dZeej:dZeej>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 rxrJr:rs rK test_boxcarzTestGetWindow.test_boxcar7sP   x 32<<?+   {B2 62<<?+rMctj5tjddtt j ddd|}dddt |jt|jd y#1swY7xYw) Nrr)r-rrFrrr<rr) rrrrrrrrGrrHrVs rK test_cheb_oddzTestGetWindow.test_cheb_odd@sw  $ $ & O  # #7 F""#3R2NA O " rzz-rzzz:A  O Os 5BB ctj5tjddtt j ddd|}dddt |jtdy#1swY+xYw) Nrr)r-rrFrrrr) rrrrrrrrGrrVs rKtest_cheb_evenzTestGetWindow.test_cheb_evenJsk  $ $ & 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_dpssRs4!!+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_floatXs\!!#rb1~~b#u4 ": D$RXXdjj-A-E-E F D$ 'rMNxr0g@r5ctjtd5tjd||dddy#1swYyxYw)Nz^Parameter Nx=.*rr/rC)pytestrrrr)rJrfr:s rKtest_invalid_parameter_NXz'TestGetWindow.test_invalid_parameter_NXcs:]]:-? @ 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')rhrrrrr)rJr:xp_s rKtest_invalid_inputsz!TestGetWindow.test_invalid_inputsjs ]]:-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_periodic~s  4 :q%BGO #7rBEJ #7DRPRWX  "5 :qR8%@ :q$2FN #6bA5I #65RP rMcd}|jd}tjd|dz|}t|rdnd}t t |5t ||jd|z| dddy#1swYyxYw) N)r#g @r'rCz ^window mustz^window.shape=rr)window)arangerrrrrrrv)rJr:osfactorsigr mesgs rKtest_array_as_windowz"TestGetWindow.test_array_as_windowstiin  (a-BG!(~2B :T 2 ? S#))A,1# > ? ? ?s A77Bc ttd|jgdfd|jgd|jttd|jgdfdd|jgd|jt j t 5tdgdfd| dddy#1swYyxYw) Nr>r;rr=r<F)rs)rU皙?rrUrC)r rrGrHrhrrrIs 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)rr8rr9r)rJr:r s rKtest_xp_defaultzTestGetWindow.test_xp_defaultsSA&#rzz***A"-|!#rzz2 222rMN)"rOrPrQ__doc__rRrrrrWr-rYr[r&rar#rerhmark parametrizerrHr/riror rur}r>rrHrr3rrrrMrKrLrLs46(w~~&,',w' ( w'L(Lw||$$%$ w~~&('( [[TBZRZZ]#;<w||$2%=2 F(w''() w~~&?'?w--.F/Fw../Q0Qw'6(63rMrLz 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 Tr8rBFr0 r7rg@r:ir<r%rFrg)\(?r^r_rH)getattrrrrhskiprrrrrrrr rvrrr rGrHr=allrimag zeros_like)rx window_nameparamsr:w1r{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_paramsr s'@ j*faB?@rM)r!r rrrrr.rr"rrr1r0poissonr2tuktriangler3rz window,winstrct|r|dvrtj|dt|d|}|jddk(sJy)N)r1rrrBrCr)rrhrrrv)r:rxrr s rKtest_not_needs_paramsr.sJbzf * 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_symmetricr=s S  q!&&rvva$q'k*+rzz#EER  rvva$q'k*+rzz#EER SrM)]rrnumpyrrrhrr 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`rtrmrrrrrrr2rr!r&rr3r>rLrrrr_winstrrr)rrrs000rKrsP  *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`YY*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'_3D,'QR$0 K WWk:KP?=S?=D:@@ ,  Z/2F;:7??# S$ SsVs "U;U