Ug<ddlZddlZddlZddlmZmZmZmZm Z ddlm Z ddl m Z mZmZmZmZddlmZddlmZejde jZGdd ZGd d ZGd d ZdS)N)assert_allclose assert_equal assert_warnsassert_array_almost_equalassert_array_equal)raises)RegularGridInterpolatorinterpnRectBivariateSplineNearestNDInterpolatorLinearNDInterpolator)matrix)ComplexWarningmethodc eZdZdZdZdZdZedZe j dgddZ e j d ee jgd gd gd gfee jgd fgdZdZedZdZdZdZe j de jgddfe jgddfe jgddfe jgddfe jgddfgdZdZd Zd!Zd"Zd#Zd$Zd%Zd&Ze j dd'd(gd)Zd*Zd+Z d,Z!e j d-d.e j"e j#ge j dd(d'gd/Z$e j dd'd(gd0Z%e j dd'd(gd1Z&e j 'd2ee j d3d4d5fd6d7fd8d9fd2d:fgd;Z(d<Z)ed=Z*e j 'd4ed>Z+ee j d?d@dAgdBZ,dCZ-e j dDe j.e j/e j0e j1ge j dEe j.e j/gdFZ2dGZ3d.S)HTestRegularGridInterpolatorcdgdz}tjgd}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fS)N?? dnpasarraynewaxisselfpointsvaluesvalues0values1values2values3s _/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/scipy/interpolate/tests/test_rgi.py_get_sample_4dz*TestRegularGridInterpolator._get_sample_4ds!#LLL))BJ BJ>?QQQ BJ>?RZBJ>?RZQQQ>?GbL(7S=87T>Iv~cdgdzdgdzz}tjgd}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fSNr)r@$@rrrrr s r(_get_sample_4d_2z,TestRegularGridInterpolator._get_sample_4d_2"s!#}o&99LLL))BJ BJ>?QQQ BJ>?RZBJ>?RZQQQ>?GbL(7S=87T>Iv~r*cdgdz}tjgd}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fS)Nrrr?@@@rrrrrr s r(_get_sample_4d_3z,TestRegularGridInterpolator._get_sample_4d_3-s56:???@@BJ BJ>?QQQ BJ>?RZBJ>?RZQQQ>?GbL(7S=87T>Iv~r*cdgdz}tjddg}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fS)N)rrrrrrrrrr s r(_get_sample_4d_4z,TestRegularGridInterpolator._get_sample_4d_48s!S#J''BJ BJ>?QQQ BJ>?RZBJ>?RZQQQ>?GbL(7S=87T>Iv~r*cJ|\}}tjgdgdgdg}t|||}||}t|||}||}t ||dS)N皙?r<r?皙?r<g?g?rrrrr)r7rrr tolistr)r!rr"r#sampleinterpv1v2s r(test_list_inputz+TestRegularGridInterpolator.test_list_inputCs..00///1D1D1D///122))/06888VFMMOO $ $()/06888VF^^Br*r)cubicquinticpchipc|\}}d}tjt|5t |||dddn #1swxYwYt ||}t jgdgdgdg}tjt|5|||ddddS#1swxYwYdS)Nzpoints in dimensionmatchrAr;r>r@)r9pytestr ValueErrorr rr)r!rr"r#rMrDrCs r(test_spline_dim_errorz1TestRegularGridInterpolator.test_spline_dim_errorTs~..00%]:U 3 3 3 C C #FF6 B B B B C C C C C C C C C C C C C C C)88///1D1D1D///122 ]:U 3 3 3 * * F6& ) ) ) ) * * * * * * * * * * * * * * * * * *s#AAA'CC Czpoints_values, sampler;r>r@r<r<r/"@c||\}}t||d}||}t||d}||}t||dS)NlinearrAslinear)r r)r! points_valuesrCr"r#rDrErFs r(test_linear_and_slinear_closez9TestRegularGridInterpolator.test_linear_and_slinear_closedso't,,(III VF^^( JJJ VF^^Br*c|\}}tjgdgdgdg}t||d}t t 5||ddddn #1swxYwYt ||dgd d t ||d gd d t ||dgdd dS)Nr;r>r@rUrA)nu)rYrrr)rYrYrYV瞯<atol)rrYrr)rrr)rrYrYr)rrrg-=)r)rarrayr assert_raisesrOr)r!r"r#rCrDs r(test_derivativesz,TestRegularGridInterpolator.test_derivativeszsw,,..333333333566) JJJ : & & ! ! F6a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! v,777!  / / / /v,777$ 5 2 2 2 2 v,777!  / / / / / /sA55A9<A9c|dkrtjd|\}}|d|zz }tjgdgdgdg}t |||}t ||j|}t ||j|}||}||d||zz} t|| dS) NrJ*pchip does not make sense for complex data@r;r>r@rA?) rNskipr7rrr realimagr) r!rr"r#rCrDrinterpiinterprErFs r( test_complexz(TestRegularGridInterpolator.test_complexs W   KD E E E..00"V)#///1D1D1D///122)GGG)&&+fMMM)&&+fMMM VF^^ WV__r''&//1 1Br*cgdgd}}tj||d\}}d||}t||f|d}t||f|d}|dd g}|dd g} tj|| d d rJdS) N)rYr-rij)indexingc|dz|dzzS)Nrxys r(zATestRegularGridInterpolator.test_cubic_vs_pchip..sq!tad{r*rHrArJr3r-+=rr]rtol)rmeshgridr allclose) r!rrrsxgygr#rHrJ vals_cubic vals_pchips r(test_cubic_vs_pchipz/TestRegularGridInterpolator.test_cubic_vs_pchips||\\\1QD111B***B33'AwGGG'AwGGGUC8__ UC8__ ;z:EJJJJJJJJr*c|\}}t||}tjgd}d}t |||dS)NrQH@r0r rrrr!r"r#rDrCwanteds r(test_linear_xi1dz,TestRegularGridInterpolator.test_linear_xi1ds_..00(88///00!&&..&99999r*c|\}}t||}tjgdgdgdg}tjgd}t |||dS)Nr;r>r@rgq@g\@r)r rrrrs r(test_linear_xi3dz,TestRegularGridInterpolator.test_linear_xi3ds,,..(88///1D1D1D///12222233!&&..&99999r*zsample, wanted)r<r<r=r=g0@)r<r<r<r<rrrrrrrrr\@)r<皙?g333333?r=g|@c|\}}t||d}t|||dS)NnearestrA)r)r r)r!rCrr"r#rDs r( test_nearestz(TestRegularGridInterpolator.test_nearestsK,,..( JJJ!&&..&99999r*c|\}}t||}tjgdgdg}tjddg}t |||dS)Nrrrrrrs r(test_linear_edgesz-TestRegularGridInterpolator.test_linear_edgessy,,..(88---/?/?/?@AARK((!&&..&99999r*cddg}tjgd}|ddtjf}|tjddf}||dzz}ttt ||ddg}ttt ||ddg}ttt ||gd}ttt ||ddg}ttt ||ddS) Nr)rrrr)r)rrg?r)rrr undefmethodrA)rrrr_rOr )r!r"r#r$r%s r(test_valid_createz-TestRegularGridInterpolator.test_valid_creates -LLL))BJ'QQQ'GbL(j"966JJJ"L1j"966JJJ#\2j"966JJJ;;;j"966JJJ -j"966* , , , , , ,r*cv|\}}t||}tjgdgdg}t t ||dtjgdgdg}t t ||tjgdgdg}t t ||dS)Nrrr)rrr)rrr)rrr皙?)r)r rrr_rO)r!r"r#rDrCs r(test_valid_callz+TestRegularGridInterpolator.test_valid_calls,,..(88---/?/?/?@AAj&&-@@@\\\<<<899j&&111---/@/@/@ABBj&&11111r*c^|\}}t||dd}tjgdgdgdgdg}tjgd}t ||d |tjgd }t ||d |dS) NF bounds_error fill_value皙rrrrrrr@皙irrrr)rr&@rrrA)gfffff[gfffff@ggrTrrs r(test_out_of_bounds_extrapz5TestRegularGridInterpolator.test_out_of_bounds_extraps,,..(e48:::1113G3G3G1113I3I3IKLL11122!&& "B"B"BFKKK>>>??!&&"A"A"A6JJJJJr*c^|\}}t||dd}tjgdgdgdgdg}tjgd}t ||d |tjgd }t ||d |dS) NFrrrrr)rrrrrrA)g333333(g33333`@ggyXrTrrs r(test_out_of_bounds_extrap2z6TestRegularGridInterpolator.test_out_of_bounds_extrap2s..00(e48:::1113G3G3G1113I3I3IKLL///00!&& "B"B"BFKKK999::!&&"A"A"A6JJJJJr*c|\}}t||dtj}tjgdgdgdg}tjtjtjtjg}t ||d|t ||d|tjgd gd gd g}tjgd }t |||dS) NFrrrrrrArTr;r>r@r)r)r rnanrrrs r(test_out_of_bounds_fillz3TestRegularGridInterpolator.test_out_of_bounds_fills&,,..(e46F<<<1113G3G3G333566RVRVRV455!&& "B"B"BFKKK!&&"A"A"A6JJJ///1D1D1D///12222233!&&..&99999r*c||\}}t||d}tj|}d|D}t j|}|d}t||}t jgdgdgdg}t||||dS)NrrAcg|]}|Srprp.0ps r( zJTestRegularGridInterpolator.test_nearest_compare_qhull..000a000r*r;r>r@) r)r itertoolsproductrrreshaper rr!r"r#rD points_qhull values_qhull interp_qhullrCs r(test_nearest_compare_qhullz6TestRegularGridInterpolator.test_nearest_compare_qhulls,,..( JJJ (&1 00<000 z,// ~~b)) ,\<HH ///1D1D1D///122!&&..,,v2F2FGGGGGr*cx|\}}t||}tj|}d|D}t j|}|d}t||}t jgdgdgdg}t||||dS)Ncg|]}|Srprprs r(rzITestRegularGridInterpolator.test_linear_compare_qhull.. rr*rr;r>r@) r)r rrrrrr rrs r(test_linear_compare_qhullz5TestRegularGridInterpolator.test_linear_compare_qhulls,,..(88 (&1 00<000 z,// ~~b)) +L,GG ///1D1D1D///122!&&..,,v2F2FGGGGGr*rrTc*tjddd}tjddd}td}t||f||}|dd g}t||f|j|}|dd g}t ||dS) Nrr-rYrrrArffffff?)rlinspaceMyValuer _vr)r!rrrrsr#rDrErFs r(test_duck_typed_valuesz2TestRegularGridInterpolator.test_duck_typed_values(s K1a  K1a (!QGGG VS#J  (!Q6JJJ VS#J  Br*cDtjdtjddd}tjddd}tjdd}t ||f|dt tt ||f|ddS) Nrr-rrYrry?@)rrandomseedrrandr r_rO)r!rrrsr#s r(test_invalid_fill_valuez3TestRegularGridInterpolator.test_invalid_fill_value6s t K1a  K1a 1%% A1==== j"9!ff 7 7 7 7 7 7r*ctjdd}d|jD}t||t||ddS)Nrz>f4dtypec6g|]}tj|Srp)rarange)rns r(rzCTestRegularGridInterpolator.test_fillvalue_type..Fs 5551")A,,555r*rr)ronesshaper )r!r#r"s r(test_fillvalue_typez/TestRegularGridInterpolator.test_fillvalue_typeCsVU33355 555///2>>>>>>r*c d}tjddd}tjddd}|tj||dd}t||f|ddd }t |tjddgdd gddgggd d t |tjddgddgddgggdd t |tjddg|jd d|_t |ddgddggddgdt |ddgddggddgdt||f|ddd }t |ddgddggdd gddS)!Nc ||zSNrprqs r(fz;TestRegularGridInterpolator.test_length_one_axis..fOs q5Lr*rYrrmTrnsparserTFerrrr)r- rur\gffffff?333333@)333333@333333@rrr333333?g'@g?g)@r[r3gffffff?rg?gffffff@rlr)rrrxr rr^r)r!rrrrsdatarDs r(test_length_one_axisz0TestRegularGridInterpolator.test_length_one_axisKs:    K1a  K2r " "q"+aT$???@(!Qh6;MMM rx!Q!Q!R(ABBCC" " $ $ $ $ rx!SAs8aW(EFFGG&" $ $ $ $ rxc 3344)" $ $ $ $ !C1d)455d % 1 1 1 1 c S$K899d % 1 1 1 1)!Qi6;NNNc RI677A" $ $ $ $ $ $r*rNc|d|d}tjddtjzd}tj|}t |f|ddfi|}t |dgf|dddffi|}tjddtjzdzd}||} tjd} |tj|| gj} t| | tj d} |tj|| gj} |t| | dSt| |dS) NFrrrrr-rrrYr) rrpisinr zerosvstackTrr) r!rroptionsrrzfafbx1azay1bzbs r(test_length_one_axis2z1TestRegularGridInterpolator.test_length_one_axis2vsT",U#%% K1RU7B ' ' F1II $aT1QQQ4 ; ;7 ; ; $a!XqDz E EW E Ek"agai-- RWWhsmm R 3*%%' ( (Bgcll R 3*%%' ( (   B # # # # # B + + + + +r*cZtgdfgddd|}tj|tjgsJtjd}|dd z}|dd k}tj||<tjd 5||}dddn #1swxYwYt||tjt|||||gd}dg}tjd }t||f|dd|}tj|tjdgsJtj|dtjgsJdS)N)rYr-rlrrYFrlJrsizerrignoreinvalid)rlrY) r risnanrr default_rngerrstaterr) r!rrrngrriresrsrs r( test_nan_x_1dz)TestRegularGridInterpolator.test_nan_x_1ds $YYYL,,,116v G G Gx26( $$$$$i##J// JJCJ  " JJCJ 3 &v! [ * * *   !A$$C                 SVRV$$$S!Waa1"hh''' II Ewv #QFDQ16v G G Gx261+'''''x1bf+'''''''s: CCCc^tjgdtjgd}}d}tj||dd\}}|||}t||f||d}tjd 5|d tjgd d gg} dddn #1swxYwYt | d d dtj| dsJtj d} | ddzd z }| ddz}| ddk} | ddk} | | z} tj|| <tj|| <tj||gj }tjd 5||} dddn #1swxYwYt| | tjt| | ||| dS)N)rrYr-)rYrlrc|dz|dzzSNr-rprqs r(rz4TestRegularGridInterpolator.test_nan_x_2d..fsa4!Q$; r*rmTrFrrrrr3rYr-rur\rrrrrr) rr^rxr rrrrrrrr)r!rrrrsrrzr{rrDrri1i2rrs r( test_nan_x_2dz)TestRegularGridInterpolator.test_nan_x_2dsx ""BHYYY$7$71   QD>>>BqRyy(!Q06ULLL[ * * * 2 2&3-!Q011C 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2A....xAi##J// JJCJ  "1 $ JJCJ  " ZZSZ ! !C ' ZZSZ ! !C ' G"" HaV    [ * * *   &))C                 SVRV$$$S!WffQrUmm,,,,,s$B))B-0B- GG"Gr)ndimsfuncr-c$d|dzzd|dzzzSNr-rlrprqs r(rtz$TestRegularGridInterpolator.sQ!Va!q&j0r*rlc*d|dzzd|dzzz|z Srrp)rrrsrs r(rtz$TestRegularGridInterpolator.s AQJQ!V3a7r*rc0d|dzzd|dzzz|z |zSrrprrrsras r(rtz$TestRegularGridInterpolator.s%q16zAQJ6:Q>r*c6d|dzzd|dzzz|z ||zzSrrp)rrrsrrbs r(rtz$TestRegularGridInterpolator.s)!a1f*q16z"9A"=A"Er*c|dkr|dvrtjdtjd}dd|d|f}fd t |D}|tj|d d d }t||| }||} d|D} |tj| d d d } t| | | } | |} t| | dS)Nr>rHrIz-too slow; OOM (quintic); or nearly so (cubic)*rYrr-rc<g|]}tjdS) )rr)r_ sample_high sample_lows r(rzITestRegularGridInterpolator.test_descending_points_nd..s7333 !K KDD333r*rmTrrAc$g|] }|dddS)Nrrp)rxis r(rzITestRegularGridInterpolator.test_descending_points_nd..s"AAA"R"XAAAr*) rNrerrruniformrangerxr r)r!rrrr test_pointsascending_pointsascending_valuesascending_interpascending_resultdescending_pointsdescending_valuesdescending_interpdescending_resultr r!s @@r(test_descending_points_ndz5TestRegularGridInterpolator.test_descending_points_nds A::&$888 KG H H Hi##B''  kk*kE kKK 33333%*5\\333 4.>6:48":":":;33C3C:@BBB,+K88AA0@AAA D"+/@7;59#;#;#;<44E4E;ACCC.-k::+->?????r*c"d}tjgd}tjgd}||f}|tj|ddd}d}tjt |5t ||ddddS#1swxYwYdS) Nc$d|dzzd|dzzzSrrprqs r( val_func_2dzJTestRegularGridInterpolator.test_invalid_points_order..val_func_2dsqAv:AF * *r*rr4r@@rr4r6r4r5rmTr(must be strictly ascending or descendingrL)rr^rxrNrrOr )r!r2rrrsr"r#rMs r(test_invalid_points_orderz5TestRegularGridInterpolator.test_invalid_points_orders + + + H*** + + H*** + +Qbk6D157778: ]:U 3 3 3 4 4 #FF 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s&BB Bcttjdgtjd|d}tj|dgsJdS)NrFr r)r rrrr)r!rrDs r(test_fill_valuez+TestRegularGridInterpolator.test_fill_value sZ(")A,,06ULLLxt %%%%%%%r*cb|dkrtjddgdzdgdzz}tjd}|d}|d}t |||d }||}t |jd | g}td D];} t ||d| f|d }| ||rHrIr<r=r2)rr.r/r?rr@A@B@) rr.r/r?rr@rPrQ/cFg|]}tt|Srp)tuplereversedrs r(rzGTestRegularGridInterpolator.test_nonscalar_values_2..As&999QeHQKK((999r*rrlr-rrr  rFr rYr.raxisrurF) rNrerrrr remptyr%item expand_dimsr)r!rrMr"rtrailing_pointsr#rCrDrIrJrrKrFs r(test_nonscalar_values_2z3TestRegularGridInterpolator.test_nonscalar_values_21s ) ) ) K ( ( (DDD  :99&999Fi##D)) Q1a:/::;;A(6;=== F6NNw1///////Xfl233' ( (v|B'(( 1 1A6<+,, 1 10Q 9J8>>CEEE"6&>>..001a4  1 ^BQ ' ' '2E6::::::r*c d}ddg}tjd}d}|ddg|R}|d}t|||d }||}|jd g|RksJtj|jd d} t |jd D]]} t |jd D]@} t||d| | f|d }||| | | f<A^tj| d} t|| d|dS)NrTr=r2r)rlrrrr-Fr rYrXr.rrYrurF) rrrr rr[r%r\r]r) r!rr"rr^r#rCrDrIrJrrKrFs r(test_nonscalar_values_linear_2Dz;TestRegularGridInterpolator.test_nonscalar_values_linear_2D\s059i##D)) Q4O4455A(6;=== F6NNw1///////Xfl233' ( (v|B'(( 1 1A6<+,, 1 10Q 9J8>>CEEE"6&>>..001a4  1 ^BQ ' ' '2E6::::::r*rxi_dtypecrd}tjddd}tjddd}tj||dd \}}|||}||}t ||f|} tjd d gd d gg|} t | | ddgddS)Nc$d|dzzd|dzzzSrrprqs r(rz:TestRegularGridInterpolator.test_float32_values..fsq!t8a!Q$h& &r*rYrrrrmTrrg@ffffff @g@rg Y`@g{,c@gHz>r\)rrrxastyper r^r) r!rrbrrrrsrzr{rrDptss r(test_float32_valuesz/TestRegularGridInterpolator.test_float32_values{s ' ' ' K1b ! ! K1b ! !QD>>>BqRyy{{5!!(!Q66hc c $+3555 s lL%AMMMMMMr*ctjddd}tjddd}tj||dd\}}||z}tt5t ||f|ddddn #1swxYwYtt 5t ||f|d d dddn #1swxYwYtt 5t ||f|d d d dddn #1swxYwYtt 5t ||f|d d diddddS#1swxYwYdS)NrrlrrmTrc|Srrprrs r(rtz=TestRegularGridInterpolator.test_bad_solver..s1r*)solverrUc|Srrprls r(rtz=TestRegularGridInterpolator.test_bad_solver..r*)rrmc|Srrprls r(rtz=TestRegularGridInterpolator.test_bad_solver..ror*woof)rrmrqr)r solver_args)rrrxr_rOr TypeError)r!rrrsrzr{rs r(test_bad_solverz+TestRegularGridInterpolator.test_bad_solvers K1a  K1a QD>>>BBw: & & F F #QFD E E E E F F F F F F F F F F F F F F F9 % %   #AY{{                    9 % %   #AY{{                    9 % %   #AYfb\                      sH!BB B"CC  C $DD D 'E  EE)4__name__ __module__ __qualname__r)r0r7r9parametrize_rgi_interp_methodsrGrNmark parametrizerPrrrWr`rjr~rrrrrrrrrrrrrrrrrrrr fail_slowr/r8r:rLr_rafloat32float64 complex64 complex128rirtrpr*r(rrs            $  $#  [X'D'D'DEE * *FE * [ )))***)))+ zrz*?*?*?@@ A        ///($  $# K K K:::::: [ RZ,,, - -v 6 RZ,,, - -s 3 RZ,,, - -s 3 RZ,,, - -v 6 RZ,,, - -v 6    ::  : :::,,,$222 K K K K K K : : : H H H H H H [X 8'<==   >=   7 7 7???)$)$)$V [\D"&"%+@AA [X)'<==,,>=BA,4 [X 8'<==((>=(< [X 8'<=="-"->="-H [1# [. 0 01 7 78 > >? E EF 1 @@ $#@B 4 4 4$&&$#&  [1#;;$#;:$ []UDM::';';;:$#';R;;;> [ RZr}= [Z"*bj)ABBNNCB  N.r*rc,eZdZdZdZdZdZddZdS)rz" Minimal indexable object cd|_||_tjtj|||_dSr )ndimrrrprodrr)r!rs r(__init__zMyValue.__init__s;  )BGENN++33E::r*c|j|Sr)r)r!idxs r( __getitem__zMyValue.__getitem__sws|r*cdSrrp)r!s r(__array_interface__zMyValue.__array_interface__str*Nc td)NzNo array representation) RuntimeError)r!rcopys r( __array__zMyValue.__array__s4555r*)NN)rurvrw__doc__rrrrrpr*r(rrs_;;; 666666r*rceZdZdZdZedZdZdZdZ dZ dZ d Z d Z d Zed Zejd edZedZdZedZdZdZejdddgdZedZdZdZdZdZdZ dZ!dZ"ejd d!d"gd#Z#d$S)% TestInterpNc tjgd}tjgd}tjgdgdgdgdgdgdg}|||fS)N)rr4r6r4r5g@)rYr-rYr-rYrY)rYr-rlr-rYrY)rYr-r-r-rYrY)rr^)r!rrrsrs r(_sample_2d_datazTestInterpN._sample_2d_datas H... / / H... / / H""""""""""""""""""   !Qwr*c ,|\}}}t|||}tjgdgdgj}t t ||f||d||dddf|dddfdS)NrYffffff@rrrf333333?rlrYrfrr4r.rrl splinef2drArrY)rr rr^rrr ev)r!rrrsrlutr#s r(test_spline_2dzTestInterpN.test_spline_2ds&&((1a!!Q** X6666668999: !'1a&!R "L"L"L"%&&AAAqD2aaad8"<"< > > > > >r*c||\}}}tjgdgdgj}t ||f|||}t ||f|||}t |||dS)NrrrArD)rrr^rr rBr)r!rrrrsrr#rErFs r(rGzTestInterpN.test_list_inputs&&((1a X6666668999: aVQ6 2 2 2  XXZZ $ahhjj"))++f    B//////r*c tjgd}tjgd}tjgdgdgdgdgdg}t|||}tjgdgdgj}t ||f||ddd }||ddd f|ddd f}d |d d<t ||ttt||f||ddd dS)Nr6rYr-rYr-rYrYr-rlr-rYrYr-r-r-rYrYrrrrfrrlrYrfrgr.rrlrFR?@rrrYr-r) rr^r rr rrr_rO)r!rrrsrrr#actualexpecteds r(test_spline_2d_outofboundsz&TestInterpN.test_spline_2d_outofboundssJ H*** + + H*** + + Hooo%oo8 9 9!!Q** X6667779:::; !QB{&+@@@66"QQQT(Bqqq!tH--1 !&(333 j'Aq61b#(T ; ; ; ; ; ;r*cdgdzdgdzz}tjgd}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fSr,rr s r(_sample_4d_datazTestInterpN._sample_4d_datas!#}o&99LLL))BJ BJ>?QQQ BJ>?RZBJ>?RZQQQ>?GbL(7S=87T>Iv~r*c|\}}t||}tjgdg}t |||d}t |||dS)NrQrTrArr rrr rr!r"r# interp_rgrCrs r(test_linear_4dzTestInterpN.test_linear_4d su--//+FF;; 000122AAA!))F"3"3V<<<<rrArBFr rCrDc Hg|]}td|fdS.Fr r rrKrr"rCr#s r(rz5TestInterpN.test_nonscalar_values..jsK===./ffS!VnfV#(***===r*r rYrrurF) rNrerrrr rrr%r^rHr) r!rrrIrJrFr"rCr#s ` @@@r(rLz!TestInterpN.test_nonscalar_valuesVs/ Y   K ( ( (11A5 ,9 9i##D))O,,I&& FFF6!& ( ( (QWi8888=======3888=== Xb\\ # #Aq! , ,2E6::::::r*cdvrtjdgdtjd}d}|dddd g|R|d t d }|jd g|RksJfdtjdD}t|tj |j ddS)N>rHrIr<rOrrVrrr rWrFr rYcfg|],fdtjdD-S)c Jg|]}td|fd Srr)rrrKrr"rCr#s r(rzBTestInterpN.test_nonscalar_values_2...sN01sAqy 16&%*,,,r*rX)r%rrs @r(rz7TestInterpN.test_nonscalar_values_2..sp2225:6<;K5L5L222r*rrurF) rNrerrrr rr%rrr) r!rrr^rIrJr"rCr#s ` @@@r(r_z#TestInterpN.test_nonscalar_values_2ps/ ) ) ) K ( ( (DDD i##D)) Q1a:/::;;A FFF6 N N Nw1///////2222222v|B/00222 2:b>>+%HHHHHHr*c8|\}}tjdtjddddd}tjddd}t t t|||ddS) NrrlrrrrrrA)rrrrrr_rOr )r!r"r#rCs r( test_non_scalar_values_splinef2dz,TestInterpN.test_non_scalar_values_splinef2ds--// t1aA..2q))j'666( * * * * * *r*cv|dkrtjd|\}}}||f}|d|zz }tjgdgdgj}t ||||}t ||j||}t ||j||} |d| zz} t|| dS)NrJrbrcrrrArd) rNrerrr^rr rfrgr) r!rrrrsr#r"rCrEv2rv2irFs r(rjzTestInterpN.test_complexs W   KD E E E++-- 1fQ"V)#::::::<===> VVVF ; ; ;ffk6&AAAffk6&AAA 2c6\Br*c|\}}}||f}|d|zz }tjgdgdgj}t jd5t |||dddddS#1swxYwYdS)NrcrrcomplexrLrJrA)rrr^rrNdeprecated_callr r!rrrsr#r"rCs r(test_complex_pchipzTestInterpN.test_complex_pchips++-- 1fQ"V)#::::::<===>   #) 4 4 4 < < FFF7 ; ; ; ; < < < < < < < < < < < < < < < < < A>c |\}}}||f}|d|zz }tjgdgdgj}t t 5t |||dddddS#1swxYwYdS)NrcrrrrA)rrr^rrrr rs r(test_complex_spline2fdz"TestInterpN.test_complex_spline2fds++-- 1fQ"V)#::::::<===>  . ) ) @ @ FFF; ? ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @sA99A=A=rrTrctjddd}tjddd}td}t||f|ddg| }t||f|jddg| }t ||dS) Nrr-rrYrrrrrA)rrrr rr)r!rrrrsr#rErFs r(rz"TestInterpN.test_duck_typed_valuess K1a  K1a  aVVc3Z ? ? ? aVVYc 6 B B BBr*ctjddd}tjddd}ttjdd}tjddd}t ||f|||}t ||ftj|||}|dkrt||d d dSt||dS) Nrr-rrYrrlrArIg-C6 ?g>rv)rrrrrr rr)r!rrrrsr#rCrErFs r(test_matrix_inputzTestInterpN.test_matrix_inputs K1a  K1a  q!,,--1a(( aVVVF ; ; ; aVRZ// G G G Y   BD 9 9 9 9 9 9 B # # # # #r*cVtjgdg}tjddgddgddgg}tdggdf||}gd}t||d tjd dgd dgd dgg}tdggdf||d d}t||d dS)N)r<rYrrYg@g @gffffff@)r-rlr)gQ?gffffff@gffffff @r[r\rr3gffffffFr)rr^r r)r!r#r#rrs r(rz TestInterpN.test_length_one_axiss <<<.)) X3x!SAs84 5 5sIII&33 V%0000XSzC:c{; < <sIII&#(T;;; V%000000r*c dd}tjgd}tjgd}tjgd}tjgd}||||f}|tj|ddd}dd tjtjd d d tjd d d f}t |||} |ddd } |ddd } |ddd } |ddd } | | | | f}|tj|ddd}t |||}t | |dS)Nc0d|dzzd|dzzz|z |z Srrprs r( value_func_4dz9TestInterpN.test_descending_points..value_func_4ds'qAv:AF *Q.2 2r*)rrYr-rl)rrrr)rr<r?rrmTrr<rrrrr)rr^rxrHrr r)r!rx1x2x3x4r"r#rhcorrect_result x1_descend x2_descend x3_descend x4_descendpoints_shuffledvalues_shuffled test_results r(test_descending_pointsz"TestInterpN.test_descending_pointssj 3 3 3Xlll # # Xooo & & Xooo & & X&&& ' 'b"b! [&4 = = =?Cbk!R&;&;<<{1c1%%' 55"X "X "X "X %z:zJ'- [/D F F FHoDD >;77777r*cttjgd}tjgd}tjgdgdgdgdgdg}tjgdgdgj}d}tjt | 5t ||f||ddddS#1swxYwYdS) Nr3r6rrrrrr7rL)rr^rrNrrOr )r!rrrsrr#rMs r(r8z%TestInterpN.test_invalid_points_orders' H*** + + H*** + + Hooo%oo8 9 9 X6667779:::; ; ]:U 3 3 3 # # QFAr " " " # # # # # # # # # # # # # # # # # #s B--B14B1cdg}ddg}tjd}d}tt|5t |||ddddS#1swxYwYdS)N)rrYrrY)rYrYrlzaThe requested sample points xi have dimension 3, but this RegularGridInterpolator has dimension 1rL)rrr_rOr )r!r"r#r#msgs r(test_invalid_xi_dimensionsz&TestInterpN.test_invalid_xi_dimensionssQ WY  9 :S 1 1 1 ( ( FFB ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sAAActjddd}tjddd}tjddd}|||f}tjd}tjgd}|D]}d|j_d|j_d|j_t |||t|||dS) NrrrrrrrrGz@(\@gffffff?Frrrr^flags writeabler r )r!rrrsrr"r#pointds r(test_readonly_gridzTestInterpN.test_readonly_grid)s K1a  K1a  K1a Q##+++,, & &A %AG  !&  % &&&///66666r*c`tjddd}tjddd}||f}tjd}tjddg}|D]}d|j_d|j_d|j_t |||t|||dS) NrrrrrrrrFr)r!rrrsr"r#rrs r(test_2d_readonly_gridz!TestInterpN.test_2d_readonly_grid8s K1a  K1a Q$&& & &A %AG  !&  % &&&///66666r*ctjddd}tj|tj|fjdddf}|jjrJtjddd}tjddd}|||f}tjd}tj gd}t|||t|||dS)Nrrrrrrr) rrr empty_likerrr c_contiguousrr^r r )r!rrrsrr"r#rs r(test_non_c_contiguous_gridz&TestInterpN.test_non_c_contiguous_gridGs K1a  Iq"-**+ , , . 3 3 5 5aaad ;7'''' K1a  K1a Q##+++,,&&&///66666r*rz>f8z>>$00$#0;;;&===222===222    2 2 2$22$#2& [1#;;$#;0$II$#I@***$  $# $ < < < @ @ @ [ 9   $$$$#$ 111,8884 # # #((( 7 7 7 7 7 7 7 7 7 [Wuen55 7 765 7 7 7r*r)rrNnumpyr numpy.testingrrrrrrr_scipy.interpolater r r r r scipy.sparse._sputilsrscipy._lib._utilrryrz _ALL_METHODSrxrrrrpr*r(r s JJJJJJJJJJJJJJ******LLLLLLLLLLLLLL)(((((++++++"(!8!8 %2""[ [ [ [ [ [ [ [ |66666666(V7V7V7V7V7V7V7V7V7V7r*