o
    5ήc                     @   s  d dl Z d dlZd dlmZ d dlmZmZmZmZm	Z	m
Z
mZ d dlmZ d dlmZ d dlmZmZmZmZ g dZG dd	 d	ZG d
d dZG dd dZG dd dZG dd dZG dd dZG dd dZeg dZeg dZ G dd dZ!eg deg deg deg deg ddeg ddeg deg dddd Z"d!d" Z#G d#d$ d$Z$G d%d& d&Z%G d'd( d(Z&G d)d* d*Z'G d+d, d,Z(G d-d. d.Z)G d/d0 d0Z*G d1d2 d2Z+G d3d4 d4Z,G d5d6 d6Z-G d7d8 d8Z.i d9eg d:d;eg d<d=eg d>d?eg d@dAeg dBdCeg dDdEeg dFdGeg dHdIeg dJdKg dLdMg dNdOg dPdQg dRdSg dTdUg dVdWg dXdYg dZg d[g d\d]Z/G d^d_ d_Z0g d`g dagdbdcgfg ddg deg dfgg dgfg dhg dig djg dkg dlgg dmfg dng dog dpg dqgg drfdsZ1G dtdu duZ2G dvdw dwZ3dxdy Z4dzd{ Z5d|d} Z6d~d Z7dd Z8dS )    N)array)assert_array_almost_equalassert_array_equalassert_allcloseassert_equalassert_assert_array_lesssuppress_warnings)raises)fft)windows
get_windowresamplehann))boxcar )triangr   )parzenr   )bohmanr   )blackmanr   )nuttallr   )blackmanharrisr   )flattopr   )bartlettr   )barthannr   )hammingr   )kaiser   )dpss)   )gaussian      ?)general_gaussian)      ?r    )chebwinr   )cosiner   )r   r   )exponentialr   )taylorr   )tukeyr"   c                   @      e Zd Zdd ZdS )TestBartHannc                 C   sF   t tjdddg d t tdg d t tddg d d S )	N   Tsym)r   gA,q?	5xB$?r0   gA,q?r      )r   HzG?\(\?      ?r3   r2   r   F)r   r2   r3   r4   r3   r2   )r   r   r   selfr   r   F/tmp/pip-target-vg8gfxp4/lib/python/scipy/signal/tests/test_windows.py
test_basic*   s   
zTestBartHann.test_basicN__name__
__module____qualname__r8   r   r   r   r7   r,   (       r,   c                   @   r+   )TestBartlettc                 C   sB   t tdg d t tdg d t tddg d d S )Nr-   )r   皙?皙?r@   r?   r   r1   )r   UUUUUU?UUUUUU?r4   rB   rA   r   F)r   rA   rB   r4   rB   rA   )r   r   r   r5   r   r   r7   r8   6   s
   zTestBartlett.test_basicNr9   r   r   r   r7   r>   4   r=   r>   c                   @   r+   )TestBlackmanc                 C   sn   t tjdddg ddd t tjdddg dd	d t tdg d
dd t tddg ddd d S )Nr-   Fr.   )r   p=
ף?)\(?r4   rE   rD   +=atolr1   )r   Fi'?b@c?Zgs?rK   rJ   rI   g:0yE>)r   쳩ֲ?,,?rM   rL   r   T)r   rD   rE   r4   rE   rD   r   )r   r   r   r5   r   r   r7   r8   ?   s   

zTestBlackman.test_basicNr9   r   r   r   r7   rC   =   r=   rC   c                   @   r+   )TestBlackmanHarrisc                 C   ^   t tddg d t tjdddg d t tdg d t tjdddg d	 d S )
Nr-   F)iUMu?0_^}?縯猨?r4   rR   rQ   r1   r.   )rP   gjP?g'm$M?gu?gu?g*m$M?gjP?)rP   ^?gU׈g?gU׈g?rS   rP   T)rP   rQ   rR   r4   rR   rQ   rP   )r   r   r   r5   r   r   r7   r8   P      
zTestBlackmanHarris.test_basicNr9   r   r   r   r7   rN   N   r=   rN   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )
TestTaylorc                 C   sT   t tdddd t tdddtg d t tdddtg d d	S )
zTests windows of small length that are normalized to 1. See the
        documentation for the Taylor window for more information on
        normalization.
        r   r       r4      )%6'HA?)T
