bijddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl Z ddl mZmZmZmZmZmZmZmZmZmZmZmZddlmZe j8j:Ze j8j<Z ddZgdZ dZ!d Z"d Z#d Z$Gd d Z%GddZ&GddZ'GddZ(e eGddZ)e edZ*dZ+dZ,GddZ-y)N)xp_assert_equalxp_assert_closeassert_array_almost_equalassert_almost_equalmake_xp_test_case)raises) KroghInterpolatorkrogh_interpolateBarycentricInterpolatorbarycentric_interpolateapproximate_taylor_polynomialCubicHermiteSplinepchipPchipInterpolatorpchip_interpolateAkima1DInterpolator CubicSplinemake_interp_spline)_run_concurrent_barrierc|i}tjjd}gd}tt dt |dz}|j |t |dzzd|jd|zj|} tj|} |tur5|jd|zj|} ||| | fd|i|| } n||| fd|i|| } |xsd| jd|z|z| j|dddz} | j| k(sJ|r|| jdkDrm|tur8|t|t| t fd|i|t| n-|t|t| fd|i|t| | jdkDr|| jd|dt |zz| j|dddz}| tdf|| jzzdz}|j|}tj | |\} } t#| | yyy) N)rrraxisr)nprandom RandomStatelistrangeleninsertrand transposezerosrshapesizeslicendimreshapebroadcast_arraysr)interpolator_clsx_shapey_shape deriv_shaper extra_argsrngxsyxidydxyi target_shapebs_shapeyvs _/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/scipy/interpolate/tests/test_polyint.py check_shaperBs[ ))   %CA U1c'l1n %&AHHTS\!^ $a(4'>#--a0A ' B--sxx$.*44Q7 B aD Bt Bz B2 F < a < < DW$55qr8JJ t/47 8 ZZ !##B+AA +{)r rr!)rrc 2d}tttttt |fD]p}t D]e}t D]Z}tt| t|D]6}|t k7rt|||d|dD]}d|i}t|||d||8\gry)Nct|||SNr)rr8r:rs rA spl_interpztest_shapes..spl_interpFs!!QT22rC)naturalclampedbc_type) r r rrrrSHAPESr&r'rB)rKips1s2rbcextras rA test_shapesrUDs3!"9;M); D G GB G!3r7(CG4GD[(#BBd;"8GB%.OE'BD$FG G G G GrCc ttfD]Qdfd }tD]@}tD]5}tt | t |D]}t |||d|7BSy)Nc,|||jSN) derivatives)r8r:rrPs rAinterpolator_derivsz/test_derivs_shapes..interpolator_derivsWsaD>-- -rCrrD)r r rOr&r'rB)rZrQrRrrPs @rAtest_derivs_shapesr[Usv "9:I . IB I!3r7(CG4ID 3RT4HI I I IrCc &dd}dd}dd}dd}dd}dd}dd}dd}dd }dd } dd } dd } dd } |||||||||| | | | f D]K} tD]@}tD]5}tt| t|D]}t| ||d|7BMy)Nc0t|||jSrX)r derivativerJs rA krogh_derivz&test_deriv_shapes..krogh_derivas At,777rCc0t|||jSrX)r r^rJs rA bary_derivz%test_deriv_shapes..bary_derivds&q!T2===rCc8t|||jSrXrr^rJs rA pchip_derivz&test_deriv_shapes..pchip_derivgsQ4 ++--rCc:t|||jdSNrrcrJs rA pchip_deriv2z'test_deriv_shapes..pchip_deriv2jsQ4 ++A..rCc8t|||jSrXrantiderivativerJs rApchip_antiderivz*test_deriv_shapes..pchip_antiderivmsQ4 //11rCc:t|||jdSrfrirJs rApchip_antideriv2z+test_deriv_shapes..pchip_antideriv2psQ4 //22rCc4Gddt}||||S)NceZdZdZy)9test_deriv_shapes..pchip_deriv_inplace..Pc0tj||dS)Nr)r__call__)selfr8s rArrzBtest_deriv_shapes..pchip_deriv_inplace..P.__call__us(11$1==rCN)__name__ __module__ __qualname__rrr rCrAPrpts  > rCrw)r)r8r:rrws rApchip_deriv_inplacez.test_deriv_shapes..pchip_deriv_inplacess ! At}rCc8t|||jSrX)rr^rJs rA akima_derivz&test_deriv_shapes..akima_derivzs"1a.99;;rCc8t|||jSrX)rrjrJs rAakima_antiderivz*test_deriv_shapes..akima_antideriv}s"1a.==??rCc8t|||jSrX)rr^rJs rA cspline_derivz(test_deriv_shapes..cspline_derivs1a&1133rCc8t|||jSrX)rrjrJs rAcspline_antiderivz,test_deriv_shapes..cspline_antiderivs1a&5577rCc:t|||jSrH)rr^rJs rA bspl_derivz%test_deriv_shapes..bspl_derivs!!QT2==??rCc:t|||jSrH)rrjrJs rAbspl_antiderivz)test_deriv_shapes..bspl_antiderivs!!QT2AACCrCr rD)rOr&r'rB)r_rardrgrkrmrxrzr|r~rrrrPrQrRrs rAtest_deriv_shapesr`s8>./23<@48@DJ \CV 0+/^M6 6B 6!3r7(CG46DBB56 6 66rCcTgd}gd}tttfD]0}|||}t||t j |2gd}t |||}t||t j |t||dt j |y)Nrrrr)rr?r)ryry@r)r r rrr"asarrayr)r8r:rPpr<s rA test_complexrsAA "9;G- q!H!bjjm,- D1a&AAaD"**Q-(AaGRZZ-.rCceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZy) TestKroghctjjgd|_tjddd|_tjddd|_|j|j |_yN)rErrrrdrEr" polynomial Polynomial true_polylinspacetest_xsxsysrss rA setup_methodzTestKrogh.setup_methodsU112CD{{2a, ++b1%..)rCct|j|j}t|j |j ||j yrXr rrrrrrsrws rA test_lagrangezTestKrogh.test_lagrange5 dggdgg .DNN4<<84<<IrCct|j|j}t|j d|ddt|j t j d|t j ddy)NFcheck_0d)r rrrrr"arrayrs rA test_scalarzTestKrogh.test_scalarsY dggdgg .DNN1-qteDDNN288A;7288A;RWXrCc$t|j|j}|j|j}t |j dD]:}t|jj||j||<yNr) r rrrYrr&r,rrderivrsrwDis rAtest_derivativeszTestKrogh.test_derivativessp dggdgg . MM$,, 'qwwqz" &A  7 4 4Q 7 E !! & &rCcRt|j|j}|j|jt |jdz}t |jdD]:}t|jj||j||<y)Nrr) r rrrYrr'r&r,rrrrs rAtest_low_derivativeszTestKrogh.test_low_derivativess~ dggdgg . MM$,,s477|A~ 6qwwqz" &A  7 4 4Q 7 E !! & &rCct|j|j}d}|j|j|}t |D]+}t |j|j|||-yN )r rrrYrr&rr^rsrwmrrs rAtest_derivativezTestKrogh.test_derivativesa dggdgg .  MM$,,q )q CA  T\\! .worker_fnH 477OrCr)r rrrrsrwr s` rAtest_concurrencyzTestKrogh.test_concurrencyEs* dggtww /  Iq1rCN)rtrurvrrrrrrrrrrrrrrrrrrrrrrr rCrArrsu* JY &&C= / /J /*9- , ;L ?3:2rCrceZdZdZy) TestTaylorcd}ttjd|dd}t|dzD]$}t |dd|j }&t |ddy)NrErr)r r"expr&rr)rsdegreerrs rAtest_exponentialzTestTaylor.test_exponentialOs] )"&&!VQ Cvax A !Q ' A  AaD#rCN)rtrurvrr rCrArrNs$rCrceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZy)TestBarycentricctjjgd|_tjddd|_tjddd|_|j|j |_yrrrs rArzTestBarycentric.setup_methodYsU112CD{{2q#. ++b!Q'..)rCct|j|jd}t||j|j |jy)Nr) random_state)r rrrrrrs rArzTestBarycentric.test_lagrange_s7 #DGGTWW1 E$,, )EFrCc t|j|jd}t|d|j ddt|t j d|j t j ddy)Nrr7rFr)r rrrrr"rrs rArzTestBarycentric.test_scalardsZ #DGGTWW! <!dnnQ/%@"((1+rxx{(CeTrCc$t|j|j}|j|j}t |j dD]:}t|jj||j||<yr) r rrrYrr&r,rrrrs rArz TestBarycentric.test_derivativesiso #DGGTWW 5 MM$,, 'qwwqz" IA 3DNN003DLLA1Q4 H IrCcVt|j|j}|j|jt |jdz}t |jdD]<}t|jj||j||d>y)Nrr-q=r) r rrrYrr'r&r,rrrrs rArz$TestBarycentric.test_low_derivativesos #DGGTWW 5 MM$,,DGG Q 7qwwqz" (A 3DNN003DLLAaD!& ( (rCct|j|j}d}|j|j|}t |D]+}t |j|j|||-yr)r rrrYrr&rr^rs rArzTestBarycentric.test_derivativewsa #DGGTWW 5  MM$,, *q AA ALLq91Q4 @ ArCc Pt|j|j}tt |jdt |jzD]N}t |j |j|tjt |jPy)NrE) r rrr&r'rr^rr"r+rs rArz$TestBarycentric.test_high_derivative~sn #DGGTWW 5s477|Qs477|^4 9A ALLq9HHS%67 9 9rCc |j}tjjgd}tjjgd}tj||j ||j ||j fd}t |j |d}|j|j}t|jdD]}t||tj|j||j|j||j|j||jfddyNrrrrIrr r) rr"rrrrr rYrr&r,rrrs rArz%TestBarycentric.test_ndim_derivativess ((8 (();< XXuTWW~uTWW~uTWW~FR P #DGGRa 8 MM$,, 'qwwqz" (A AaDHHnekk!nT\\&B&4ekk!nT\\&B&4ekk!nT\\&B&D+-."'  ( (rCc |j}tjjgd}tjjgd}tj||j ||j ||j fd}t |j |d}t|jD]}t|j|j|tj|j||j|j||j|j||jfddyr$) rr"rrrrr r&rrr^rrrs rArz$TestBarycentric.test_ndim_derivatives ((8 (();< XXuTWW~uTWW~uTWW~FR P #DGGRa 8qss (A ALLq9HHnekk!nT\\&B&4ekk!nT\\&B&4ekk!nT\\&B&D+-."'  ( (rCct|j}|j|jt |j |j ||j yrX)r rset_yirrrrrs rA test_delayedzTestBarycentric.test_delayeds? #DGG , DNN4<<8!DLL/JrCct|jdd|jdd}|j|jdd|jddt |j |j ||j y)Nr)r rradd_xirrrrs rA test_appendzTestBarycentric.test_appendsc #DGGBQK! = dggabk*DNN4<<8!DLL/JrCc gd}tjddgddgddgg}t}|||}t|jdDcgc]}|||dd|f}}tj ddd}t ||tj|Dcgc] }|| c}jycc}wcc}w)Nrrrrrrr) r"rr r&r,rrrr) rsrrBIrwrrrrs rArzTestBarycentric.test_vectors  XX1v1v1v. / $ r2J(-bhhqk(: ;1bR1X ; ;++b!S)AgJJJB'?q' '?@BB D<(@s CCcpt|j|j}tj|ddk(sJtj|tj ddk(sJtj|dgdk(sJtj|ddgdk(sJyr)r rrr"r,rrs rArz'TestBarycentric.test_shapes_scalarvalues #DGGTWW 5xx!~###xx"((1+'2---xx1#4'''xx1a& "d***rCct|j|j}|j}t j |j d|fk(sJt j |j t jd|fk(sJt j |j dg|dfk(sJt j |j ddg|dfk(sJyr)r rrrr"r,rYrrs rArz2TestBarycentric.test_shapes_scalarvalue_derivatives #DGGDGG 4 CCxx a()aT111xx bhhqk23t;;;xx qc*+!u444xx qe,-!A666rCc Xt|jtj|jtj d}tj |ddk(sJtj |dgdk(sJtj |ddgdk(sJyr)r rr"rrrr,rs rArz'TestBarycentric.test_shapes_vectorvalues #DGGRXXdggryy|-L Mxx!~%%%xx1#6)))xx1a& "f,,,rCc4t|jtj|jdg}tj |ddk(sJtj |dgdk(sJtj |ddgdk(sJyr)r rr"rrr,rs rArz*TestBarycentric.test_shapes_1d_vectorvaluesy #DGGRXXdggs-C Dxx!~%%%xx1#6)))xx1a& "f,,,rCc t|jtj|jtj d}|j }tj|jd|dfk(sJtj|jdg|ddfk(sJtj|jddg|ddfk(sJyr) r rr"rrrrr,rYrs rArz2TestBarycentric.test_shapes_vectorvalue_derivatives #DGGBHHTWWRYYq\,J K CCxx a()aV333xx qc*+1ay888xx q!f-.1a);;;rCc t|j|jd}t}t ||j ||j|j|j dt |j |j d||j|j|j ddt |j|j d||j|j|j ddgdy)Nrrr)rr7r)r rrr rrr^rY)rsrwbis rArzTestBarycentric.test_wrappers #DGGTWW! < $$,,DGGTWWdllPQ)RS T\\15477DGGT\\qaH J dllA6477DGGT\\1v1M OrCcdtjddz}tjdd}t||d}t|tjdy)Nir g@g#@)r"rr rr)rsr8r:values rAtest_int_inputzTestBarycentric.test_int_inputsG 299Q# # IIa '1j9E2::c?3rCcnd}tj|dzjtj}tj|tj z|z }d|z}|dxxdzcc<|dxxdzcc<t |}|jd}t|jd|zz |y)NiLrrr?r) r"rastypefloat64cospir wir)rsrjr8wrwfactors rAtest_large_chebyshevz$TestBarycentric.test_large_chebyshevs  IIa!e  # #BJJ / FF1ruu9q= !AI !  "  #A &aADDAJ/3rCctddgddg}tjd5||j}dddt |j j y#1swY.xYw)Nrrrraise)divide)r r"errstater;rr=ravel)rsrwr=s rA test_warningzTestBarycentric.test_warnings\ $QFQF 3 [[ ( 144B  B -   s A''A0ctjgd}tjgd}tjtd5t ||dddy#1swYyxYw)N)皙?r:?r:rz)Interpolation points xi must be distinct.r)r"rrr ValueErrorr )rsxisrs rAtest_repeated_nodez"TestBarycentric.test_repeated_node sRhh+, XXl # ]]:!LN - #C , - - -s A  A)cjtjj}fd}td||y)Nc*|jyrXrr s rAr z3TestBarycentric.test_concurrency..worker_fnr rCr)r rrrrs` rArz TestBarycentric.test_concurrencys* #DGGTWW 5  Iq1rCN)rtrurvrrrrrrrrrr(r+rrrrrrrr8rCrIrOrr rCrArrXs}* G U I (A9 ( (K K D+7- - <O4 42 .-2rCrcJeZdZd dZdZdZdZdZdZdZ dZ d Z d Z y ) TestPCHIPctjjd}tj|j|}|j|}t ||||fS)Nr)r"r#r$sortr)rsnptsr7r;r=s rA _make_randomzTestPCHIP._make_randomsNii##D) WWSZZ% & ZZ R}b"$$rCc*|j\}}}tt|dz D]e}||||dz}}||||dz}}||kDr||}}tj||d} || } || dzk| |dzkzj reJy)NrrrrWr&r'r"rall rsrr;r=rx1x2y1y2xpyps rAtest_overshootzTestPCHIP.test_overshoot%s%%' 2rs2wqy! CAUBqsGBUBqsGBBwRBRR(B2B2:%"U *:;@@B BB CrCc|j\}}}tt|dz D]_}||||dz}}||||dz}}tj||d} || } ||z | dd| ddz zdkDj r_Jy)NrrrrYr[s rA test_monotonezTestPCHIP.test_monotone1s%%' 2rs2wqy! ;AUBqsGBUBqsGBRR(B2BUr!"v2A/!388: ::  ;rCctjgdgdg}tjd}t|d|d|}|dz}t|d|d|}t ||ddy) N) rrr/<OWcr) rlirr&-57rrr?+=)rrtol)r"rrrr)rsdataxxcurvedata1curve1s rA test_castzTestPCHIP.test_cast;swxx@DFG YYs^'d1gtAw'+s *uQxq*2.vE>rCc&d}tjtj|}t |dddf|dddf}d}tjtj|}t |dddf||dddfddy)Na 7.99 0.00000E+0 8.09 0.27643E-4 8.19 0.43750E-1 8.70 0.16918E+0 9.20 0.46943E+0 10.00 0.94374E+0 12.00 0.99864E+0 15.00 0.99992E+0 20.00 0.99999E+0 rra^ 7.9900 0.0000 9.1910 0.4640 10.3920 0.9645 11.5930 0.9965 12.7940 0.9992 13.9950 0.9998 15.1960 0.9999 16.3970 1.0000 17.5980 1.0000 18.7990 1.0000 20.0000 1.0000 g-C6 ?rtr)r"loadtxtioStringIOrr)rsdataStrrupch resultStrresults rAtest_nagzTestPCHIP.test_nagGs  zz"++g./D1ItAaCy)  BKK 23qs S!%5BTJrCctjgd}tjgd}tjgd}t||t||fD]}|d|dfD]}||ddk7rJ!y)N)r|rKg?gffffff?)皙uq@g@@@@@)rrgp@rrrr)r"rr)rsr8r^r_ppts rAtest_endslopeszTestPCHIP.test_endslopesjs~ HH+ , XX3 4 XX4 5B<q". %BdAbE] %!Qx1}$} % %rCcFtjd}tj|}tj5tj dt ||}dddtjddd}t|dk(sJy#1swY7xYw)Nrerrorr er|) r"rrwarningscatch_warningsfilterwarningsrrrZ)rsr8r:rrvs rAtest_all_zeroszTestPCHIP.test_all_zerosts IIbM MM!  $ $ &   # #G ,1+C [[As #3r7b=!!!   s "BB ctjddd}tddgddg}t||d|zdy)Nrrr6rrr)r"rrr)rsr8rs rAtest_two_pointszTestPCHIP.test_two_pointss> KK1b ! 1a&1a& !!ac.rCc `ttgdgddgdtjdgttgdgddgdtjdgttjtgdgddgddgtjdgdggy) Nrrr)rrErr:rrrrrg @)rrr"rrs rAtest_pchip_interpolatez TestPCHIP.test_pchip_interpolates! iSEq A JJt   " iSEq A JJu   " JJ(Iu1a&Q R JJs| $ &rCctddgddg}|j}t|tjdgy)Nrrrr:)rrootsrr"r)rsrrs rA test_rootszTestPCHIP.test_rootss6 1a&2q' " GGI2::se,-rCN)) rtrurvrWrbrdrzrrrrrrr rCrArSrSs7% C; ?!KF% "/ &.rCrSc^eZdZe d dZdZdZdZdZdZ dZ dZ d Z d Z d Zy )TestCubicSplinec &|j}|j}tj|}|j |j dgdg|j dz zz}|dd}t|dddf|dddf|dzz|dddf|dzzz|dddf|zz|dddfz||t|dddfd|dddfz|dzzd|dddfz|zz|dddfz||t|dddfd|dddfz|z|dddfz|||jdk(r3|dk(r.|dk(r)t|dtj|d||y|d k(rjt||dd||dd||t||dd||dd||t||dd||dd||y|dk(rq|jdk(rL||d||dz |dz }tj|}t||dd|||nt|d |d ||n|d k(r7t||ddtj||dd||nm|d k(r7t||ddtj||dd||n1|\} } t||d| tj| |||dk(rq|jdk(rL||d||dz |dz }tj|}t||dd|||yt|d|d||y|d k(r7t||ddtj||dd||y|d k(r=t||ddtj||ddd|zd|zy|\} } t||d| tj| ||y)zVCheck that spline coefficients satisfy the continuity and boundary conditions.rrrNrrr} not-a-knotperiodic)rr)rrrMrL)rr)rr) r8cr"diffr0r,r/rr-rr) Sbc_startbc_endtolr8rdxdxislopeorderr7s rAcheck_correctnessz!TestCubicSpline.check_correctnesssJ CC CC WWQZ ZZ! qvvz(:: ;"g !QR%!AssF)c1f"4qCRCy367I"I!SbS& C#(*+AssF)#4:=C I!QR%!a3B3i-#q&"8Aa"fI +#,./3B3i#8>A M!QR%!a3B3i-#"5!SbS& "A s , 66Q;8|3,8N AaD"--!"5Cc J  z ! AadAJ!B% #C H AadAJ!B% #C H AadAJ!B% #C H  | #vv{1Q41QqT7*be3 5)!A$ E#F$4sE  " !A$ BMM!AaD!*5Cc K  " !A$ BMM!AaD!*5Cc K$LE5 AadENBJJu,=Cc R \ !vv{1Q41QqT7*be3 5)!A$ E#F%!E(3G y AaeQKq2{)C!$3 0 y AaeQKq2{)C!"3QsU 4"LE5 AaeUORZZ->Ss SrCc t|j}||=tj|}|j dtj|}|j ddddd|fd|fg}|ddD]$}t ||||} |j | ||&|D]/} |D](} t |||| | f} |j | | | d *1y) NrrrrLrMrrrrNg+=r)r%r,r"emptyfillrr) rsr8r:rr5 first_deriv second_derivbc_allrSrrrs rA check_all_bczTestCubicSpline.check_all_bcs177m  hh{+ xx , "         !* .BAqtR8A  " "1b" - . GH  G14(F9KL&&q(F&F G GrCctjgd}tjgd}dd|jfD]}|j|d||d|dtjd|df}|d||ddddf<|d|dz |ddddf<|d|dz|ddddf<|d|dz|ddddf<|j|d||dy)N)rrr:rrg@g@r)rgrrg@rrr:rrrr)r"rr-rr)rsr8r:rYs rA test_generalzTestCubicSpline.test_generals HH4 5 HH4 5Q +A   aeQrUA .!Q#A2AAaAgJ2AAaAgJ2AAaAgJ2AAaAgJ   aeQ * +rCcdD]}tjddtjz|}tj|}t ||d}|j |ddtj d|df}||ddddf<|dz|ddddf<|dz |ddddf<|dz|ddddf<t ||dd}|j |ddy) N)rrrErrrrNrrEr)r"rr>r=rrr)rsrr8r:rrs rA test_periodiczTestCubicSpline.test_periodics >A Aq255y!,Aq AAq*5A  " "1j* =!Q#AAaAgJQAaAgJQAaAgJQAaAgJAqq*=A  " "1j* = >rCcL|jdd|jzd|j}|j|}t ||d}t |d|dd|jzzd t ||}t |||j|d y) Nrrrdtyperrrr)decimal)rr>r<r=rr)rsr`r8r:rs rAtest_periodic_evalz"TestCubicSpline.test_periodic_evals KK1ruu9b K ; FF1I 1j 1AaD!ABEE M"2B? 1 AaD"&&)R8rCctjgd}tjgd}t||dd}|j|ddy)N)rL?gffffff?@@@g333333@g@g@g@g@g@g @gffffff"@g%@g&@g333333'@g(@g333333)@g*@g*@)rg?g?rrg@g333333@g333333@gffffff@r@gffffff@rg333333?gffffff?rLgffffff?g333333?r:g?rr)rN extrapolate)r"rrrrsr8r:rs rA*test_second_derivative_continuity_gh_11758z:TestCubicSpline.test_second_derivative_continuity_gh_11758sN HHO P HHP Q 1jj I q*j9rCctjgd}tjgd}t||d}|j|ddt |j d|tjgdy)N)rrg@r)rrg.@rrrrr)Hrr)r"rrrrr^rs rAtest_three_pointsz!TestCubicSpline.test_three_points&se HH% & HH% & 1j 1 q*j9 Q*BHH5J,KLrCctjgd}tjddgddgddgg}t||d}|j|ddt||dddfd}t||dddfd}tjddd }t ||dddf||t ||dddf||y) N)r|rrrr|rrrrrrrrE)r"rrrrr)rsr8r:rS0S1qs rA#test_periodic_three_points_multidimz3TestCubicSpline.test_periodic_three_points_multidim2s HH_ % HHsCj3*sCj9 : 1j 1 q*j9 AadGZ 8 AadGZ 8 KK1a !QT BqE*!QT BqE*rCctjgdt}tjgdt}t||}|j |tjgd}t||}|j |t||dzd}|j |dd tjgd}t||d d g}|j |d d y) N)rrrrr)rrrr)y?r|y?y?r)rLry@rrLr)ry@?)ry?)r"rintrrrs rA test_dtypeszTestCubicSpline.test_dtypes?s HH\ - HH]# . 1  q! HH1 2 1  q! 16+? @ q)W5 HH] # 1}m&D E q-?rCctjjd}tj|j d}d|j dz}t ||}|j |dy)Nrr)r-g@gvIh%<=r)r"r#r$rUuniformrr)rsr7r8r:rs rA test_small_dxzTestCubicSpline.test_small_dxPscii##A& GGCKKSK) * #++3+' ' 1  qe,rCc *tjgd}tjgd}tjgd}tjtjdddg}tjgd}tjtjdddg}gd}dg}dg} ttt ||ttt ||ttt ||ttt ||ttt ||ttt |ddtj f|ttt || gd } | D]} ttt ||d | d tj||f} d } d dgdff}d dd d ggff}ttt || d | d ttt || d |d ttt || d |d ttt ||d dd y)Nr)y??rrrrrr)rrrrrr))rrM))rr)rr)rr)r|r|z not-a-typorT)rMrrM)rrrr)r"rnan assert_raisesrMrnewaxisc_)rsr8r:xcxnxoyny3r\r^wrong_bcrNrbc1bc2bc3s rAtest_incorrect_inputsz%TestCubicSpline.test_incorrect_inputsWs HH\ " HH\ " XX' ( XXrvvq!Q' ( XXl # XXrvvq!Q' ( SSj+r15j+r15j+q"5j+r15j+q"5j+qBJJ/?Cj+r26"   KG *k1aGT J K EE!Q$K!1i.)11vh-(j+q!QTBj+q!QTBj+q!QTB j+q!Q DIrCN)rrrs)rtrurv staticmethodrrrrrrrrrrrr rCrArrsS;G#?T?TBG. + >9: M +@"-&JrCrc|jgd}|jgd}|jgd}t|||}t|||dddt||d|dddy)N)rrr)rrr)rrrFr)rB check_dtypertr)rrr)r`r8r:r<r9s rA#test_CubicHermiteSpline_correctnessrsd 9A :A ::i D1a&AAaD!ENAaGTu%eTrCcgd}gd}gd}ttt|||ddtjg}ttt|||y)Nr)rrrE)rrrrrr)rrMrr"r)r8r:r< dydx_with_nans rA&test_CubicHermiteSpline_error_handlingrsBAA D*0!Q=266NM*0!Q FrCc*tjddg}tjddg}tjddg}t|||}|jd}|jj d d k(sJ|j d k(sJy) NgMbP?gMb`?gBܻ>gdJ>g/"g%"T)rrr)r"rrrrr,r-)r8r:dyrrs rAtest_roots_extrapolate_gh_11185rs % A ..12A ; * +B1a$A D!A 3399Q<1   66Q;;rCcfeZdZejj dejdejdgejj dgdejj dgdejj dee gd Z ejj dejdejdgejj dgdejj de e gd Z y ) TestZeroSizeArraysr:)rrrE)rrErrN)rrrLrMrrclsctjd}tjd}||||}||jdk(sJ||j|j|jddzk(sJtj|d|}|||||}|jd||jfz|j|dzdz} ||jdk(sJ||j| k(sJy)Nrrrrr)rNrr"rr-r,moveaxis) rsrr:rNrr8xvalobjytshs rAtest_zero_sizez!TestZeroSizeArrays.test_zero_sizes IIbMyy|!Q(4y~~"""4y$**qwwqr{"::::[[At $!Rt4 XXet_ } ,rxxQ/@ @4y~~"""4y"$$$rCctjd}tjd}|||}||jdk(sJ||j|j|jddzk(sJtj|d|}||||}|jd||jfz|j|dzdz}||jdk(sJ||j|k(sJy)NrrrrrIr) rsrr:rr8rrrrs rAtest_zero_size_2z#TestZeroSizeArrays.test_zero_size_2s IIbMyy|!Qi4y~~"""4y$**qwwqr{"::::[[At $!Rd# XXet_ } ,rxxQ/@ @4y~~"""4y"$$$rCN)rtrurvrmark parametrizer"r+rrrrrrr rCrArrs [[S8288J#7#+288J#7#9: [[YMO [[VY/ [[U%7$EF%G0O: %  [[S8288J#7#+288J#7#9: [[VY/ [[U%68K$LM%N0:%rCr)NrN).rrnumpyr"scipy._lib._array_apirrrrrrrrscipy.interpolater r r r r rrrrrrrscipy._lib._testutilsrrskip_xp_backendsxfail_xp_backendsrBrOrUr[rrrrrrSrrrrrr rCrAr s +  :;;//KK11LM&R %G"I26j /h2h2V$$C2C2L|.|.~;_J_J _JD%&U'UG /%/%rC