0^jI$ ddlZddlmZmZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZddlmZddlmZddlmZmZmZmZddlmcmZej8j:Zej<dZeee e e ej8jAdee fe e fgej8jAdgd ej8jAd gd ej8jAd dd gej8jAdgdej8jAdddgdZ!ej8jAdee ge e ggej8jAdgd ej8jAdejDejFejHejJejLgej8jAd dd gej8jAdgdej8jAdddgdZ'ee e eeej8jAde e feefgej8jAdgd ej8jAdgdej8jAdgdej8jAdddgdZ(ej8jAde e geeggej8jAdgd ej8jAdgdej8jAdejDejFejHejJejLgej8jAdgdej8jAdddgdZ)ej8jAdeeee ee eegej8jAdgd ej8jAdgddZ*ej8jAdeeee ee eegej8jAdgd dZ+ej8jAdgdej8jAd eed!ee d ee d!gd"Z,eeej8jAdgdd#Z-ej8jAdgdej8jAdeeee gd$Z.ej8jAdgdej8jAdeeee gd%Z/ej8jAdee e e e e eegd&Z0y)'N)assert_allcloseassert_array_equal)dctidctdctnidctndstidstdstnidstn)fftpack)xp_copyxp_assert_closemake_xp_test_casemake_xp_pytest_paramzforward, backwardtype)rn)rrr axisrnorm)Nbackwardorthoforward orthogonalizeFTcl|jtjj||}||||||} || ||||} t | |dgdz} d| |<|jtj tj| | d} || |||||} t | |y)N)rrr rrrrredgemoder )asarraynprandomrandrpad)rrrrrrr xpxyzr,y2z2s o/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/scipy/fft/tests/test_real_transforms.pytest_identity_1dr4s 299>>!Q'(A4d]KADt$mLAAq (Q,CCI BFF2::a=#F; rtolatol)r)r*r+astypecopyrr) rrrr6rrr7r.x_origr/y_origr0s r3test_identity_1d_overwriterC-s q!##E*A VVXF4d;GA VVXFDt$KHA 14d31f%1f%648r5z shape, axes) )rrr)rDr)rDN)rDrr)r N)rFrE)rrN)rHr)rH)rrc.|jtjj|}|tj||}||||||} || ||||} t | ||dg|j z} nBt |trdg|j z} d| |<ndg|j z} |D]} d| | < |jtjtj| | d} || |||||}t ||y)N)axesrr r#r"r$r%r') r(r)r*takerndim isinstanceintr,)rrrshaperKrr r-r.r/r0r,ar1r2s r3test_identity_ndrREs* 299##E*+A t$4d]KADt$mLAAq |h D# hD h ACF  BFF2::a=#F; >#&'A e D > D )1r5)rrrz func, typerc|jtjjd}||||d}||||d}t ||y)Nr_Tr`Fra)rVrrr-r.y1r1s r3test_orthogonalize_nooprfsI 299>>#&'A adT :B adU ;BBr5c|jtjjd}t ||}t j |djtt j |djtt|d|d}t|d|d}t j |djtt j |djtt||y) Nr_r-rrTr`F) r(r)r*r+rxpxatmultiplySQRT_2rdivider)rr-r.x2rer1s r3test_orthogonalize_dct1rps 299>>#&'A r BFF2qM6"FF2rNF# QQT 6B Rad% 8BFF2qM FF2rN&!Br5c|jtjjd}||d|d}||d|d}t j ||j dk(rdndjtt||y) Nr_rTr`Frrri) r(r)r*r+rjrkrYrnrmr)rVrr-r.rer1s r3test_orthogonalize_dcst2rrso 299>>#&'A aad$ 7B aad% 8BFF2DMMU*q3::6BBr5c6|jtjjd}t ||}t j ||jdk(rdndjt||d|d}||d|d }t||y) Nr_rhrrrirTr`F) r(r)r*r+rrjrkrYrlrmr)rVrr-r.rorer1s r3test_orthogonalize_dcst3rts{ 299>>#&'A r BFF2DMMU*q3<rs= DDD99--;;// 13c4(,T{S$K.HI.23!Q(!GH5$-8 9I)4/J)  ,T{S$K.HI.2::rzz2::#%<<#@A!Q(!GHu 697I)A/J9"4e,,e}tUm.LM.  !GH5$-89I /N-"<,e}tUm.LM. 2::rzz2::#%<<#@A!GH 6&7IA /N&*"6s";"6s";"6t"<"6t"<">?.!GH*I/ ? *"6s";"6s";"6t"<"6t"<">?. / ? !AB(>C!AB"6s";"6s";"=>>C#tT5#tT5!QR2S2r5