:j/:ddlZddlZddlZddlmZddlmZmZm Z m Z ddl m Z ddlm ZddlmZmZmZmZmZddlmZmZddlmZdd lmZej8j;d ej<Ze eGd d Z Gd dZ!GddZ"y)N)ComplexWarning)xp_assert_equalxp_assert_closeassert_array_almost_equalmake_xp_test_case)skip_xp_invalid_arg)raises)RegularGridInterpolatorinterpnRectBivariateSplineNearestNDInterpolatorLinearNDInterpolator)_regrid_ndbspline_call_like_bivariate)matrix)_run_concurrent_barriermethodc eZdZdZdZdZdZedZe jjdgddZ e jjd ee jgd gd gd gfee jgd fgdZdZedZdZdZdZe jjdgddfgddfgddfgddfgddfgdZdZd Zd!Zd"Zd#Zd$Zd%Zd&Ze jjdd'd(gd)Zd*Zd+Z e jjd,e jBe jDge jjd-gd.e jjdd(d'gd/Z#d0Z$e jjd1d2e jJe jLge jjdd(d'gd3Z'e jjdd'd(gd4Z(e jjdd'd(gd5Z)e jjUd6ee jjd7d8d9fd:d;fdd?fgd@Z+dAZ,edBZ-e jjUd>edCZ.ee jjdDdEdFgdGZ/dHZ0e jjd,e jBe jDe jbe jdge jjdIe jBe jDgdJZ3dKZ4dLZ5y2)MTestRegularGridInterpolatorc dgdz}|jgd}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fS)N?? dasarraynpnewaxisselfxppointsvaluesvalues0values1values2values3s u/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/scipy/interpolate/tests/test_rgi.py_get_sample_4dz*TestRegularGridInterpolator._get_sample_4d s!#L)BJJ BJJ>?Q BJJ>?RZZBJJ>?RZZQ>?GbL(7S=87T>Iv~cdgdzdgdzz}|jgd}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fSNr)r@$@rrrrr#s r,_get_sample_4d_2z,TestRegularGridInterpolator._get_sample_4d_2+s!#}o&99L)BJJ BJJ>?Q BJJ>?RZZBJJ>?RZZQ>?GbL(7S=87T>Iv~r.c dgdz}|jgd}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fS)Nrrr?@@@rrrrrr#s r,_get_sample_4d_3z,TestRegularGridInterpolator._get_sample_4d_36s56:?@BJJ BJJ>?Q BJJ>?RZZBJJ>?RZZQ>?GbL(7S=87T>Iv~r.c dgdz}|jddg}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fS)N)rrrrrrrrrr#s r,_get_sample_4d_4z,TestRegularGridInterpolator._get_sample_4d_4As!S#J'BJJ BJJ>?Q BJJ>?RZZBJJ>?RZZQ>?GbL(7S=87T>Iv~r.c|jt\}}tjgdgdgdg}t||j |}||j }t|||}||}t ||y)N)r%皙?r@r?皙?r@g?g?rrrrr)r;r!r r tolistr)r$rr&r'sampleinterpv1v2s r,test_list_inputz+TestRegularGridInterpolator.test_list_inputLs..".5/1D/12))/068FMMO $()/068F^Br.r)cubicquinticpchipc|j\}}tfd|D}d}tjt|5t |||dddt ||}j gdgdgdg}tjt|5|||dddy#1swY^xYw#1swYyxYw)Nc3@K|]}j|ywNr .0pr%s r, zDTestRegularGridInterpolator.test_spline_dim_error..`4bjjm4zpoints in dimensionmatchrEr?rBrD)r=listpytestr ValueErrorr r )r$rr%r&r'rZrHrGs ` r,test_spline_dim_errorz1TestRegularGridInterpolator.test_spline_dim_error]s..r24V44%]]:U 3 C #FF6 B C)8/1D/12 ]]:U 3 * 6& ) * * C C * *sB3 B?3B<?Czpoints_values, sampler?rBrDr@r@r3"@c||\}}tfd|Dj|}}t||d}||}t||d}||}t||y)Nc3@K|]}j|ywrQrRrSs r,rVzLTestRegularGridInterpolator.test_linear_and_slinear_close..~s<bjjmP(I F^( J F^Br.c4|j\}}tfd|D}jgdgdgdg}t||d}t t 5||ddddt ||d jgd j d t ||djgdj d t ||djgdj d y#1swYxYw)Nc3@K|]}j|ywrQrRrSs r,rVz?TestRegularGridInterpolator.test_derivatives..rWrXr?rBrDrdrE)nu)rirrr)rriridtypeV瞯<atol)rrirr)r3rr)rririr)rrrg-=)r-r[r r assert_raisesr]rfloat64)r$r%r&r'rGrHs ` r,test_derivativesz,TestRegularGridInterpolator.test_derivativess,,R04V4453356) J : & ! 6a  ! v,7 ;bjj A Ov,7 > D5 R v,7 ;bjj A O ! !s ! DDc|dk(rtjd|j\}}tfd|D}|d|zz }j gdgdgdg}t |||}t |j ||}t |j||}||} ||d ||zz} t| | y) NrN*pchip does not make sense for complex datac3@K|]}j|ywrQrRrSs r,rVz;TestRegularGridInterpolator.test_complex..rWrX@r?rBrDrE?) r\skipr;r[r r realimagr) r$rr%r&r'rGrHrinterpiinterprIrJs ` r, test_complexz(TestRegularGridInterpolator.test_complexs W  KKD E..r24V44"V)#/1D/12)G)&"''&/&Q)&"''&/&Q F^ V_r'&/1 1Br.cJ|jgd|jgd}}|j||d\}}d||}t||f|d}t||f|d}|dd g} |dd g} |j|j | | z d krJy) N)rir1rij)indexingc|dz|dzzS)Nrxys r,zATestRegularGridInterpolator.test_cubic_vs_pchip..sq!tad{r.rLrErNr7r1+=)r meshgridr allabs) r$r%rrxgygr'rLrN vals_cubic vals_pchips r,test_cubic_vs_pchipz/TestRegularGridInterpolator.test_cubic_vs_pchipszz,'L)A1QD1B*B3'AwG'AwGC8_ C8_ 66"&&j!89EABBBBr.c|j\}}tfd|D}t||}jgd}jdgj}t |||y)Nc3@K|]}j|ywrQrRrSs r,rVz?TestRegularGridInterpolator.test_linear_xi1d..rWrXr_H@rkr4r[r r rqrr$r%r&r'rHrGwanteds ` r,test_linear_xi1dz,TestRegularGridInterpolator.test_linear_xi1dsh..r24V44(8/0VHBJJ7!&.&9r.c|j\}}tfd|D}t||}jgdgdgdg}jgd}t |||y)Nc3@K|]}j|ywrQrRrSs r,rVz?TestRegularGridInterpolator.test_linear_xi3d..rWrXr?rBrDrgq@g\@r-r[r r rrs ` r,test_linear_xi3dz,TestRegularGridInterpolator.test_linear_xi3dsl,,R04V44(8/1D/1223!&.&9r.zsample, wanted)r@r@rArAg0@)r@r@r@r@rrrrrrrrr\@)r@皙?g333333?rAg|@c|j\}}tfd|Dj|}}t||d}j|gj}t |||y)Nc3@K|]}j|ywrQrRrSs r,rVz;TestRegularGridInterpolator.test_nearest..s=rzz!}=rXnearestrErk)r-tupler r rqr)r$rGrr%r&r'rHs ` r, test_nearestz(TestRegularGridInterpolator.test_nearestsj,,R0=f==rzz&?Q( JVHBJJ7!&.&9r.c|j\}}tfd|D}t||}jgdgdg}jddg}t |||y)Nc3@K|]}j|ywrQrRrSs r,rVz@TestRegularGridInterpolator.test_linear_edges..rWrXrrrrrrs ` r,test_linear_edgesz-TestRegularGridInterpolator.test_linear_edgessh,,R04V44(8-/?@ARK(!&.&9r.cddg}tjgd}|ddtjf}|tjddf}||dzz}ttt ||ddg}ttt ||ddg}ttt ||gd}ttt ||ddg}ttt ||dy) Nr)rrrr)r)rrg?r)rrr undefmethodrE)r!r r"rpr]r )r$r&r'r(r)s r,test_valid_createz-TestRegularGridInterpolator.test_valid_creates -L)BJJ'Q'GbL(j"966J"L1j"966J#\2j"966J;j"966J -j"966* ,r.c|j\}}tfd|D}t||}jgdgdg}t t 5||ddddjgdgdg}t t 5||dddjgdgdg}t t 5||dddy#1swYxxYw#1swYMxYw#1swYyxYw)Nc3@K|]}j|ywrQrRrSs r,rVz>TestRegularGridInterpolator.test_valid_call..rWrXrrr)rrr)rrr)rrr皙?)r-r[r r rpr])r$r%r&r'rHrGs ` r,test_valid_callz+TestRegularGridInterpolator.test_valid_calls,,R04V44(8-/?@A : & * 6= ) *\<89 : &  6N -/@AB : &  6N   * *    s$ C C) C5C&)C25C>c|j\}}tfd|D}t||dd}jgdgdgdgdgj}jgd j}t ||d |jgd j}t ||d |y)Nc3@K|]}j|ywrQrRrSs r,rVzHTestRegularGridInterpolator.test_out_of_bounds_extrap..rWrXF bounds_error fill_value皙rrrrrrr@皙irrrrrk)rr&@rrrE)gfffff[gfffff@ggrc)r-r[r r rqrrs ` r,test_out_of_bounds_extrapz5TestRegularGridInterpolator.test_out_of_bounds_extraps,,R04V44(e48:13G13IK"$**.1D!& "BFK>bjjQ!&"A6Jr.c|j\}}tfd|D}t||dd}jgdgdgdgdgj}jgd j}t ||d |jgd j}t ||d |y)Nc3@K|]}j|ywrQrRrSs r,rVzITestRegularGridInterpolator.test_out_of_bounds_extrap2..rWrXFrrrrrrk)rrrrrrE)g333333(g33333`@ggyXrcrrs ` r,test_out_of_bounds_extrap2z6TestRegularGridInterpolator.test_out_of_bounds_extrap2s..r24V44(e48:13G13IK"$**./rzzB!& "BFK9L!&"A6Jr.c|j\}}tfd|D}t||dj}j gdgdgdg}j jjjg}t ||d|t ||d |j gd gd gd g}j gd }t |||y)Nc3@K|]}j|ywrQrRrSs r,rVzFTestRegularGridInterpolator.test_out_of_bounds_fill..!rWrXFrrrrrrErcr?rBrDr)r-r[r nanr rrs ` r,test_out_of_bounds_fillz3TestRegularGridInterpolator.test_out_of_bounds_fills,,R04V44(e46FF<13G356RVVRVVRVV45!& "BFK!&"A6J/1D/1223!&.&9r.cl|jt\}}t||d}tj|}|Dcgc]}|}}tj |}|j d}t||}tj gdgdgdg}t||||ycc}w)NrrEr?rBrD) r-r!r itertoolsproductr reshaper r r$r&r'rH points_qhullrU values_qhull interp_qhullrGs r,test_nearest_compare_qhullz6TestRegularGridInterpolator.test_nearest_compare_qhull.s,,R0( J ((&1 #/0a0 0zz,/ ~~b) ,\<H /1D/12!&.,v2FG 1s B1ch|jt\}}t||}tj|}|Dcgc]}|}}tj |}|j d}t||}tj gdgdgdg}t||||ycc}w)Nrr?rBrD) r-r!r rrr rrrrs r,test_linear_compare_qhullz5TestRegularGridInterpolator.test_linear_compare_qhull:s,,R0(8 ((&1 #/0a0 0zz,/ ~~b) +L,G /1D/12!&.,v2FG 1s B/rrcctjddd}tjddd}td}t||f||}|dd g}t||f|j|}|dd g}t ||d y) Nrr1rirrrErffffff?F check_dtype)r!linspaceMyValuer _vr)r$rrrr'rHrIrJs r,test_duck_typed_valuesz2TestRegularGridInterpolator.test_duck_typed_valuesFs KK1a  KK1a (!QG S#J (!Q6J S#J BE2r.c2tjjdtjddd}tjddd}tjj dd}t ||f|dt tt ||f|dy) Nrr1rrirry?@)r!randomseedrrandr rpr])r$rrr's r,test_invalid_fill_valuez3TestRegularGridInterpolator.test_invalid_fill_valueTsv t KK1a  KK1a 1% A1= j"9!ff 7r.ctjdd}|jDcgc]}tj|}}t ||t ||dycc}w)Nrz>f4rkrr)r!onesshapearanger )r$r'nr&s r,test_fillvalue_typez/TestRegularGridInterpolator.test_fillvalue_typeasLU3(. 51"))A,55/2>6sArlndimrir1rc4tj||}td|D}tjd|}tjd|z|}tjd|}|dk(rtj n|} t |||d| } t |||d } tj|g} tj|g| } | | | | fD]}t|| |d xxd z cc<tj|g} tj|g| } | | }t|| tjtd5| | dddtj|d <tj|g} tjtjg| } | | }t|| tjtd5| | dddy#1swYxYw#1swYyxYw)Nrkc3HK|]}tj|gywrQ)r!r )rTxis r,rVzGTestRegularGridInterpolator.test_length_one_axis_all..ws5Brzz2$'5s "g$I$I?)ri)rrgaa?rcFrrrTrrrriz9^One of the requested xi is out of bounds in dimension 0$rY) r!rrr fullrqr rr\r r]r)r$rlrrx0r&valr'fillpromoted_dtype interp_fill interp_errrGrresults r,test_length_one_axis_allz4TestRegularGridInterpolator.test_length_one_axis_allisYYt5 )5"55jjE*uTzc:zz$e,(.'9u. F6$ - F6 RD!SE8"6*Jv,>? ,F FF + , 1 RD!TF.9V$' ]] M   v   1RD!RVVHN;V$' ]] M   v        s! H0 HH Hc d}tjddd}tjddd}|tj||dd}t||f|ddd }t |tj ddgdd gddggtj gd d t |tj ddgddgddgggdd t |tj ddg|jdddd d|_t |ddgddggddgdt |ddgddggddgdt||f|ddd }t |ddgddggtj d d!gdy)"Nc ||zSrQrrs r,fz;TestRegularGridInterpolator.test_length_one_axis..fs q5Lr.rirrTrsparsercFerr)r8 rrngffffff?333333@)333333@333333@rrr)r check_shapecheck_0dro333333?g'@g?g)@rmr7gffffff?rg?gffffff@r:r)r!rrr rarrayr r)r$rrrdatarHs r,test_length_one_axisz0TestRegularGridInterpolator.test_length_one_axiss  KK1a  KK2r ""++aT$?@(!Qh6;M rxx!Q!Q!R(ABC <0" $ rxx!SAs8aW(EFG&" $ rxxc 34))$)uu" $ !C1d)45d % 1 c S$K89d % 1)!Qi6;Nc RI67 C8," $r.rNc|d|d}tjddtjzd}tj|}t |f|ddfi|}t |dgf|dddffi|}tjddtjzdzd}||} tj d} |tj || gj} t| | tjd} |tj || gj} | t| | yt| tj| |y) NFrrrrr1rrrir) r!rpisinr zerosvstackTrr full_like) r$rroptionsrzfafbx1azay1bzbs r,test_length_one_axis2z1TestRegularGridInterpolator.test_length_one_axis2s ",U#% KK1RUU7B ' FF1I $aT1Q4 ;7 ; $a!XqDz EW Ekk"agai- Whhsm  3*%'' (Bggcl  3*%'' (   B # B R < =r.ctgdfgddd|}tj|tjgsJtjj d}|j dd z}|j dd kD}tj||<tj d 5||}dddtj|jsJt|||||gd}dg}tjd }t||f|dd|}tjtj|tjdgsJtjtj|dtjgsJy#1swYxYw)NrrriFrlJrsizerrignoreinvalid)rri) r r!isnanrr default_rngerrstaterrr) r$rrrngriresrr s r, test_nan_x_1dz)TestRegularGridInterpolator.test_nan_x_1dso $YL,116v Gxx266( $$$ii##J/ JJCJ  " JJCJ 3 &vv! [[ *  A$C  xxA##%%%QB1aR5*  Ewwv #QFDQ16v Gvvbhhq"&&!~.///vvbhhq!RVV~.///#  s 4 F66F?c*tjgdtjgd}}d}tj||dd\}}|||}t||f||d}tjd 5|d tj gd d gg} dddt  d d dtj| dsJtjjd} | jddzd z }| jddz}| jddkD} | jddkD} | | z} tj || <tj || <tj||gj}tjd 5||} dddtj| | jsJt| | ||| dy#1swYcxYw#1swY[xYw)N)rrir1)rirrc|dz|dzzSNr1rrs r,rz4TestRegularGridInterpolator.test_nan_x_2d..fsa4!Q$; r.rTrFrr$r%r7rir8rrnrr!rr"rrr) r!r rr r)rrr'rr(rrr)r$rrrrrrr rHr,r*i1i2r+rs r, test_nan_x_2dz)TestRegularGridInterpolator.test_nan_x_2dsxx "BHHY$71 QD>BRy(!Q06UL[[ * 23-!Q01C 2A%0xxAii##J/ JJCJ  "1 $ JJCJ  " ZZSZ !C ' ZZSZ !C ' G"" HHaV    [[ *  )C  xxA##%%%QB1"EB/ 2 2  s>G<& H <H Hr)ndimsfuncr1c$d|dzzd|dzzzSNr1rrrs r,rz$TestRegularGridInterpolator.8sQ!Va!q&j0r.rc*d|dzzd|dzzz|z Sr8r)rrrs r,rz$TestRegularGridInterpolator.9s AQJQ!V3a7r.rc0d|dzzd|dzzz|z |zSr8rrrras r,rz$TestRegularGridInterpolator.:s%q16zAQJ6:Q>r.rc6d|dzzd|dzzz|z ||zzSr8r)rrrr<bs r,rz$TestRegularGridInterpolator.;s)!a1f*q16z"9A"=A"Er.c|dk\r|dvrtjdtjj d}d}d}|j ||d|f}t |Dcgc]}tj||d } }|tj| d d d } t| | | } | |} | D cgc] } | ddd }} |tj|d d d }t||| }||}t| |ycc}wcc} w)Nr>rLrMz-too slow; OOM (quintic); or nearly so (cubic)*rirr1r" rTrrEr) r\rxr!rr(uniformrangerrr r)r$rr5r6r* sample_low sample_high test_points_ascending_pointsascending_valuesascending_interpascending_resultrdescending_pointsdescending_valuesdescending_interpdescending_results r,test_descending_points_ndz5TestRegularGridInterpolator.test_descending_points_nd5sB A:&$88 KKG Hii##B'  kk*kE kK &+5\3 !KK KD33 .>6:48":;33C3C:@B,K80@A"R"XAA "++/@7;59#;<44E4E;AC.k:(*;<+3Bs %D8D cd}tjgd}tjgd}||f}|tj|ddd}d}tjt |5t ||dddy#1swYyxYw) Nc$d|dzzd|dzzzSr8rrs r, val_func_2dzJTestRegularGridInterpolator.test_invalid_points_order..val_func_2d_sqAv:AF * *r.rr8r@@rr8r:rUrVrTr(must be strictly ascending or descendingrY)r!r rr\r r]r )r$rSrrr&r'rZs r,test_invalid_points_orderz5TestRegularGridInterpolator.test_invalid_points_order^s~ + HH* + HH* +Qbkk6D1578: ]]:U 3 4 #FF 3 4 4 4s , BB cttjdgtjd|d}tj|dgsJy)NrFrr)r r!rrr')r$rrHs r,test_fill_valuez+TestRegularGridInterpolator.test_fill_valueks@("))A,06ULxxt %%%r.c|dk(rtjddgdzdgdzz}tjj d}|jd}|jd}t |||d }||}|j d k(sJ|g}td D]-} t ||d | f|d }|j||/tj|jddd} t|| d|yNrM Way too slow.rrrr7r8r9r1)rr2r3.@r9@r)rrrrr1)rrrFr)rrr1r1.rirrroerr_msg) r\rxr!rr(r rrCappendr  transposer) r$rr&r*r'rGrHvvsjrJs r,test_nonscalar_valuesz1TestRegularGridInterpolator.test_nonscalar_valuesqs Y  KK (11A5 ,9 9ii##D)O,I&(6;= 6Nww)#+V+# q &A,VVCF^4::?AF IIfVn %  & XXb\ # #Aq! ,2E6:r. flip_pointsFTc |dvrtjdgd}|r!|Dcgc]}tt|}}tj j d}d}|j dddd g|}|j d }t|||d } | |} | jd g|k(sJt j|jdd} t|jdD]O} t|jdD]2} t||d| | f|d } | |j| | | f<4Qt j| d}t| |d|ycc}w)N>rLrMr^r_r6)rr2r3r`rraA@B@) rr2r3r`rrarmrn/rrr1rrr1 rFrrir.raxisrrb)r\rxrreversedr!rr(r remptyrCitem expand_dimsr)r$rrjr&rUr*trailing_pointsr'rGrHrfrgr+rhrJs r,test_nonscalar_values_2z3TestRegularGridInterpolator.test_nonscalar_values_2sr ) ) KK (D 289QeHQK(9F9ii##D) Q1a:/:;A(6;= 6Nww1/////XXfll23' (v||B'( 1A6<<+, 10Q 9J8>>CE"&>..01a4  1 1 ^^BQ '2E6:3:sE*c \d}ddg}tjjd}d}|jddg|}|jd}t|||d }||}|jd g|k(sJtj |jd d} t |jd D]O} t |jd D]2} t||d| | f|d }||j| | | f<4Qtj| d} t|| d|y)Nrcr_r6r)rrrrr1Frrirrr.rrsrrb) r!rr(r rrvrCrwrxr) r$rr&r*ryr'rGrHrfrgr+rhrJs r,test_nonscalar_values_linear_2Dz;TestRegularGridInterpolator.test_nonscalar_values_linear_2Ds=059ii##D) Q4O45A(6;= 6Nww1/////XXfll23' (v||B'( 1A6<<+, 10Q 9J8>>CE"&>..01a4  1 1 ^^BQ '2E6:r.xi_dtypecXd}tjddd}tjddd}tj||dd \}}|||}|j|}t ||f|} tj d d gd d gg|} t | | ddgdddy)Nc$d|dzzd|dzzzSr8rrs r,rz:TestRegularGridInterpolator.test_float32_values..fsq!t8a!Q$h& &r.rirrrrTrrg@ffffff @g@rkg Y`@g{,c@gHz>F)rortolr)r!rrastyper r r) r$rlr}rrrrrr rHptss r,test_float32_valuesz/TestRegularGridInterpolator.test_float32_valuess ' KK1b ! KK1b !QD>BRy{{5!(!Q6hhc c $+35 s lL%A!% Ar.cTtjddd}tjddd}tj||dd\}}||z}tt5t ||f|ddddtt 5t ||f|d d dddtt 5t ||f|d d d dddtt 5t ||f|d d didddy#1swYxYw#1swYlxYw#1swYMxYw#1swYyxYw)NrrrrTrc|SrQrrs r,rz=TestRegularGridInterpolator.test_bad_solver..s1r.)solverrdc|SrQrrs r,rz=TestRegularGridInterpolator.test_bad_solver..r.)rrc|SrQrrs r,rz=TestRegularGridInterpolator.test_bad_solver.. rr.woof)rrrr@)r solver_args)r!rrrpr]r TypeError)r$rrrrr s r,test_bad_solverz+TestRegularGridInterpolator.test_bad_solvers! KK1a  KK1a QD>BBw: & F #QFD E F9 %  #AY{   9 %  #AY{   9 %  #AYfb\    F F        s0C:D2DD:DDDD'c|jt\}}tjgdgdgdgdgt||d}ddgfd}t d ||y) Nr?rBrD)r r@rCrrdrErcb|j}|dz}|||j|usJy)Nr1rE)_spline)tidrHsplinermethodsrGs r, worker_fnz?TestRegularGridInterpolator.test_concurrency..worker_fns6^^FS1W%F 6& )>>V+ ++r.r)r-r!r r r)r$r&r'rHrrrGs @@r,test_concurrencyz,TestRegularGridInterpolator.test_concurrencyse,,R0333356) Ji( , Iv6r.)6__name__ __module__ __qualname__r-r4r;r=parametrize_rgi_interp_methodsrKr\mark parametrizer^r!r rfrrr}rrrrrrrrrrrrrrrfloat32rqrrrrrr-r4 fail_slowrPrYr[rirzr| complex64 complex128rrrrr.r,rrs    $ $  [[X'DE*F*  [[ %&%' zrzz*?@ A     O*$ $ " C:: [[ !6 * !3 ' !3 ' !6 * !6 *   : ::,$ K K : H H [[X 8'<= 3> 3 7? [[Wrzz2::&>? [[VY/ [[X)'<=;>0@;z*$X [[\D"&&"%%+@A [[X)'<=>>B>4 [[X 8'<=0>0< [[X 8'<="C>"CH [[2# [[. 01 78 >? EF 1 = $=B 4$&$&  [[1#;$;:$ [[]UDM:';;$';R;> [[ RZZr}}= [[Z"**bjj)ABAC  A087r.rc*eZdZdZdZdZdZddZy)rz" Minimal indexable object cd|_||_tjtj|j ||_yr0)rrr!rprodrr)r$rs r,__init__zMyValue.__init__,s3  ))BGGEN+33E:r.c |j|SrQ)r)r$idxs r, __getitem__zMyValue.__getitem__1swws|r.cyrQr)r$s r,__array_interface__zMyValue.__array_interface__4sr.Nctd)NzNo array representation) RuntimeError)r$rlcopys r, __array__zMyValue.__array__7s455r.)NN)rrr__doc__rrrrrr.r,rr's; 6r.rceZdZdZdZedZdZdZdZ dZ dZ d Z d Z d Zed Zej"j%d edZedZdZedZdZdZej"j3dddgdZeedZdZdZdZdZ dZ!dZ"dZ#ej"j3d d!d"gd#Z$y$)% TestInterpNc tjgd}tjgd}tjgdgdgdgdgdgdg}|||fS)N)rr8r:rUrVg@)rir1rir1riri)rir1rr1riri)rir1r1r1riri)r!r )r$rrrs r,_sample_2d_datazTestInterpN._sample_2d_data<sR HH. / HH. / HH""""""   !Qwr.c |j\}}}t|||}t|||}tjgdgdgj }t t||f||d|j|dddf|dddftt||f||dt||dddf|dddfdy) Nriffffff@rrr333333?rrirrrUr2rr splinef2drErriFgrid) rr rr!r rrr evrr)r$rrrlut lut_customrs r,test_spline_2dzTestInterpN.test_spline_2dKs&&(1a!!Q*Q1% XX66899: !'1a&!R "L"%&&AqD2ad8"< >A2kB"@&0"QT(Bq!tH+0#2 3r.cF|j\}}}tjgdgdgj}t ||f|||}t |j |j f|j |j |}t |||y)NrrrE)rc)rr!r rr rFr)r$rrrrrrIrJs r,rKzTestInterpN.test_list_inputYs&&(1a XX66899: aVQ6 2  XXZ $ahhj"))+f  B/r.c ftjgd}tjgd}tjgdgdgdgdgdg}t|||}t|||}tjgdgdgj}t ||f||ddd }|j |ddd f|ddd f}d |d dt||t||ddd f|ddd fd} d | d dt|| d | d dt|| ttt ||f||ddd y)NrWrir1rir1ririr1rr1ririr1r1r1ririrrrrrrrirrgr2rrrFR?@rrrir1rr) r!r r rrr rrrrrpr]) r$rrrrrractualexpectedexpected_customs r,test_spline_2d_outofboundsz&TestInterpN.test_spline_2d_outofboundsds: HH* + HH* + HHo%8 9!!Q*Q1% XX679::;! !QB{&+@66"QT(Bq!tH-1 !&(38 1a4"QT(8%!0%!0 j'Aq61b#(T ;r.c"dgdzdgdzz}tjgd}|ddtjtjtjf}|tjddtjtjf}|tjtjddtjf}|tjtjtjddf}||dzz|dzz|dzz}||fSr0)r!r r")r$r&r'r(r)r*r+s r,_sample_4d_datazTestInterpN._sample_4d_datas!#}o&99L)BJJ BJJ>?Q BJJ>?RZZBJJ>?RZZQ>?GbL(7S=87T>Iv~r.c|j\}}t||}tjgdg}t |||d}t |||y)Nr_rcrErr r!r r rr$r&r' interp_rgrGrs r,test_linear_4dzTestInterpN.test_linear_4dsQ--/+FF; 012A!)F"3V} }} t|tj| jd|ycc} wcc} }w)N>rLrMr^rlrrprrr1rqrFrrirrr.rrb) r\rxr!rr(r rrCrr r) r$rr&r*ryr'rGrfrhr+rgs r,rzz#TestInterpN.test_nonscalar_values_2s& ) ) KK (D ii##D) Q1a:/:;A FFF6 Nww1///// v||B/026;6<<;K5L01sAqy 16&%*,22 2::b>++%H 2s(DD $DDc"|j\}}tjjdtjj ddddd}tjj ddd}t t t|||dy) NrrrrrrrrE)rr!rrrrpr]r )r$r&r'rGs r, test_non_scalar_values_splinef2dz,TestInterpN.test_non_scalar_values_splinef2dsi--/ t1aA.2q)j'666( *r.cl|dk(rtjd|j\}}}||f}|d|zz }tjgdgdgj }t ||||}t ||j||}t ||j||} |d| zz} t|| y)NrNrtrvrrrErw) r\rxrr!r rr ryrzr) r$rrrr'r&rGrIv2rv2irJs r,r}zTestInterpN.test_complexs W  KKD E++- 1fQ"V)#::<==>Q VVVF ;ffkk6&Affkk6&A 2c6\Br.c |j\}}}||f}|d|zz }tjgdgdgj}t j t d5t|||ddddy#1swYyxYw)NrvrrryrYrNrE)rr!r rr\r r]r r$rrr'r&rGs r,test_complex_pchipzTestInterpN.test_complex_pchip/s++- 1fQ"V)#::<==>Q  ]]:V 4 < FFF7 ; < < Q  \\. ) @ FFF; ? @ @ @s A88Brrcrctjddd}tjddd}td}t||f|ddg| }t||f|jddg| }t ||d y) Nrr1rrirrrrrEFr)r!rrr rr)r$rrrr'rIrJs r,rz"TestInterpN.test_duck_typed_valuesEsq KK1a  KK1a  aVVc3Z ? aVVYYc 6 BBE2r.ctjddd}tjddd}ttjj dd}tjj ddd}t ||f|||}t ||ftj |||}|dk(rt||d d y t||y ) z8np.matrix inputs are allowed for backwards compatibilityrr1rrirrrErMg-C6 ?g>)rorN)r!rrrrr r r)r$rrrr'rGrIrJs r,test_matrix_inputzTestInterpN.test_matrix_inputSs KK1a  KK1a  q!,-1a( aVVVF ; aVRZZ/ G Y  BD 9 B #r.c:tjgdg}tjddgddgddgg}tdggdf||}gd}t||d tjd dgd dgd dgg}tdggdf||d d}t||d y)N)r@rirrig@g @gffffff@)r1rr)gQ?gffffff@gffffff @rmrnrr7gffffffFr)r!r r r)r$r'rr,rs r,rz TestInterpN.test_length_one_axisfs <.) XX3x!SAs84 5sI&3 V%0XXSzC:c{; <sI&#(T; V%0r.c Pd}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 | |y)Nc0d|dzzd|dzzz|z |z Sr8rr;s r, value_func_4dz9TestInterpN.test_descending_points..value_func_4d}s'qAv:AF *Q.2 2r.)rrir1r)rrrr)rr@rCr rTrr@r rrrr)r!r rrerr r)r$rx1x2x3x4r&r'rcorrect_result x1_descend x2_descend x3_descend x4_descendpoints_shuffledvalues_shuffled test_results r,test_descending_pointsz"TestInterpN.test_descending_points|s! 3XXl # XXo & XXo & XX& 'b"b! [[&4 =?Cbkk!R&;<{{1c1%' 5"X "X "X "X %z:zJ' [[/D FHoD  4r.cttjgd}tjgd}tjgdgdgdgdgdg}tjgdgdgj}d}tjt | 5t ||f||dddy#1swYyxYw) NrTrWrrrrrrXrY)r!r rr\r r]r )r$rrrrrZs r,rYz%TestInterpN.test_invalid_points_orders HH* + HH* + HHo%8 9 XX679::;! ; ]]:U 3 # QFAr " # # #s B..B7cdg}ddg}tjd}d}tt|5t |||dddy#1swYyxYw)N)rrirri)ririrzaThe requested sample points xi have dimension 3, but this RegularGridInterpolator has dimension 1rY)r!rrpr]r )r$r&r'rmsgs r,test_invalid_xi_dimensionsz&TestInterpN.test_invalid_xi_dimensionssRQ WWY 9 :S 1 ( FFB ' ( ( (s AActjddd}tjddd}tjddd}|||f}tjd}tjgd}|D]}d|j_d|j_d|j_t |||t|||y) NrrrrrrrrGz@(\@gffffff?Fr!rrr flags writeabler r )r$rrrr&r'pointds r,test_readonly_gridzTestInterpN.test_readonly_grids KK1a  KK1a  KK1a Q#+, &A %AGG  &!&  % &//6r.crtjddd}tjddd}||f}tjd}tjddg}|D]}d|j_d|j_d|j_t |||t|||y) NrrrrrrrrFr)r$rrr&r'rrs r,test_2d_readonly_gridz!TestInterpN.test_2d_readonly_grids KK1a  KK1a Q$& &A %AGG  &!&  % &//6r.ctjddd}tj|tj|fjj dddf}|j jrJtjddd}tjddd}|||f}tjd}tjgd}t|||t|||y)Nrrrrrrr) r!rr empty_likerrr c_contiguousrr r r )r$rrrr&r'rs r,test_non_c_contiguous_gridz&TestInterpN.test_non_c_contiguous_grids KK1a  IIq"--*+ , . . 3 3 5ad ;77'''' KK1a  KK1a Q#+,&//6r.rlz>f8zr,s +/*LLN)9"(!8!8 %22"*+E 7E 7,E 7P66(f7f7r.