?r4   rY   rX   r-   )δX?繡1E?PD3m?r\   r[   rZ   Nr   r   r)   npr   r5   r   r   r7   test_normalizeda   s   zTestTaylor.test_normalizedc                 C   sH   t tjdddddtg d t tjdddddtg d d	S )
zTest windows of small length that are not normalized to 1. See
        the documentation for the Taylor window for more information on
        normalization.
        rW   r    rV   Fnorm)	P ?sCp?g)8ux?rc   rb   r-   ))S~?r4   kt#?re   r4   rd   Nr]   r5   r   r   r7   test_non_normalizeds   s   zTestTaylor.test_non_normalizedc           
   	   C   s   d}d}t j|ddddd}t||}dtt|t|  }tt|dk}t|||  }d	t|d
k | | }d	t|dk | | }	t	|ddd t	|ddd t	|	ddd dS )a  This 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
        i   i      #   F)nbarsllra   r/      r   r    gX$g_2gAfAr   rG   gLJ?皙?gZӼ@N)
r   r)   r   r^   log10absamaxargmaxdiffr   )
r6   M_winN_fftwfspec
first_zeroPSLLBW_3dBBW_18dBr   r   r7   test_correctness   s   
zTestTaylor.test_correctnessN)r:   r;   r<   r_   rf   r{   r   r   r   r7   rU   _   s    rU   c                   @   r+   )
TestBohmanc                 C   F   t tdg d t tjdddg d t tddg d d S )	Nr-   )r   g$I ?6 D?r~   g I ?r   r1   Tr.   )r   PG?4}|?r4   6}|?r   r   F)r   r   r   r4   r   r   )r   r   r   r5   r   r   r7   r8      s   
zTestBohman.test_basicNr9   r   r   r   r7   r|      r=   r|   c                   @   r+   )
TestBoxcarc                 C   sB   t tdg d t tdg d t tddg d d S )Nr-   r   r   r   r   r   r   r1   r   r   r   r   r   r   r   F)r   r   r   r5   r   r   r7   r8      s   zTestBoxcar.test_basicNr9   r   r   r   r7   r      r=   r   )5CV?CԳ ?Q&E?:*?;Bu?9#?=[?Z
G?ٵݒ?Lg?(+ ?T=? zR&5?scH?u?]/M`?xC8???9?mo?[?T:?sK!q?/M.?×?)s?r4   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   )6?2?w.v?f*#?灖`?d${!??M?ut\?Ljh?xak?WY?qj?<)[?@ի?MEs?甽/?{l?[^6?:Ug$?: U\?I.!v?Clp?gF?稦$p?VLy?.?r4   r4   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   c                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )TestChebWinc                 C   s   t  G}|td ttddg d ttddg d ttddg d ttddg d	 ttddd
g d W d    d S 1 sMw   Y  d S )NThis window is not suitabler-   d   )}	ɺ?L >?r4   r4   r   r   r1   )S?D1mPC?%5R?r4   r   r   r   
   )r4   TEm?	-o?r   r   r4   )r4   F3?NO?	r]?r   r   r4   F)r4   r   r   r   r   r   )r	   filterUserWarningr   r   r&   )r6   supr   r   r7   r8      s$   "zTestChebWin.test_basicc                 C   R   t  }|td tjddd}W d    n1 sw   Y  t|tdd d S )Nr   5   atrg   decimal)r	   r   r   r   r&   r   cheb_odd_true)r6   r   cheb_oddr   r   r7   test_cheb_odd_high_attenuation   
   z*TestChebWin.test_cheb_odd_high_attenuationc                 C   r   )Nr   6   (   r   rg   r   )r	   r   r   r   r&   r   cheb_even_true)r6   r   	cheb_evenr   r   r7   test_cheb_even_high_attenuation   r   z+TestChebWin.test_cheb_even_high_attenuationc                 C   ^   t g d}t }|td tjddd}W d    n1 s!w   Y  t||dd d S )N)r4   m?燿&k?gc[?r   r   r4   r   r1   r   r   rg   r   r   r	   r   r   r   r&   r   )r6   cheb_odd_low_at_truer   r   r   r   r7   test_cheb_odd_low_attenuation     z)TestChebWin.test_cheb_odd_low_attenuationc                 C   r   )N)r4   R?N(D!T?#R?r   r   r   r4   r      ir   rg   r   r   )r6   cheb_even_low_at_truer   r   r   r   r7   test_cheb_even_low_attenuation  r   z*TestChebWin.test_cheb_even_low_attenuationN)r:   r;   r<   r8   r   r   r   r   r   r   r   r7   r      s    	r   )|dy?vS?{?r4   r   )1B?%?r   r   )́R?8,6V?r4   r   )\Kr?
oh?r   r   )r   r   r4   r   r   )r   r   r4   r   r   ))rg   N皙?F)rg   Nr   T)rg   Nr4   F)rg   Nr4   T)rg   r    r   F)rg   r    r   T)rg   r    r4   F)rg   r    r4   T)rW   Nr   T)rW   Nr4   T)rW   r    r   T)rW   r    r4   Tc                  C   sL   t  D ]\} }|d u rtttjg| R   qtj|  }t||dd qd S NrF   rtol)exponential_dataitemsassert_raises
ValueErrorr   r(   r   )kvwinr   r   r7   test_exponential6  s   
r   c                   @   r+   )TestFlatTopc                 C   ^   t tjdddg d t tjdddg d t tdg d t tddg d	 d S )
Nr-   Fr.   )bF;s*?dR	p^?r4   r   r   r1   )r   `p|ܢ?J?r  r  r   )r   <#U_&k?r  r  r   T)r   r   r   r4   r   r   r   )r   r   r   r5   r   r   r7   r8   A  s   
zTestFlatTop.test_basicNr9   r   r   r   r7   r   ?  r=   r   c                   @   r+   )TestGaussianc                 C   s^   t tddg d t tddg d t tddg d t tddd	g d
 d S )Nr-   r4   )|~?w$?"$j=?r  r  r  r1   g333333?)r  
"?;eܜ?r4   r
  r	  r     )r   ?yLE?r4   r  r  r   F)r   r  r  r4   r  r  )r   r   r!   r5   r   r   r7   r8   U  s   zTestGaussian.test_basicNr9   r   r   r   r7   r  S  r=   r  c                   @   r+   )TestGeneralCosinec                 C   s<   t tdg dg d t tjdg dddg d d S )NrW   r#   333333?r   )r?   r  r   r  r?   rg   Fr.   r?   r  r   r  )r   r   general_cosiner5   r   r   r7   r8   j  s   zTestGeneralCosine.test_basicNr9   r   r   r   r7   r  h  r=   r  c                   @   r+   )TestGeneralHammingc                 C   sN   t tddg d t tjddddg d t tjddd	dg d
 d S )NrW   ffffff?r?   r  r4   r  r?         ?Fr.   )r#   y&"?x?r  r  r-   T)r#   r  r  r  r  r#   )r   r   general_hammingr5   r   r   r7   r8   s  s   zTestGeneralHamming.test_basicNr9   r   r   r   r7   r  q  r=   r  c                   @   r+   )TestHammingc                 C   rO   )
Nr-   F){Gz?ףp=
?p=
ף?r4   r  r  r1   r.   )r  CW4?	/5?4Cъ?r   r  r  )r  6 iv?P0?r"  r!  r  T)r  r  r  r4   r  r  r  )r   r   r   r5   r   r   r7   r8     rT   zTestHamming.test_basicNr9   r   r   r   r7   r  ~  r=   r  c                   @   r+   )TestHannc                 C   s^   t tjdddg d t tjdddg d t tddg d t tdg d	 d S )
Nr-   Fr.   )r         ?r  r4   r  r$  r1   )r   \$:?H`r?8T^j?r'  r&  r%  Tr   Xhd?Tͻ?r*  r)  r   r   r$  r  r4   r  r$  r   )r   r   r   r5   r   r   r7   r8     s   
zTestHann.test_basicNr9   r   r   r   r7   r#    r=   r#  c                   @   r+   )
TestKaiserc                 C   st   t tddg d t tddg d t tddg d t tddg d t tddd	g d
 d S )Nr-   r#   )l?<3N?i>%?r/  r.  r-  r1   )r-  	$?6?r4   r1  r0  r-  g@)§Aը?O F?$?r4  r3  r2  )r2  *'?=a?r4   r6  r5  r2  F)r2  r5  r6  r4   r6  r5  )r   r   r   r5   r   r   r7   r8     s   zTestKaiser.test_basicNr9   r   r   r   r7   r,    r=   r,  c                   @      e Zd Zdd Zdd ZdS )TestKaiserBesselDerivedc                 C   s   d}t j|dd}t jd|dd}t|| t|d |d  d || d d  d  d tt jdtjd dd d	 tdd  tt jd
