bi<ddlZddlZddlZddlZddlmZddlZddlm Z m Z ddl m Z ddl mcmZddlmcmZddlZddlZej.dkZdZGddZGd d ZGd d Zy) N)raises)xp_assert_closeassert_almost_equal)check_free_memorylctjjtjjtjj t d|S)Ndata)ospathjoinabspathdirname__file__)basenames `/home/cdr/jupyterlab/.venv/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)Nrrrrr&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@rr&s rtest_complex_smoketestz0TestLinearNDInterpolation.test_complex_smoketest,sf HHN:: ' IIaggaj 3 1H 0X * *1a 0 3Ar"rcdtjgdtj}tj|jdtj}|d|zz }t j |}tj||}||}t|||j|usJyr0) rr r!r"r#qhullDelaunayr$r%rtri)r'r(r)r6 interpolatorr*s rtest_tri_inputz(TestLinearNDInterpolation.test_tri_input6s HHN:: ' IIaggaj 3 1HnnQ44S!< !_Ar"3&&&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..ipQsa%1*BB2B2B2B2B!AAYB ,ay|$ay|$AbEAYR! ,ay!b&)*ay!b&)*AbEHrrr- rr r!broadcast_arrayslinspaceravelTcopyr$r%r)r'pointsrJxxyyxizirIs @r test_squarez%TestLinearNDInterpolation.test_squareCs6bjjI+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)rZ)rZ)r[r[)g@r?rrTrescalerr&s rtest_smoketest_rescalez0TestLinearNDInterpolation.test_smoketest_rescaleps[ HHB:: ' IIaggaj 3 >X * *1a >q AAr"rcltjgdtj}tjgdtj}tjtjddddddftjddddddf\}}|j }|j }tj||gj j}tj|||}tj||d |}t||y) N)r)rd) r`)rarrr=rrarKr`Tr\rL)r'rRrIrSrTrUrV zi_rescaleds rtest_square_rescalez-TestLinearNDInterpolation.test_square_rescaleys$$R[[B%;AdF%C%'[[C%h33FF "  B ,rctjgdtj}tj|jdtj}|d|zz }t j |}tj|j||}tj|j|d|}t||yNrYrrr1Tr\) rr r!r"r#r4r5r$r%rRrr'r(r)r6r* 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 NrYrrr1zORescaling is not supported when passing a Delaunay triangulation as ``points``.)matchTr\) rr r!r"r#r4r5pytestr ValueErrorr$r%r'r(r)r6rks rtest_tri_input_rescalez0TestLinearNDInterpolation.test_tri_input_rescales HH?:: ' IIaggaj 3 1HnnQ9 ]]:U 3 C ?H ) )#q$ ? B C C C B<<Cctjjdtjjdd}tjjddtjjdzz}t j ||}t jt j|}t|dd|ddyNr@y?r) rrandomseedrandr$r%pickleloadsdumpsr)r'r(r)rJip2s 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'rihracJtj||f\}}|||f}|SN)rmgrid)r7 slice_rows slice_colsgrid_xgrid_yress r do_interpz;TestLinearNDInterpolation.test_threading..do_interps-XXj*&<=NFF/0CJri4)targetargs)rrr"meshgridvstackrOrPr$r% threadingThreadslicestartr ) r'r_ticks phi_ticksr_gridphi_gridrrRrIr7worker_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+r.r2r8rWr^rcrhror|rlmarkslowskipif _IS_32BITrrArrrrsr### '+,Z#-$ , C 9 [[ [[ *>?'@'rrceZdZdZdZy)TestEstimateGradients2DGlobalc tjgdt}tj|}ddfddfddfd d fg}t |D]{\}\}}||ddd f|ddd f}t j||d }|jdk(sJt|tj|dddfd |zzddd|}y)N)r)rr@r<)r-r@rg?g333333?g?rcd|zdzSNrr-rAr(r)s rz>TestEstimateGradients2DGlobal.test_smoketest..s!A#'rrc d|zSNrrArs rrz>TestEstimateGradients2DGlobal.test_smoketest..s !a%rr<c d|zSNrArs rrz>TestEstimateGradients2DGlobal.test_smoketest..s "q&rr:cdd|zzd|zzSNr?L,@rArs rrz>TestEstimateGradients2DGlobal.test_smoketest..s!ac'E!G+r)r?rrr-ư>)tol)r@gh㈵>zitem )rtolatolerr_msg) rr floatr4r5 enumerater$estimate_gradients_2d_globalr#r) r'r(r6funcsjfuncgradrHdzs rr+z,TestEstimateGradients2DGlobal.test_smoketests HH@GL NnnQ "6 *  ( & ) +Z 8   )/ OA|dQqsVQqsV$A66sA4HB88v% %% BHHTN47+ad2Ds    rctjtd}tjj |j d}t j|}tj5tjddtjtj||ddddy#1swYyxYw)Nzestimate_gradients_hang.npyrignorez$Gradient estimation did not converger-)maxiter)rloadrrurwr#r4r5warningscatch_warningsfilterwarningsr$GradientEstimationWarningr)r'rRrIr6s rtest_regression_2359z2TestEstimateGradients2DGlobal.test_regression_2359s#@AB Q0nnV$ $ $ & J  # #622   1 1#vq I  J J Js 4>B;;CN)rrrr+rrArrrrs 0JrrceZdZ d dZdZdZdZdZdZe jjdd Z d Z d Zd Zy)TestCloughTocher2DInterpolatorNc tjjd}|tjgdt}|s/t j |||dddf|dddfd|}n>t j |dddf|dddff||dddf|dddfd|}|jdd } |s || } n|| dddf| dddf} || dddf| dddf} t| | fi|y#t$r0td t| | z td |jwxYw) Nrs)rr:r<r;rr)rg?rrr-r)rr]2r@z_check_accuracy: abs(a-b):zip.grad:) rru RandomStater rr$CloughTocher2DInterpolatorrwrAssertionErrorprintabsr) r'rr(r alternater]kwrngrJpabs r_check_accuracyz.TestCloughTocher2DInterpolator._check_accuracysNii##D) 9& %&A 44QQqsVQqsV8L9=wPB44a!fa!f5E59!AaC&!AaC&5I9=wPB HHRO1A1QqS61QqS6"A 1Q31Q3   Aq 'B '  .AE ; *bgg &  s : D9Ec ddddg}t|D]m\}}|j|dddd||j|dddd d | |j|dddd |d |j|dddd d d|oy)Ncd|zdzSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..,s1qrc d|zSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..-s Qrc d|zSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest...s arcdd|zzd|zzSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest../sQqS57*rgvIh%<=gHz> Function rrrrTzFunction (alternate) )rrrrrzFunction (rescaled) rrrrr]zFunction (alternate, rescaled) )rrrrr]rrrr'rrrs rtest_linear_smoketestz4TestCloughTocher2DInterpolator.test_linear_smoketest)s !   *  !' GAt  %d1# !   %d/s3 !   %d.qc2D !   %dt9!= !  rc ddddg}t|D]8\}}|j|dddd| |j|dddd|d :y) Nc |dzSNr@rArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..G Arc |dzSrrArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..Hrrc|dz|dzz SrrArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..IA1rc ||zSrrArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..J 1r& .>g)\(?rrrTrrrs rtest_quadratic_smoketestz7TestCloughTocher2DInterpolator.test_quadratic_smoketestDs~   $   !' GAt  $TYqc? !   $TYqc?TX !   rc@tjgdtj}tj|jdtj}|d|zz }t j |}tj|||}t||yr0) rr r!r"r#r4r5r$rr)r'r(r)r6r*s rr8z-TestCloughTocher2DInterpolator.test_tri_inputUst HHN:: ' IIaggaj 3 1HnnQ 8X 0 0a 8 ;Ar"rctjgdtj}tj|jdtj}|d|zz }t j |}d}tjt|5tj||d|dddy#1swYyxYwrj) rr r!r"r#r4r5rlrrmr$rrns rroz5TestCloughTocher2DInterpolator.test_tri_input_rescale`s HH?:: ' IIaggaj 3 1HnnQ9 ]]:U 3 I EH / /Q Ea H I I Irpctjgdtj}tj|jdtj}|d|zz }t j |}tj|j||}tj|j|d|}t||yre) rr r!r"r#r4r5r$rrRrrfs rrhz;TestCloughTocher2DInterpolator.test_tripoints_input_rescalems HH?:: ' IIaggaj 3 1HnnQ ?X 0 0Q ? BUX88QPTUVWX B +rr[c jdddddg}tjjd}tjtjgdt |j d d f}t|D]:\}}|j||d d d d||j||d d d d|d<y)Nc |dzSrrArs rrz;TestCloughTocher2DInterpolator.test_dense..}rrc |dzSrrArs rrz;TestCloughTocher2DInterpolator.test_dense..~rrc|dz|dzz SrrArs rrz;TestCloughTocher2DInterpolator.test_dense..rrc ||zSrrArs rrz;TestCloughTocher2DInterpolator.test_dense..rrctjdtjz|ztjdtjz|zzSr)rcospisinrs rrz;TestCloughTocher2DInterpolator.test_dense..s5"%% *266!BEE'!)+<<ri)rr:r<r;rir@rg{Gzt?g{Gz?r)r(rrrrT)r(rrrrr]) rrurr_r rrwrr)r'rrgridrrs r test_densez)TestCloughTocher2DInterpolator.test_denseys   $  <  ii##D)uuRXX:%HXXeQ'()!' GAt  $T RSQTo !   $T#A3 !   rctjjdd}tjjd}ttt j ||y)Nrtr?)rrurandn assert_raisesrmr$r)r'r(r)s rtest_wrong_ndimz.TestCloughTocher2DInterpolator.test_wrong_ndims< IIOOB " IIOOB j("E"Eq!Lrcbtjjd}|jdd}|jdd|jdzz}t j ||}t jt j|}t|dd|ddyrr) rrurrwr$rrxryrzr)r'rr(r)rJr{s rr|z*TestCloughTocher2DInterpolator.test_picklesii##D) HHRO HHRL2chhrl? *  0 0A 6ll6<<+,BsCL#c3-8rc@tjdddtjddz fg}tjgd}tj||}d|j d<d }tjdtj |zdtj|zg}tjdtj tjdz |z zdtjtjdz |z zg}||}||}t||tjjd } | jdd} | jd} | j|jj| dddfz}| j|| z}| j|| z}tj||}d|j d<||} ||} t| |t| |y) Nrr<rr?r@)r-rrr.rr-)rr sqrtr$rrrrrrrurrdotrP)r'rRrIrJalphap1p2v1v2rArw1w2s rtest_boundary_tri_symmetryz9TestCloughTocher2DInterpolator.test_boundary_tri_symmetrys 66CA+>?@)$  0 0 @  XXsRVVE]*C"&&-,?@ A XXsRVVBEE!GeO44cBFF2557U?rsh * F3..$$  [[5 * @@F'J'JT@ @ r