o ?Hh<@sddlZddlZddlZddlmZddlmZddlZddl m Z m Z ddl m Z ddlmmZddlmmZddlZddlZejdkZddZGd d d ZGd d d ZGd ddZdS)N)suppress_warnings)raises)xp_assert_closeassert_almost_equal)check_free_memorylcCs tjtjtjtd|S)Ndata)ospathjoinabspathdirname__file__)basenamere/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/scipy/interpolate/tests/test_interpnd.py data_filesrc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ e jje jje jjeddddZdS)TestLinearNDInterpolationcCsHtjgdtjd}tj|jdtjd}t|||}t||dS)Nrr)r)r?)rr)?333333?dtypernparrayfloat64arangeshapeinterpndLinearNDInterpolatorrselfxyyirrrtest_smoketests z(TestLinearNDInterpolation.test_smoketestcCs~tjgdtjd}tj|jdtjd}t|dddf|dddff||dddf|dddf}t||dS)Nrrrrr#rrrtest_smoketest_alternate#s Fz2TestLinearNDInterpolation.test_smoketest_alternatecCsTtjgdtjd}tj|jdtjd}|d|}t|||}t||dSNrrr@rr#rrrtest_complex_smoketest,s  z0TestLinearNDInterpolation.test_complex_smoketestcCsptjgdtjd}tj|jdtjd}|d|}t|}t||}||}t |||j |us6JdSr+) rrrrr qhullDelaunayr!r"rtri)r$r%r&r0 interpolatorr'rrrtest_tri_input6s     z(TestLinearNDInterpolation.test_tri_inputcstjgdtjd}tjgdtjdfdd}ttddddddftddddddf\}}|}|}t||gj}t ||}t ||||dS) N)rrr)r)r)r)rrg?g@gg@c s||dk}|}||}||}||}||}d|}dd||d|d|||<d||ddd|dd|||<|S)Nr)rr) r%r&t1t2x1y1x2y2zvaluesrripQs$   z1TestLinearNDInterpolation.test_square..iprr) rrrbroadcast_arrayslinspaceravelTcopyr!r"r)r$pointsrBxxyyxizirr@r test_squareCs z%TestLinearNDInterpolation.test_squarecCsLtjgdtjd}tj|jdtjd}tj||dd|}t||dS)Nr)rQ)rQ)rRrR)g@r7rrTrescalerr#rrrtest_smoketest_rescaleps z0TestLinearNDInterpolation.test_smoketest_rescalecCstjgdtjd}tjgdtjd}ttddddddftddddddf\}}|}|}t||gj}t |||}tj ||dd |}t ||dS) N)r)rd) rV)rWrrr6rrWrCrVTrSrD)r$rJrArKrLrMrN zi_rescaledrrrtest_square_rescaleysz-TestLinearNDInterpolation.test_square_rescalecCvtjgdtjd}tj|jdtjd}|d|}t|}t|j ||}tj|j |dd|}t ||dSNrPrrr,TrS) rrrrr r.r/r!r"rJrr$r%r&r0r' yi_rescalerrrtest_tripoints_input_rescales    z6TestLinearNDInterpolation.test_tripoints_input_rescalecCtjgdtjd}tj|jdtjd}|d|}t|}d}tjt |dt j ||dd|WddS1s?wYdS NrPrrr,zORescaling is not supported when passing a Delaunay triangulation as ``points``.)matchTrS) rrrrr r.r/pytestr ValueErrorr!r"r$r%r&r0rarrrtest_tri_input_rescale   "z0TestLinearNDInterpolation.test_tri_input_rescalecCsltjdtjdd}tjddtjd}t||}tt|}t |dd|dddSNr8y?r) rrandomseedrandr!r"pickleloadsdumpsr)r$r%r&rBip2rrr test_pickles  z%TestLinearNDInterpolation.test_picklezit fails on 32-bit)reasonc Cs0tdtddd}tddd}t||\}}dd}t||fj}||}t||}t j ||t ddt ddfd} t j ||t ddt ddfd} t j ||t ddt ddfd} t j ||t ddt ddfd} | | | | | | | | dS) Ni'rihrWcSs"tj||f\}}|||f}|SN)rmgrid)r1 slice_rows slice_colsgrid_xgrid_yresrrr do_interps z;TestLinearNDInterpolation.test_threading..do_interpi4)targetargs)rrrmeshgridvstackrGrHr!r" threadingThreadslicestartr ) r$r_ticks phi_ticksr_gridphi_gridrzrJrAr1worker_thread_1worker_thread_2worker_thread_3worker_thread_4rrrtest_threadings@   z(TestLinearNDInterpolation.test_threadingN)__name__ __module__ __qualname__r(r*r-r2rOrUrYr^rerqrbmarkslow thread_unsafeskipif _IS_32BITrrrrrrs    -    rc@seZdZddZddZdS)TestEstimateGradients2DGlobalc Cstjgdtd}t|}dddfdddfddd fd dd fg}t|D]>\}\}}||ddd f|ddd f}tj||dd}|jdksNJt |t|dddfd |ddd|dq(dS)N)r)rr8r5)r)r8rg?g333333?g?rcS d|dSNrr)rr%r&rrr z>TestEstimateGradients2DGlobal.test_smoketest..rcSd|SNrrrrrrrr5cSd|SNrrrrrrrr3cSdd|d|SNr7L,@rrrrrr)r7rrr)ư>)tol)r8gh㈵>zitem %d)rtolatolerr_msg) rrfloatr.r/ enumerater!estimate_gradients_2d_globalr r) r$r%r0funcsjfuncgradr?dzrrrr(s"      "  z,TestEstimateGradients2DGlobal.test_smoketestcCstttd}tj|jd}t|}t}| t j dt j ||ddWddS1s3wYdS)Nzestimate_gradients_hang.npyrz$Gradient estimation did not converger))maxiter) rloadrrjrlr r.r/rfilterr!GradientEstimationWarningr)r$rJrAr0suprrrtest_regression_2359s "z2TestEstimateGradients2DGlobal.test_regression_2359N)rrrr(rrrrrrs rc@sneZdZ  dddZddZdd Zd d Zd d ZddZe j dddZ ddZ ddZddZdS)TestCloughTocher2DInterpolatorNrFc KsNtjd}|durtjgdtd}|s.tj|||dddf|dddfd|d}n&tj|dddf|dddff||dddf|dddfd|d}|dd } |sa|| } n|| dddf| dddf} || dddf| dddf} z t| | fi|WdSt yt d t | | t d |j w) Nrh)rr3r5r4rr)rg?rrr)r)rrT2r8z_check_accuracy: abs(a-b):zip.grad:) rrj RandomStaterrr!CloughTocher2DInterpolatorrlrAssertionErrorprintabsr) r$rr%r alternaterTkwrngrBpabrrr_check_accuracys2  &"   ""  z.TestCloughTocher2DInterpolator._check_accuracyc Csddddddddg}t|D]8\}}|j|dddd|d |j|dddd d |d |j|dddd |d d|j|dddd d d|dqdS)NcSrrrrrrrr(rzFTestCloughTocher2DInterpolator.test_linear_smoketest..cSrrrrrrrr)rcSrrrrrrrr*rcSrrrrrrrr+rgvIh%<=gHz> Function %drrrrTzFunction (alternate) %d)rrrrrzFunction (rescaled) %drrrrrTz!Function (alternate, rescaled) %d)rrrrrTrrrr$rrrrrrtest_linear_smoketest%s*    z4TestCloughTocher2DInterpolator.test_linear_smoketestc Csdddddddddg}t|D]\}}|j|dddd |d |j|dddd |d d qdS) NcS|dSNr8rrrrrr>rzITestCloughTocher2DInterpolator.test_quadratic_smoketest..cS|dSrrrrrrr?rcS|d|dSrrrrrrr@cS||SrsrrrrrrAr& .>g)\(?rrrTrrrrrrtest_quadratic_smoketest;s  z7TestCloughTocher2DInterpolator.test_quadratic_smoketestcCs^tjgdtjd}tj|jdtjd}|d|}t|}t|||}t ||dSr+) rrrrr r.r/r!rr)r$r%r&r0r'rrrr2Js   z-TestCloughTocher2DInterpolator.test_tri_inputcCr_r`) rrrrr r.r/rbrrcr!rrdrrrreUrfz5TestCloughTocher2DInterpolator.test_tri_input_rescalecCrZr[) rrrrr r.r/r!rrJrr\rrrr^bs   z;TestCloughTocher2DInterpolator.test_tripoints_input_rescalerRc Csddddddddddg}tjd}tjtjgdtd |d d f}t|D]\}}|j||d d dd|d|j||d d dd|ddq-dS)NcSrrrrrrrrrrz;TestCloughTocher2DInterpolator.test_dense..cSrrrrrrrrsrcSrrrrrrrrtrcSrrsrrrrrrurcSs(tdtj|tdtj|Sr)rcospisinrrrrrvs(i)rr3r5r4rir8rg{Gzt?g{Gz?r)r%rrrrT)r%rrrrrT) rrjrr_rrrlrr)r$rrgridrrrrr test_densens$  z)TestCloughTocher2DInterpolator.test_densecCs.tjdd}tjd}tttj||dS)Nrir7)rrjrandn assert_raisesrcr!r)r$r%r&rrrtest_wrong_ndims z.TestCloughTocher2DInterpolator.test_wrong_ndimcCsftjd}|dd}|dd|d}t||}tt|}t |dd|dddSrg) rrjrrlr!rrmrnror)r$rr%r&rBrprrrrqs   z*TestCloughTocher2DInterpolator.test_picklecCsRtdddtddfg}tgd}t||}d|jd<d }tdt|dt|g}tdttjd|dttjd|g}||}||}t ||tj d } | dd} | d} | |jj| dddf}| || }| || }t||}d|jd<||} ||} t | |t | |dS) Nrr5rr7r8)r)rrr.rr))rrsqrtr!rrrrrrrjrrdotrH)r$rJrArBalphap1p2v1v2rArw1w2rrrtest_boundary_tri_symmetrys,  "6       z9TestCloughTocher2DInterpolator.test_boundary_tri_symmetry)NrFF)rrrrrrr2rer^rbr fail_slowrrrqrrrrrrs !   r)rsysnumpyr numpy.testingrrbrrscipy._lib._array_apirrscipy._lib._testutilsrscipy.interpolate._interpnd interpolate _interpndr!scipy.spatial._qhullspatial_qhullr.rmrmaxsizerrrrrrrrrs$    E%