tjd dd d ddg tt jdtjd dd d g d d S )Nr         @beta)zkaiser bessel derivedr9  Ffftbinsr    r4   r   rg   gU?gU3|\?r-   r  )gHZ1?gKf?g"iF5?)r   kaiser_bessel_derivedr   r   r^   pisqrt)r6   Mrt   w2r   r   r7   r8     s    
0z"TestKaiserBesselDerived.test_basicc                 C   s   d}d}t t|d tj|d dd W d    n1 sw   Y  d}t t|d tj|d ddd	 W d    d S 1 sAw   Y  d S )
Nr   zHKaiser-Bessel Derived windows are only defined for even number of pointsmatchr   r9  r:  zCKaiser-Bessel Derived windows are only defined for symmetric shapesF)r;  r/   )r   r   r   r>  )r6   rA  msgr   r   r7   test_exceptions  s   "z'TestKaiserBesselDerived.test_exceptionsN)r:   r;   r<   r8   rF  r   r   r   r7   r8    s    r8  c                   @   r+   )TestNuttallc                 C   r   )
Nr-   Fr.   )A"7?v;O<g?$2bUs?r4   rJ  rI  r1   )rH  gx&yxXW?g??絑?rK  g??g&yxXW?)rH  gEJ?g=`T?g>`T?gEJ?rH  T)rH  rI  rJ  r4   rJ  rI  rH  )r   r   r   r5   r   r   r7   r8     s   
zTestNuttall.test_basicNr9   r   r   r   r7   rG    r=   rG  c                   @   r+   )
TestParzenc                 C   r}   )	Nr-   )e/?r$  88?rN  r$  rM  r1   Tr.   )%OZQ%w?2|&?݁?r4   rQ  rP  rO  F)rO  rP  rQ  r4   rQ  rP  )r   r   r   r5   r   r   r7   r8     s   
zTestParzen.test_basicNr9   r   r   r   r7   rL    r=   rL  c                   @   r+   )
TestTriangc                 C   sF   t tddg d t tdg d t tjdddg d d S )	Nr-   T)UUUUUU?r#   竪?rT  r#   rS  r1   )r$  r#   r  r   r  r#   r$  Fr.   )r$  r#   r  r   r  r#   )r   r   r   r5   r   r   r7   r8     s   
zTestTriang.test_basicNr9   r   r   r   r7   rR    r=   rR  )rg   r#   T)        r4   r4   rU  )rg   ?T)rU  ?rW  rU  )rg   r4   T)rU  r  r  rU  )rg   r#   F)rU  r4   r4   r4   )rg   rV  F)rU  rnOC?r4   rX  )rg   r4   F)rU  r#   r4   r#   )rW   rU  T)r4   r4   r4   r4   r4   )rW   r@   T)rU  Yx?r4   rY  rU  )rW   r4   T)rU  r#   r4   r#   rU  )r-   r   r   )r1   r   r   )r-   r$  )r   r   r   r   r   r   )r1   r$  )r   r   r   r   r   r   r   )r-   )r   r*  r4   r4   gRͻ?r   )r1   )r   r  r4   r4   r4   r  r   )r-   r  )r   gxų	&?r4   r4   gvų	&?r   )r1   r  )r   g#}`yq?)!?r4   rZ  g#}`yq?r   r(  r+  ))r-   r   )r1   r   c                   @   r7  )	TestTukeyc                 C   sL   t  D ]\}}|d u rtttjg|R   qtj| }t||dd qd S r   )
tukey_datar   r   r   r   r*   r   )r6   r   r   r   r   r   r7   r8   -  s   
zTestTukey.test_basicc                 C   sD   t dd}t d}t|| t dd}t d}t|| d S )Nr   r   r   )r   r*   r   r   r   )r6   tuk0box0tuk1han1r   r   r7   test_extremes6  s   


