tKg4<ddlZddlZddlZddlmZmZmZmZddl m Z ddl Z ddl m Z ddlmcmZddlmcmZddlZddlZej.dkZdZGddZGd d ZGd d Zy) N) assert_equalassert_allcloseassert_almost_equalsuppress_warnings)raises)check_free_memorylctjjtjjtjj t d|S)Ndata)ospathjoinabspathdirname__file__)basenames i/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/scipy/interpolate/tests/test_interpnd.py data_filers8 77<<(AB **ceZdZdZdZdZdZdZdZdZ dZ d Z d Z e jje jj!ed d Zy)TestLinearNDInterpolationctjgdtj}tj|jdtj}t j |||}t||y)Nrr)r)r?)rr)?333333?dtypernparrayfloat64arangeshapeinterpndLinearNDInterpolatorrselfxyyis rtest_smoketestz(TestLinearNDInterpolation.test_smoketestsY HHN:: ' IIaggaj 3 0X * *1a 0 3Ar"rcDtjgdtj}tj|jdtj}t j |dddf|dddff||dddf|dddf}t||y)Nrrrr r(s rtest_smoketest_alternatez2TestLinearNDInterpolation.test_smoketest_alternate#s HHN:: ' IIaggaj 3 ?X * *AacFAacF+;Q ?!A#!A# OAr"rctjgdtj}tj|jdtj}|d|zz }t j |||}t||yNrrr@r r(s rtest_complex_smoketestz0TestLinearNDInterpolation.test_complex_smoketest,sf HHN:: ' IIaggaj 3 1H 0X * *1a 0 3Ar"rc@tjgdtj}tj|jdtj}|d|zz }t j |}tj|||}t||yr2) r!r"r#r$r%qhullDelaunayr&r'rr)r*r+trir,s rtest_tri_inputz(TestLinearNDInterpolation.test_tri_input6st HHN:: ' IIaggaj 3 1HnnQ 2X * *3 21 5Ar"rcJtjgdtj}tjgdtjfd}tjtjddddddftjddddddf\}}|j }|j }tj||gj j}tj||}t||||y)N)rrr/r/r/r/rrg?g@gg@c||zdk}|}||}||}||}||}d|z} dd|z |z z d|zz d|zz||< d||zdz z dd|z zz dd|z zz||<|S)Nr/r) r*r+t1t2x1y1x2y2zvaluess ripz1TestLinearNDInterpolation.test_square..ipOsa%1*BB2B2B2B2B!AAYB ,ay|$ay|$AbEAYR! ,ay!b&)*ay!b&)*AbEHrrr/ r!r"r#broadcast_arrayslinspaceravelTcopyr&r'r)r)pointsrLxxyyxizirKs @r test_squarez%TestLinearNDInterpolation.test_squareAs6bjjI+2::> *$$R[[Ar%:1T6%B%'[[Ar%:46%BDB XXZ XXZ XXr2h  ! ! & & ( :X * *66 :2 >B2r +rc tjgdtj}tj|jdtj}t j ||d|}t||y)Nr)r\)r\)r]r])g@rArrTrescaler r(s rtest_smoketest_rescalez0TestLinearNDInterpolation.test_smoketest_rescalens[ HHB:: ' IIaggaj 3 >X * *1a >q AAr"rcltjgdtj}tjgdtj}tjtjddddddftjddddddf\}}|j }|j }tj||gj j}tj|||}tj||d |}t||y) N)r)rd) rb)rcrrr?rrcrMrbTr^rN)r)rTrKrUrVrWrX zi_rescaleds rtest_square_rescalez-TestLinearNDInterpolation.test_square_rescalews$$R[[B%;AdF%C%'[[C%h33FF "  B ,rctjgdtj}tj|jdtj}|d|zz }t j |}tj|j||}tj|j|d|}t||yNr[rrr3Tr^) r!r"r#r$r%r6r7r&r'rTrr)r*r+r9r, yi_rescales rtest_tripoints_input_rescalez6TestLinearNDInterpolation.test_tripoints_input_rescales HH?:: ' IIaggaj 3 1HnnQ 9X * *3::q 9! <X223::q! B +rctjgdtj}tj|jdtj}|d|zz }t j |}d}tjt|5tj||d|dddy#1swYyxYw Nr[rrr3zORescaling is not supported when passing a Delaunay triangulation as ``points``.)matchTr^) r!r"r#r$r%r6r7pytestr ValueErrorr&r'r)r*r+r9rms rtest_tri_input_rescalez0TestLinearNDInterpolation.test_tri_input_rescales HH?:: ' IIaggaj 3 1HnnQ9 ]]:U 3 ?H ) )#q$ ? B4 3 3 B<<Cctjjdtjjdd}tjjddtjjdzz}t j ||}t jt j|}t|dd|ddyNrBy?r) r!randomseedrandr&r'pickleloadsdumpsrr)r*r+rLip2s r test_picklez%TestLinearNDInterpolation.test_pickles t IINN2q ! IINN2 BIINN2$6!6 6  * *1a 0ll6<<+,BsCL#c3-8rzit fails on 32-bit)reasonc tdtjddd}tjddd}tj||\}}d}tj|j |j fj }||zj }tj||}tj||tddtddf} tj||tddtddf} tj||tddtddf} tj||tddtddf} | j| j| j| j| j| j| j| jy)Ni'rihrccJtj||f\}}|||f}|SN)r!mgrid) interpolator slice_rows slice_colsgrid_xgrid_yress r do_interpz;TestLinearNDInterpolation.test_threading..do_interps-XXj*&<=NFF/0CJri4)targetargs)rr!r$meshgridvstackrQrRr&r' threadingThreadslicestartr ) r)r_ticks phi_ticksr_gridphi_gridrrTrKrworker_thread_1worker_thread_2worker_thread_3worker_thread_4s rtest_threadingz(TestLinearNDInterpolation.test_threadings % ))AtR(IIar* ;;w :  FLLNHNN,<=>@@8#**,44VVD #**aa?A$**dD 15D>BD$**adD0ABD$**dD 15t3DEG rN)__name__ __module__ __qualname__r-r0r4r:rYr`rerjrqrrnmarkslowskipif _IS_32BITrrCrrrrsr### #+,Z#-$ , C 9 [[ [[ *>?'@'rrceZdZdZdZy)TestEstimateGradients2DGlobalc tjgdt}tj|}ddfddfddfd d fg}t |D]\}\}}||ddd f|ddd f}t j||d }t|jdt|tj|dddfd |zzddd|zy)N)r)rrBr>)r/rBrg?g333333?g?rcd|zdzSNrr/rCr*r+s rz>TestEstimateGradients2DGlobal.test_smoketest..s !A#'rrc d|zSNrrCrs rrz>TestEstimateGradients2DGlobal.test_smoketest..s!a%rr>c d|zSNrCrs rrz>TestEstimateGradients2DGlobal.test_smoketest..s"q&rr<cdd|zzd|zzSNrAL,@rCrs rrz>TestEstimateGradients2DGlobal.test_smoketest..s!ac'E!G+r)rArrr/ư>)tol)rBgh㈵>zitem %d)rtolatolerr_msg) r!r"floatr6r7 enumerater&estimate_gradients_2d_globalrr%r) r)r*r9funcsjfuncgradrJdzs rr-z,TestEstimateGradients2DGlobal.test_smoketests HH@GL NnnQ "6 *  ( & ) +Z 8   )/OA|dQqsVQqsV$A66sA4HB 6 * BtAv 62 =!%D)a- I 0rcptjtd}tjj |j d}t j|}t5}|jtjdtj||ddddy#1swYyxYw)Nzestimate_gradients_hang.npyrz$Gradient estimation did not converger/)maxiter) r!loadrrwryr%r6r7rfilterr&GradientEstimationWarningr)r)rTrKr9sups rtest_regression_2359z2TestEstimateGradients2DGlobal.test_regression_2359s#@AB Q0nnV$ C JJx99= ?  1 1#vq I! s *9B,,B5N)rrrr-rrCrrrrsI, JrrceZdZ d dZdZdZdZdZdZe jjdd Z d Z d Zd Zy)TestCloughTocher2DInterpolatorNc tjjd|tjgdt}|s/t j |||dddf|dddfd|}n>t j |dddf|dddff||dddf|dddfd|}tjjdd }|s ||} n||dddf|dddf} ||dddf|dddf} t| | fi|y#t$r0td t| | z td |jwxYw) Nru)rr<r>r=rr)rg?rrr/r)rr_2rBz_check_accuracy: abs(a-b):zip.grad:) r!rwrxr"rr&CloughTocher2DInterpolatorryrAssertionErrorprintabsr) r)rr*r alternater_kwrLpabs r_check_accuracyz.TestCloughTocher2DInterpolator._check_accuracysO t 9& %&A 44QQqsVQqsV8L9=wPB44a!fa!f5E59!AaC&!AaC&5I9=wPB IINN2q !1A1QqS61QqS6"A 1Q31Q3   Aq 'B '  .AE ; *bgg &  s  D9Ec ddddg}t|D]m\}}|j|dddd|z|j|dddd d |z |j|dddd |zd |j|dddd d d|zoy)Ncd|zdzSrrCrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..$s 1qrc d|zSrrCrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..%sQrc d|zSrrCrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..&sarcdd|zzd|zzSrrCrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..'sQqS57*rgvIh%<=gHz> Function %drrrrTzFunction (alternate) %d)rrrrrzFunction (rescaled) %drrrrr_z!Function (alternate, rescaled) %d)rrrrr_rrrr)rrrs rtest_linear_smoketestz4TestCloughTocher2DInterpolator.test_linear_smoketest!s !   *  !'GAt  5t$)6): ! <  5t$+/)BQ)F ! H  5t$)AA)Et ! U  5t$+/)Lq)P ! R(rc ddddg}t|D]8\}}|j|dddd|z |j|dddd|zd :y) Nc |dzSNrBrCrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..:Arc |dzSrrCrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..;rrc|dz|dzz SrrCrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..<A1rc ||zSrrCrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..=1r& .>g)\(?rrrTrrrs rtest_quadratic_smoketestz7TestCloughTocher2DInterpolator.test_quadratic_smoketest7su   $   !'GAt  4d)6): ! <  4d)6):D ! J(rc@tjgdtj}tj|jdtj}|d|zz }t j |}tj|||}t||yr2) r!r"r#r$r%r6r7r&rrr8s rr:z-TestCloughTocher2DInterpolator.test_tri_inputFst HHN:: ' IIaggaj 3 1HnnQ 8X 0 0a 8 ;Ar"rctjgdtj}tj|jdtj}|d|zz }t j |}d}tjt|5tj||d|dddy#1swYyxYwrl) r!r"r#r$r%r6r7rnrror&rrps rrqz5TestCloughTocher2DInterpolator.test_tri_input_rescaleQs HH?:: ' IIaggaj 3 1HnnQ9 ]]:U 3 EH / /Q Ea H4 3 3rrctjgdtj}tj|jdtj}|d|zz }t j |}tj|j||}tj|j|d|}t||yrg) r!r"r#r$r%r6r7r&rrTrrhs rrjz;TestCloughTocher2DInterpolator.test_tripoints_input_rescale^s HH?:: ' IIaggaj 3 1HnnQ ?X 0 0Q ? BUX88QPTUVWX B +rrBc dddddg}tjjdtjtjgdt tjj d d f}t|D]:\}}|j||d d d d|z|j||d d d d|zd<y)Nc |dzSrrCrs rrz;TestCloughTocher2DInterpolator.test_dense..nrrc |dzSrrCrs rrz;TestCloughTocher2DInterpolator.test_dense..orrc|dz|dzz SrrCrs rrz;TestCloughTocher2DInterpolator.test_dense..prrc ||zSrrCrs rrz;TestCloughTocher2DInterpolator.test_dense..qrrctjdtjz|ztjdtjz|zzSr)r!cospisinrs rrz;TestCloughTocher2DInterpolator.test_dense..rs3"%% *266!BEE'!)+<r=rirBrg{Gzt?g{Gz?r)r*rrrrT)r*rrrrr_) r!rwrxr_r"rryrr)r)rgridrrs r test_densez)TestCloughTocher2DInterpolator.test_densejs   $  <   tuuRXX:%HYY^^E1-./!'GAt  4d)6): ! <  4d)6):D ! J(rctjjdd}tjjd}ttt j ||y)NrvrA)r!rwrandn assert_raisesror&r)r)r*r+s rtest_wrong_ndimz.TestCloughTocher2DInterpolator.test_wrong_ndims< IIOOB " IIOOB j("E"Eq!Lrctjjdtjjdd}tjjddtjjdzz}t j ||}t jt j|}t|dd|ddyrt) r!rwrxryr&rrzr{r|rr}s rrz*TestCloughTocher2DInterpolator.test_pickles t IINN2q ! IINN2 BIINN2$6!6 6  0 0A 6ll6<<+,BsCL#c3-8rcxtjdddtjddz fg}tjgd}tj||}d|j d<d }tjdtj |zdtj|zg}tjdtj tjdz |z zdtjtjdz |z zg}||}||}t||tjjd tjjdd} tjjd} | j|jj| dddfz}| j|| z}| j|| z}tj||}d|j d<||} ||} t| |t| |y) Nrr>rrArB)r/rrr.rr/)r!r"sqrtr&rrrrrrrwrxrdotrR) r)rTrKrLalphap1p2v1v2Arw1w2s rtest_boundary_tri_symmetryz9TestCloughTocher2DInterpolator.test_boundary_tri_symmetrys 66CA+>?@)$  0 0 @  XXsRVVE]*C"&&-,?@ A XXsRVVBEE!GeO44cBFF2557U?rsl ..* 3--$$  [[5 * ~~B"J"JJu u r