o i<@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__)basenamer_/home/kim/smarthome/.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?Zdtypernparrayfloat64arangeshapeinterpndLinearNDInterpolatorrselfxyyirrrtest_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*) rrrrrqhullDelaunayr r!rtri)r#r$r%r/ 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%t1t2x1y1Zx2y2zvaluesrripQs$   z1TestLinearNDInterpolation.test_square..iprr( rrrZbroadcast_arraysZlinspaceravelTcopyr r!r)r#pointsr@xxyyxizirr>r test_squareCs z%TestLinearNDInterpolation.test_squarecCsLtjgdtjd}tj|jdtjd}tj||dd|}t||dS)Nr)rM)rM)rNrN)g@r6rrTrescalerr"rrrtest_smoketest_rescaleps z0TestLinearNDInterpolation.test_smoketest_rescalecCstjgdtjd}tjgdtjd}ttddddddftddddddf\}}|}|}t||gj}t |||}tj ||dd |}t ||dS) N)r)rd) rR)rSrrr5rrSrArRTrOrB)r#rFr?rGrHrIrJZ zi_rescaledrrrtest_square_rescaleysz-TestLinearNDInterpolation.test_square_rescalecCvtjgdtjd}tj|jdtjd}|d|}t|}t|j ||}tj|j |dd|}t ||dSNrLrrr+TrO) rrrrrr-r.r r!rFrr#r$r%r/r&Z yi_rescalerrrtest_tripoints_input_rescales    z6TestLinearNDInterpolation.test_tripoints_input_rescalecCtjgdtjd}tj|jdtjd}|d|}t|}d}tjt |dt j ||dd|WddS1s?wYdS NrLrrr+zORescaling is not supported when passing a Delaunay triangulation as ``points``.)matchTrO) rrrrrr-r.pytestr ValueErrorr r!r#r$r%r/r[rrrtest_tri_input_rescale   "z0TestLinearNDInterpolation.test_tri_input_rescalecCsltjdtjdd}tjddtjd}t||}tt|}t |dd|dddSNr7y?r) rrandomseedrandr r!pickleloadsdumpsr)r#r$r%r@ip2rrr 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'rihrScSs"tj||f\}}|||f}|SN)rZmgrid)r0Z slice_rowsZ slice_colsZgrid_xZgrid_yresrrr do_interps z;TestLinearNDInterpolation.test_threading..do_interpi4)targetargs)rrrZmeshgridZvstackrCrDr r! threadingThreadslicestartr ) r#Zr_ticksZ phi_ticksZr_gridZphi_gridrorFr?r0Zworker_thread_1Zworker_thread_2Zworker_thread_3Zworker_thread_4rrrtest_threadings@   z(TestLinearNDInterpolation.test_threadingN)__name__ __module__ __qualname__r'r)r,r1rKrQrTrXr_rkr\markZslowZ thread_unsafeZskipif _IS_32BITrvrrrrrs    -    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)rr7r4)r(r7rg?g333333?g?rcS d|dSNrr(rr$r%rrr z>TestEstimateGradients2DGlobal.test_smoketest..rcSd|SNrrrrrrrr4cSd|SNrrrrrrrr2cSdd|d|SNr6L,@rrrrrr)r6rrr(ư>)tol)r7gh㈵>zitem %d)rtolatolerr_msg) rrfloatr-r. enumerater estimate_gradients_2d_globalrr) r#r$r/funcsjfuncgradr=Zdzrrrr's"      "  z,TestEstimateGradients2DGlobal.test_smoketestcCstttd}tj|jd}t|}t}| t j dt j ||ddWddS1s3wYdS)Nzestimate_gradients_hang.npyrz$Gradient estimation did not converger()maxiter) rloadrrdrfrr-r.rfilterr ZGradientEstimationWarningr)r#rFr?r/suprrrtest_regression_2359s "z2TestEstimateGradients2DGlobal.test_regression_2359N)rwrxryr'rrrrrr|s r|c@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) Nrb)rr2r4r3r}r~)rg?rrr(r)rrP2r7z_check_accuracy: abs(a-b):zip.grad:) rrd RandomStaterrr CloughTocher2DInterpolatorrfrAssertionErrorprintabsr) r#rr$r alternaterPkwrngr@pabrrr_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) %drrrrrPz!Function (alternate, rescaled) %d)rrrrrPrrrr#rrrrrrtest_linear_smoketest%s*    z4TestCloughTocher2DInterpolator.test_linear_smoketestc Csdddddddddg}t|D]\}}|j|dddd |d |j|dddd |d d qdS) NcS|dSNr7rrrrrr>rzITestCloughTocher2DInterpolator.test_quadratic_smoketest..cS|dSrrrrrrr?rcS|d|dSrrrrrrr@cS||SrmrrrrrrAr& .>g)\(?rrrTrrrrrrtest_quadratic_smoketest;s  z7TestCloughTocher2DInterpolator.test_quadratic_smoketestcCs^tjgdtjd}tj|jdtjd}|d|}t|}t|||}t ||dSr*) rrrrrr-r.r rr)r#r$r%r/r&rrrr1Js   z-TestCloughTocher2DInterpolator.test_tri_inputcCrYrZ) rrrrrr-r.r\rr]r rr^rrrr_Ur`z5TestCloughTocher2DInterpolator.test_tri_input_rescalecCrUrV) rrrrrr-r.r rrFrrWrrrrXbs   z;TestCloughTocher2DInterpolator.test_tripoints_input_rescalerNc Csddddddddddg}tjd}tjtjgdtd |d d f}t|D]\}}|j||d d dd|d|j||d d dd|ddq-dS)NcSrrrrrrrrrrz;TestCloughTocher2DInterpolator.test_dense..cSrrrrrrrrsrcSrrrrrrrrtrcSrrmrrrrrrurcSs(tdtj|tdtj|Sr)rcospisinrrrrrvs(i)rr2r4r3rir7rg{Gzt?g{Gz?r)r$rrrrT)r$rrrrrP) rrdrZr_rrrfrr)r#rrgridrrrrr test_densens$  z)TestCloughTocher2DInterpolator.test_densecCs.tjdd}tjd}tttj||dS)Nrcr6)rrdrandn assert_raisesr]r r)r#r$r%rrrtest_wrong_ndims z.TestCloughTocher2DInterpolator.test_wrong_ndimcCsftjd}|dd}|dd|d}t||}tt|}t |dd|dddSra) rrdrrfr rrgrhrir)r#rr$r%r@rjrrrrks   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) Nrr4rr6r7)r(rrr.rr()rrsqrtr rrrrrrrdrrdotrD)r#rFr?r@alphap1Zp2Zv1Zv2rArZw1Zw2rrrtest_boundary_tri_symmetrys,  "6       z9TestCloughTocher2DInterpolator.test_boundary_tri_symmetry)NrFF)rwrxryrrrr1r_rXr\rzZ fail_slowrrrkrrrrrrs !   r)rsysnumpyrZ numpy.testingrr\rrZscipy._lib._array_apirrZscipy._lib._testutilsrZscipy.interpolate._interpndZ interpolateZ _interpndr Zscipy.spatial._qhullZspatialZ_qhullr-rgrrmaxsizer{rrr|rrrrrs$    E%