zTestTukey.test_extremesN)r:   r;   r<   r8   ra  r   r   r   r7   r[  +  s    	r[  )IP?;]L?rc  rb  )g/vI<t?gE@?gE@̿g/vI<tgV?g$aȲ`?)ыBA?gnjk?rd  )g>ef?rU  g>ef)	#n?gLore  )gm/ ?g96+?g雦br?)g w?V?g>J#?g;Wi?g֙}I#?gr?V?)g:7A!(?cHX?rU  cHXS[@!(ܿ)qh4?T1a~²?gEܿrj  ri  )rf  rh  rU  gS[@!(?rg  )gu+t?X>߿g36?rk  g;St?)gML?gh5"z?g\v4?gPa?gd?)dsk8Si?p&p?5Bu?OL{?08À?w-<?kF?^R?kPz?1?Y^Q?X9?dkѐ^ś?3aJ?)IQ?-?ѫ"?給.?~9P?~n[?8Wϭ?	%fҳ?!KI?PJd?/ĳ?$~s	?pkuP?%ꙷ?
?q*?%wo?Zo??矐Ӄ?磓B@%?g3?/ߗ#@?0?@?=?.F!?it3?:.?X.?GkNVs?玃"Hk? 'F?精`?iGU?#!5#?r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r~  r}  r|  r{  rz  ry  rx  rw  rv  ru  rt  rs  rr  rq  rp  ro  rn  rm  rl  )dgUw|?gTP?gj?gwĢ?gDъO?g_{?g< ?gȁk?gꬃt?g(P;?gHñ?gBKXP?g.mˏ޴?g nik?g"=?gOu?g{?ggU?g/u?gN׏$s?g ]?gГE?goY*?gܛ?gbu?g`?gɪJS?ga`gT*r?g.{?gYn?g.|TK?g紸?gzþ?gfU?g,?gԙ=?gg'?gws䕽?g?g[\	?g#k	?g%*?gLw<w?g,h-A?g[::<?g199l?g.4䓦%?gG/?gƢhlO?g[$v?g[$vgƢhlOgG/g.4䓦%g199lg[::<񍭿g,h-AgLw<wg%*浿g#k	g[\	g㻿gws䕽gg'gԙ=g,gfUgzþg紸¿g.|TK¿gYn¿g.{¿ga`gT*r¿gɪJS¿g`¿gbugܛgoY*gГEg ]gN׏$sg/uggUg{캿gOug"=g nikg.mˏ޴gBKXPgHñg(P;gꬃtgȁkg< g_{gDъOgwĢgjgTPgUw|)d|W◰?煢	͓??
8ρ?\i?ٮ32A?!;f?+?3sXL2??J?5QG?Ao?W? }2?␉k?IǇ?F6?f?ch'?%"a?h/K?:?>}?DC1?XB?QST`?s\NK?
3?<s?o?@1O?5M?#$7?,9ṗ?C)?W]{?#Ys臈En꛿x;i6
v^
m-b3[Ehp[6a0_:WR~戺0ъҺr  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  )dgI!w.?gMu?g%Pld?g3z?g>?gd?g!DKy?g+o ?gYh{b?g{^y?gSh1?gy%@?g|hzj??gP\D?gz	?gVx<҅?g?gq&?gb˱?g8|p?g_cH?g6K	?gwp΍?g"Q	q?g@m8Migǐ3g7/`_qgS!g8QJg68gu\qg/Z~ܳgpQZgCg3`g7<w䷿ggp鷿g<Wulg~`gTg#qvg,]Bigw|gvľg3gagLHgU,gvA]PvgvA]Pv?gU,?gLH?ga?g3?gvľ?gw|?g,]Bi?g#qv?gT?g~`?g<Wul?gp?g?g7<w?g3`?gC?gpQZ?g/Z~ܳ?gu\q?g68?g8QJ?gS!?g7/`_q?gǐ3?g@m8Mi?g"Q	qgwp΍g6K	g_cHg8|pgb˱gq&ggVx<҅gz	gP\Dg|hzj?gy%@¿gSh1ÿg{^yÿgYh{bĿg+o Ŀg!DKyſgdſg>ſg3zĿg%PldĿgMuÿgI!w.ÿ)g?g"?gy?'?g `?))rg   rl   r    )r  gffffff?r  )rW   r%   rW   )r   r    rg   c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestDPSSc                 C   sT   t  D ]#\}}tj|ddi\}}t||d d|d t||d dd|d qd S )	Nreturn_ratiosTr   gHz>)rH   err_msgr   gh㈵>)r   rH   r  )	dpss_datar   r   r   r   )r6   r   r   r   ratiosr   r   r7   r8   N  s
   zTestDPSS.test_basicc                 C   s   t ddD ]e}t||d }|d }tt|d |d|f d tj||d dd	}|dkrHtt|d |d|f d t||d
