Ë
|
Kñúh± ã ó| d Z ddlZddlmZmZ ddlmZ G d d« Z G d d« Z G d d
|
« Z G d d « Z
|
y) zfTest functions for fftpack.helper module
|
|
Copied from fftpack.helper by Pearu Peterson, October 2005
|
|
é N)ÚfftÚpi)Úassert_array_almost_equalc ó* e Zd Zd Zd Zd Zd Zd Zy)Ú TestFFTShiftc ó g d¢}g d¢}t t j |« |« t t j |« |« g d¢}g d¢}t t j |« |« t t j |« |« y )N© 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 ÚfftshiftÚ ifftshift)ÚselfÚxÚys úNH:\Change_password\venv_build\Lib\site-packages\numpy/fft/tests/test_helper.pyÚtest_definitionz TestFFTShift.test_definition s` Ú +Ú +Ü!¤#§,¡,¨q£/°1Ô5Ü!¤#§-¡-°Ó"2°AÔ6Ú /Ú /Ü!¤#§,¡,¨q£/°1Ô5Ü!¤#§-¡-°Ó"2°AÕ6ó c ó¶ dD ]T }t j j |f« }t t j t j
|
|« « |« V y )N)r
|
r é éd éÓ )ÚnpÚrandomr r r r )r Únr s r Ú test_inversezTestFFTShift.test_inverse sD Ø$ò IAÜ × Ñ ! Ó&AÜ %¤c§m¡m´C·L±LÀ³OÓ&DÀaÕ Hñ Ir c ó g d¢g d¢g d¢g}g d¢g d¢g d¢g}t t j |d¬« |« t t j |d ¬« t j |d
|
¬« « t t j |d¬« |« t t j |d ¬« t j |d
|
¬« « t t j |« |« t t j |« |« y ) N)r r
|
r )r r r )r r r )r r r )r r r
|
)r r r ©r r
|
©Úaxesr ©r r )r ÚfreqsÚshifteds r Útest_axes_keywordz TestFFTShift.test_axes_keyword s´ ÚJª Ð5Ú ¢ªJÐ7Ü!¤#§,¡,¨u¸6Ô"BÀGÔLÜ!¤#§,¡,¨u¸1Ô"=Ü"%§,¡,¨u¸4Ô"@ô Bä!¤#§-¡-°¸fÔ"EÀuÔMÜ!¤#§-¡-°¸aÔ"@Ü"%§-¡-°¸dÔ"Cô Eô "¤#§,¡,¨uÓ"5°wÔ?Ü!¤#§-¡-°Ó"8¸%Õ@r c óú ddgddgddgg}ddgddgddgg}t t j |d¬« |« t t j |d¬« |« t t j |d¬« |« t t j |dg¬« |« ddgddgddgg}t t j |d¬« |« t t j |d¬« |« ddgddgddgg}t t j |d ¬« |« t t j |d ¬« |« t t j |ddg¬« |« t t j |ddg¬« |« t t j |d
|
¬« |« t t j |d
|
¬« |« t t j |« |« t t j |« |« y
|
) z1 Test 2D input, which has uneven dimension sizes r r
|
r r r é r' r) r& Nr )r r* Ú
|
shift_dim0Ú
|
shift_dim1Úshift_dim_boths r Útest_uneven_dimsz TestFFTShift.test_uneven_dims) s® ð FØ FØ Fð
|
ð FØ FØ Fð
|
|
ô
|
"¤#§,¡,¨u¸1Ô"=¸zÔJÜ!¤#§-¡-°
|
ÀÔ"CÀUÔKÜ!¤#§,¡,¨u¸4Ô"@À*ÔMÜ!¤#§-¡-°
|
À!ÀÔ"EÀuÔMð FØ FØ Fð
|
|
ô
|
"¤#§,¡,¨u¸1Ô"=¸zÔJÜ!¤#§-¡-°
|
ÀÔ"CÀUÔKð FØ FØ Fð
|
ô
|
"¤#§,¡,¨u¸6Ô"BÀNÔSÜ!¤#§-¡-°ÀVÔ"LÈeÔTÜ!¤#§,¡,¨u¸A¸q¸6Ô"BÀNÔSÜ!¤#§-¡-°ÀaÈÀVÔ"LÈeÔTô "¤#§,¡,¨u¸4Ô"@À.ÔQÜ!¤#§-¡-°ÀTÔ"JÈEÔRÜ!¤#§,¡,¨uÓ"5°~ÔFÜ!¤#§-¡-°Ó"?ÀÕGr c
|
ór
|
ddl mmm m
|
d
|
fd }d
|
fd }t d« D ] }t d« D ]w }dD ]p }t j j ||« }t t j ||« |||« « t t j ||« |||« « r y y) zG Test the new (>=v1.15) and old implementations are equal (see #10073) r )ÚarangeÚasarrayÚ concatenateÚtakeNc ó
|
| « }|j }|t t |« « }nt |t « r|f}|}|D ]: }|j
|
| }|dz dz } ||« |« f« } |||« }< |S )z& How fftshift was implemented in v1.14r
|
r ©ÚndimÚlistÚrangeÚ
|
isinstanceÚintÚshape© r r( Útmpr: r Úkr# Úp2Úmylistr4 r5 r6 r7 s r Úoriginal_fftshiftz>TestFFTShift.test_equal_to_original.<locals>.original_fftshiftZ s ø á!*CØ88DØ|ÜE $KÓ(ÜD¤#Ô&Ø wØAØò 'ØIIaLØ!e \Ù$¡f¨R°£m±V¸B³ZÐ%@ÓAÙF AÓ&ð 'ð
|
Hr c ó
|
| « }|j }|t t |« « }nt |t « r|f}|}|D ]= }|j
|
| }||dz dz z
|
} ||« |« f« } |||« }? |S )z( How ifftshift was implemented in v1.14 r
|
r r9 r@ s r Úoriginal_ifftshiftz?TestFFTShift.test_equal_to_original.<locals>.original_ifftshiftj s ø á!*CØ88DØ|ÜE $KÓ(ÜD¤#Ô&Ø wØAØò 'ØIIaLØ!a% A Ñ%Ù$¡f¨R°£m±V¸B³ZÐ%@ÓAÙF AÓ&ð 'ð
|
Hr é )r r
|
Nr) r& )N) Ú numpy._corer4 r5 r6 r7 r< r! r" Úrandr r r r ) r rE rG ÚiÚjÚ axes_keywordÚinpr4 r5 r6 r7 s @@@@r Útest_equal_to_originalz#TestFFTShift.test_equal_to_originalV s° û çBÓB÷ ð ÷ ð ô$ rò UAÜ2Yò UØ$>ò ULÜ ))..¨¨AÓ.Cä-¬c¯l©l¸3À Ó.MÙ.?ÀÀ\Ó.RôTô .¬c¯m©m¸CÀ Ó.NÙ.@ÀÀlÓ.SõUñ Uñ Uñ Ur N)Ú__name__Ú
|
__module__Ú __qualname__r r$ r, r2 rO © r r r r s ò7òIò
|
Aò+HóZ/Ur r c ó e Zd Zd Zy)Ú TestFFTFreqc óT g d¢}t dt j d« z |« t dt z t j dt « z |« g d¢}t dt j d« z |« t dt z t j dt « z |« y )Nr r r é
|
)r r Úfftfreqr ©r r s r r zTestFFTFreq.test_definition sv Ú +Ü! !¤c§k¡k°!£nÑ"4°aÔ8Ü! !¤b¡&¬3¯;©;°q¼"Ó+=Ñ"=¸qÔAÚ /Ü! "¤s§{¡{°2£Ñ"6¸Ô:Ü! "¤r¡'¬C¯K©K¸¼BÓ,?Ñ"?ÀÕCr N©rP rQ rR r rS r r rU rU s óDr rU c ó e Zd Zd Zy)Ú TestRFFTFreqc óT g d¢}t dt j d« z |« t dt z t j dt « z |« g d¢}t dt j d« z |« t dt z t j dt « z |« y )N)r r
|
r r r r )r r
|
r r r r. rW )r r Úrfftfreqr rY s r r z TestRFFTFreq.test_definition sw Ú Ü! !¤c§l¡l°1£oÑ"5°qÔ9Ü! !¤b¡&¬3¯<©<¸¼2Ó+>Ñ">ÀÔBÚ Ü! "¤s§|¡|°BÓ'7Ñ"7¸Ô;Ü! "¤r¡'¬C¯L©L¸¼RÓ,@Ñ"@À!ÕDr NrZ rS r r r\ r\ s óEr r\ c ó e Zd Zd Zy)Ú
|
TestIRFFTNc ó t j j d« \ }}|d|z z }d}t j ||¬« y )N)r rH é é y ð?)r r' )r! r" r Úirfftn)r ÚarÚaiÚar( s r Útest_not_last_axis_successz%TestIRFFTN.test_not_last_axis_success s< Ü×!Ñ! .Ó1BØ bLàô
|
|
14Ö r N)rP rQ rR rh rS r r r` r` s ó!r r` ) Ú__doc__Únumpyr! r r Ú numpy.testingr r rU r\ r` rS r r ú<module>rl sF ðñó
|
ß Ý 3÷zUñ zU÷zDñ D÷Eñ E÷ !ò !r
|
