o 5╗BiL<у@s,dZddlmZmZddlmZddlmZddlZddl Z ddlZddlm Z ddlmZmZddlmZe ejаd бgZejjZgd вZddДZd dДZeddНGddДdГГZeddНGddДdГГZeddgdНGddДdГГZeddddgdНGdd Дd ГГZeddddgdНGd!d"Дd"ГГZdS)#zМIncludes test functions for fftpack.helper module Copied from fftpack.helper by Pearu Peterson, October 2005 Modified for Array API, 2023 щ)┌ next_fast_len┌_init_nd_shape_and_axes)┌assert_equal)┌raisesN)┌array_api_compatible)┌xp_assert_close┌SCIPY_DEVICE)┌fft┌skip_if_array_api)щщщщщщщ щ щiШicCstD] }tt|Г|ГqdS)N)┌_5_smooth_numbersrr)┌nйr·[/var/www/html/Trade-python/venv/lib/python3.10/site-packages/scipy/fft/tests/test_helper.py┌test_next_fast_lens rcCsМ|}|dkrJВ t|dГ\}}|dkrn|}q td|ddГD]} t||Г\}}|dkr/n|}q#q |dksDJd|Ыd|Ыd |ЫЭГВdS) NrFTrrщzx=z is not z-smooth, remainder=)┌divmod┌range)┌xr┌x_orig┌q┌r┌drrr┌_assert_n_smooth s(№№ r!T)┌np_onlyc@sNeZdZddДZddДZddДZejje j dkded d НddДГZd dДZ dS)┌TestNextFastLencCs\tjаdбddД}|ГD]}t|Г}t|dГ|t|dГks!JВt|dГ}t|dГq dS)Ni╥cssБtddГEdHdVdS)NriшiАy)rrrrr┌nums<sА z0TestNextFastLen.test_next_fast_len..numsщFTr)┌np┌random┌seedrr!)┌selfr$r┌mrrrr9s ·z"TestNextFastLen.test_next_fast_lencCsLtjtjtjtjtjtjg}|D]}|dГ}t|Г}t|tt |ГГГqdS)Ni90) r&┌int16┌int32┌int64┌uint16┌uint32┌uint64rr┌int)r)┌ITYPES┌itypr┌testNrrr┌test_np_integersHs¤z TestNextFastLen.test_np_integerscCsJddddddddddd d ddd dЬ}|абD]\}}tt|dГ|ГqdS)Nrrrr rrrщщщiщщ@ )rrrr rrщrщr6r7щi¤r9r:Tй┌itemsrrйr)┌hamsr┌yrrr┌testnext_fast_len_smallOs ■ z'TestNextFastLen.testnext_fast_len_smalllz$Hamming Numbers too large for 32-bitT)┌reasonr┌strictc CsмiddУddУddУddУddУddУd d УddУdd УddУddУddУddУddУddУddУddУdddddddddddd Ье}|абD]\}}tt|d!Г|ГqGdS)"Ni└╟hi┴╟hiАДi└=ui+Йы2i,Йы2iB3lА ╖lА ╖lАsЎ'╖lа>х9∙lб>х9∙l lf╪nlf╪nl>Uk▒ol°qЕ. l°qЕ. l╪xQ+t0 lв!═lв!═ldopL?═lэЕhRlьЕhRьЙ2 f;┘ьь`71Y!lрЕ("ь ь ╦єX l )rFlИ2 f;┘rGl rHl`71Y!rIl l ╦єX rJl! ╦єX Tr>r@rrr┌testnext_fast_len_bigWsd ■¤¤№№√√·∙° ў Ўї єЄф z%TestNextFastLen.testnext_fast_len_bigcCs,tdddНdks JВtdddНdksJВdS)Nr%T)┌realщr;F)┌targetrL)r)r)rrr┌test_keyword_args|sz!TestNextFastLen.test_keyword_argsN)┌__name__┌ __module__┌__qualname__rr5rC┌pytest┌mark┌xfail┌sys┌maxsize┌ ValueErrorrKrOrrrrr#6s■ "r#)┌cpu_onlyc@sleZdZddДZddДZddДZddДZd d ДZddДZd dДZ ddДZ ddДZddДZddДZ ddДZdS)┌Test_init_nd_shape_and_axesc CєF|аdб}d}d}d}g}t|||Г\}}||ksJВ||ks!JВdS)Nr rй┌asarrayrй r)┌xpr┌shape┌axes┌shape_expected┌ axes_expected┌ shape_res┌axes_resrrr┌test_py_0d_defaultsДє z/Test_init_nd_shape_and_axes.test_py_0d_defaultsc Cr[)Ng@rr\r^rrr┌test_xp_0d_defaultsСrgz/Test_init_nd_shape_and_axes.test_xp_0d_defaultsc CsL|аgdвб}d}d}d}dg}t|||Г\}}||ksJВ||ks$JВdS)Nйrrr)rrr\r^rrr┌test_py_1d_defaultsЮєz/Test_init_nd_shape_and_axes.test_py_1d_defaultsc CsL|аdddб}d}d}d}dg}t|||Г\}}||ksJВ||ks$JВdS)NrrчЪЩЩЩЩЩ╣?)r)┌arangerr^rrr┌test_xp_1d_defaultsлrkz/Test_init_nd_shape_and_axes.test_xp_1d_defaultsc CsV|аgdвgdвgб}d}d}d}ddg}t|||Г\}}||ks#JВ||ks)JВdS)N)rrrr )rrr;r)rr rrr\r^rrr┌test_py_2d_defaults╕s z/Test_init_nd_shape_and_axes.test_py_2d_defaultsc CsZ|аdddб}|а|dб}d}d}d}ddg}t|||Г\}}||ks%JВ||ks+JВdS)Nrrrl)rr)rm┌reshaperr^rrr┌test_xp_2d_defaults╞sz/Test_init_nd_shape_and_axes.test_xp_2d_defaultsc CsN|аgdвб}d}d}d}gdв}t|||Г\}}||ksJВ||ks%JВdS)Nйrrrrr йrrrrr й┌zerosrr^rrr┌test_xp_5d_defaults╘sz/Test_init_nd_shape_and_axes.test_xp_5d_defaultsc CsR|аgdвб}gdв}d}d}gdв}t|||Г\}}||ks!JВ||ks'JВdS)Nrr)rщ rwrr )rrrrr rsrtr^rrr┌test_xp_5d_set_shapeсsz0Test_init_nd_shape_and_axes.test_xp_5d_set_shapec CsR|аgdвб}d}gdв}d}gdв}t|||Г\}}||ks!JВ||ks'JВdS)Nrr)r rr)r rrrtr^rrr┌test_xp_5d_set_axesюsz/Test_init_nd_shape_and_axes.test_xp_5d_set_axesc CsV|аgdвб}gdв}gdв}d}gdв}t|||Г\}}||ks#JВ||ks)JВdS)Nrr)rrwr)rrr)rrrrtr^rrr┌test_xp_5d_set_shape_axes√sz5Test_init_nd_shape_and_axes.test_xp_5d_set_shape_axescCs<|аdб}t|dddН\}}|dksJВ|gdвksJВdS)N)rrr r)rrrйr`rarirt)r)r_rr`rarrr┌test_shape_axes_subsets z2Test_init_nd_shape_and_axes.test_shape_axes_subsetcCsФ|аdб}ttddНПt|dddgddggdНWdГn1s#wYttddНПt|dgdвdНWdГn1sBwYttd dНПt|ddgdНWdГn1s`wYttd dНПt|dd gdНWdГn1s~wYttddНПt|dddgdНWdГn1sЭwYttd dНПt|ddgddggddНWdГn1s└wYttd dНПt|gdвddНWdГn1s▀wYttddНПt|аgdвбgdвdgdНWdГn 1РswYttddНПt|dgddНWdГn 1Рs$wYttddНПt|d ddНWdГdS1РsCwYdS)Nrz-axes must be a scalar or iterable of integers)┌matchrrr r{)gЁ?g@g@g@z$axes exceeds dimensionality of inputщ■ zall axes must be uniquerz.shape must be a scalar or iterable of integerszBwhen given, axes and shape arguments have to be of the same length)rrrrriz1invalid number of data points $\[0\]$ specifiedz2invalid number of data points $\[-2\]$ specified)ru┌ assert_raisesrXr)r)r_rrrr┌test_errorssZ ■■ ■ ■ ■■■ ¤ ¤ $¤z'Test_init_nd_shape_and_axes.test_errorsN)rPrQrRrfrhrjrnrorqrvrxryrzr|rАrrrrrZБs rZ┌torchz1torch.fft not yet implemented by array-api-compat)┌reasonsc@s,eZdZddДZddДZddДZddДZd S) ┌TestFFTShiftcCs||аgdвб}|аgdвб}ttа|б|Гttа|б|Г|аgdвб}|аgdвб}ttа|б|Гttа|б|ГdS)N) чrrrr щ№ щ¤ r~rw) g└rЖr~rwrrrrr ) rДrrrr щ√ rЕrЖr~rw) g└rЕrЖr~rwrrrrr йr]rr ┌fftshift┌ ifftshift)r)r_rrBrrr┌test_definitionBszTestFFTShift.test_definitioncCs8dD]}|аtjа|fбб}ttаtа|бб|ГqdS)N)rr rщdщ╙)r]r&r'rr rКrЙ)r)r_rrrrr┌test_inverseLs■zTestFFTShift.test_inversecCs└|аgdвgdвgdвgб}|аgdвgdвgdвgб}ttj|ddН|Гttj|d dНtj|d dНГttj|ddН|Гttj|d dНtj|d dНГttа|б|Гttа|б|ГdS)N)rДrr)rr rЕ)rЖr~rw)gЁ┐rЖr~)rrr)rЕrr йrrйrarйrrИ)r)r_┌freqs┌shiftedrrr┌test_axes_keywordQs zTestFFTShift.test_axes_keywordcCsж|jddgddgddgg|jdН}|jddgddgddgg|jdН}ttj|ddН|Гttj|ddН|Гttj|d dН|Гttj|dgdН|Г|jddgddgddgg|jdН}ttj|ddН|Гttj|ddН|Г|jddgddgddgg|jdН}ttj|d dН|Гttj|d dН|Гttj|ddgdН|Гttj|ddgdН|Гttj|ddН|Гttj|ddН|Гttа|б|Гttа|б|ГdS)z1 Test 2D input, which has uneven dimension sizes rrrrr rй┌dtyperРrСrПN)r]┌float64rr rЙrК)r)r_rТ┌ shift_dim0┌ shift_dim1┌shift_dim_bothrrr┌test_uneven_dims\sT¤№¤№¤№¤№zTestFFTShift.test_uneven_dimsN)rPrQrRrЛrОrФrЫrrrrrГ>s rГ┌array_api_strict┌cupyz+fft not yet implemented by array-api-strictz0cupy.fft not yet implemented by array-api-compatc@єeZdZddДZdS)┌TestFFTFreqcCє(t}z|jgdв|j|dН}|jgdв|j|dН}Wnty7|jgdв|jdН}|jgdв|jdН}Ynw|jdtjd|dН|jdН}t||Г|jd|jtjd|j|dН|jdН}t||Г|jdtjd|dН|jdН}t||Г|jd|jtjd|j|dН|jdН}t||ГdS)N) rrrrr rЕrЖr~rwйrЦ┌device) rrrrr rЗrЕrЖr~rwrХrйr_r)rr]rЧ┌ TypeErrorr ┌fftfreqr┌piйr)r_rвr┌x2rBrrrrЛПs, ■ ( (zTestFFTFreq.test_definitionNйrPrQrRrЛrrrrrЯКєrЯc@rЮ)┌TestRFFTFreqcCrа)Nrsrб)rrrrr rrХrrгr)rr]rЧrдr ┌rfftfreqrrжrзrrrrЛлs ¤ ( (zTestRFFTFreq.test_definitionNrйrrrrrлжrкrл)┌__doc__┌scipy.fft._helperrr┌ numpy.testingrrSrr┌numpyr&rV┌scipy.conftestr┌scipy._lib._array_apirr┌scipyr rT┌usefixtures┌ pytestmarkr rrr!r#rZrГrЯrлrrrr┌sHJ= J