qwj<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 j/var/www/html/newmanjeet/manjet/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 ?H ) )#q$ ? B4 3 3 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   0rctjtd}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$ $ $ &  # #622   1 1#vq I ' & &s 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..,s 1qrc d|zSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..-sQrc d|zSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest...sarcdd|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..GArc |dzSrrArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..Hrrc|dz|dzz SrrArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..IA1rc ||zSrrArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..J1r& .>g)\(?rrrTrrrs rtest_quadratic_smoketestz7TestCloughTocher2DInterpolator.test_quadratic_smoketestDsy   $   !'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 EH / /Q Ea H4 3 3rpctjgdtj}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..s3"%% *266!BEE'!)+<?@)$  0 0 @  XXsRVVE]*C"&&-,?@ A XXsRVVBEE!GeO44cBFF2557U?rsh * F3..$$  [[5 * @@F'J'JT@ @ r