d tj||d dd	}|dkrXdnd}tt|d |d|f d qd S )Nr      g @r    r4   z%s)r  	subsampler`   gQ?r   r   )ranger   r   r   r^   isclosesumr   )r6   rA  r   expectedwin_subwin_2r   r   r7   
test_unityU  s$   zTestDPSS.test_unityc                 C   sd   t jdddddd }t|d t jdddddd }t|d t jdd	dddd }t|d d S )
N   r-   rg   T)r  r   r4   r1   r   )r   r   r   )r6   lamr   r   r7   ra  j  s   

zTestDPSS.test_extremesc                 C   s   t ttjddd t ttjddd t ttjddd t ttjddd t ttjddd t ttjddd t ttjddd d S )	Nrg   r%   g?r  r   r   )r   r   r   r   	TypeErrorr5   r   r   r7   test_degenerates  s   zTestDPSS.test_degenerateN)r:   r;   r<   r8   r  ra  r  r   r   r   r7   r  L  s
    	r  c                   @   sT   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd ZdS )TestGetWindowc                 C   s<   t dd}t|t| t dd}t|t| d S )Nr      )r      )r   r   r   r^   	ones_like)r6   rt   r   r   r7   test_boxcar  s   zTestGetWindow.test_boxcarc                 C   T   t  }|td tjdddd}W d    n1 sw   Y  t|tdd d S )Nr   )r&   r   r   Fr<  rg   r   )r	   r   r   r   r   r   r   r6   r   rt   r   r   r7   test_cheb_odd  
   zTestGetWindow.test_cheb_oddc                 C   r  )Nr   )r&   r   r   Fr<  rg   r   )r	   r   r   r   r   r   r   r  r   r   r7   test_cheb_even  r  zTestGetWindow.test_cheb_evenc                 C   s.   t jdddd}t dd}t||dd d S )N)r   r  @   Fr<  r  rg   r   )r   r   r   r   r6   win1win2r   r   r7   	test_dpss  s   zTestGetWindow.test_dpssc                 C   s(   t dd}t ddd}t|| d S )Ng@r  F)r   r   r   r   r  r   r   r7   test_kaiser_float  s   zTestGetWindow.test_kaiser_floatc                 C   s(   t ttjtdd t ttjdd d S )Nr   r   brokenrg   )r   r   r   r   setr5   r   r   r7   test_invalid_inputs  s   z!TestGetWindow.test_invalid_inputsc                 C   sf   d}t d}td|d }ttdd t|t|| |d W d    d S 1 s,w   Y  d S )N   )r   g       @r    zmust have the same lengthrC  )window)r^   aranger   r   r   r   r   len)r6   osfactorsigr   r   r   r7   test_array_as_window  s   
"z"TestGetWindow.test_array_as_windowc                 C   s@   t tdg dfdg d t tdg dfdddg d d S )Nr  r  rg   r  Fr<  )r?   皙?r  r?   r   r   r5   r   r   r7   test_general_cosine  s   z!TestGetWindow.test_general_cosinec                 C   s0   t tddg d t tddddg d d S )N)r  r  rW   )r?   жn?男*?r  r  Fr<  r  r  r5   r   r   r7   test_general_hamming  s   
z"TestGetWindow.test_general_hammingN)r:   r;   r<   r  r  r  r  r  r  r  r   r  r   r   r   r7   r  ~  s    	r  c               	   C   sh  t D ]\} }tt| }t }|td |dg|R ddi}|dg|R ddi}t|d d | |dg|R ddi}|dg|R ddi}t|d d | tt|d	g|R ddid	 tt|d	g|R ddid	 tt|dg|R ddid tt|dg|R ddid t	t
|d
g|R   t	t
|dg|R   t|dg|R ddig  t|dg|R ddig  t|dg|R ddidg t|dg|R ddidg t|dg|R ddijdk t|dg|R ddijdk t|dg|R ddijdk t|dg|R ddijdk t|d	g|R ddijdk t|d	g|R ddijdk t|dg|R ddid t|dg|R ddid t|dg|R ddid t|dg|R ddid tt|dg|R ddijddd tt|dg|R ddijddd W d    n	1 sw   Y  qd S )Nr   r   r/   Tr1   Fr  	   r-   g      @ir   r   floatr   g)\(?rF   rG      )window_funcsgetattrr   r	   r   r   r   r   r  r   r   r   r   dtyper   r   r   imag)window_nameparamsr  r   w1rB  r   r   r7   test_windowfunc_basics  sN   

          r  c                  C   s   dD ]	} t tt| d qd S )N)r   ksrr>  kbdr!   gaussgsszgeneral gaussianr$   zgeneral gaussgeneral_gaussggsdssr   zgeneral cosiner  r&   chebzgeneral hammingr  r1   )r   r   r   )winstrr   r   r7   test_needs_params  s   r  c                  C   s&   dD ]} t | d}tt|d qd S )N)r   r   r   r   r   r   r'   r   r   r   r   r)   r(   poissonr*   tuktriangler1   )r   r   r  )r  r   r   r   r7   test_not_needs_params  s   
r  c                   C   s0   t jd urtdt jv  tdtjjv d S d S )Nzsignal.hann is deprecated
deprecated)dep_hann__doc__r   r   r   r   r   r   r7   test_deprecation  s   
r   c                  C   s    t t t} t| tu  d S )N)pickleloadsdumpsr  r   )	dep_hann2r   r   r7   test_deprecated_pickleable  s   r%  )9r!  numpyr^   r   numpy.testingr   r   r   r   r   r   r	   pytestr
   r   	scipy.fftr   scipy.signalr   r   r   r   r  r  r,   r>   rC   rN   rU   r|   r   r   r   r   r   r   r   r  r  r  r  r#  r,  r8  rG  rL  rR  r\  r[  r  r  r  r  r  r  r   r%  r   r   r   r7   <module>   s    $	J
9






		)	
("2>3