o
    g;5                     @   s   d dl Z d dlZd dlmZmZmZmZ d dlm	Z
 d dlZd dlm  mZ d dlm  mZ d dlZdd ZG dd dZG dd	 d	ZG d
d dZdS )    N)assert_equalassert_allcloseassert_almost_equalsuppress_warnings)raisesc                 C   s    t jt jt jtd| S )Ndata)ospathjoinabspathdirname__file__)basename r   c/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/scipy/interpolate/tests/test_interpnd.py	data_file   s   r   c                   @   s\   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd ZdS )TestLinearNDInterpolationc                 C   sH   t jg dt jd}t j|jd t jd}t|||}t|| d S )Nr   r   )      r   )r         ?)r   r   )      ?333333?dtyper   nparraydoublearangeshapeinterpndLinearNDInterpolatorr   selfxyyir   r   r   test_smoketest   s   
z(TestLinearNDInterpolation.test_smoketestc                 C   s~   t jg dt jd}t j|jd t jd}t|d d df |d d df f||d d df |d d df }t|| d S )Nr   r   r      r   r#   r   r   r   test_smoketest_alternate   s   
Fz2TestLinearNDInterpolation.test_smoketest_alternatec                 C   sT   t jg dt jd}t j|jd t jd}|d|  }t|||}t|| d S Nr   r   r                 @r   r#   r   r   r   test_complex_smoketest'   s   
z0TestLinearNDInterpolation.test_complex_smoketestc                 C   ^   t jg dt jd}t j|jd t jd}|d|  }t|}t|||}t	|| d S r+   )
r   r   r   r   r    qhullDelaunayr!   r"   r   r$   r%   r&   trir'   r   r   r   test_tri_input1      

z(TestLinearNDInterpolation.test_tri_inputc                    s   t jg dt jd}t jg dt jd  fdd}t t dddd d d f t dddd d d f \}}| }| }t ||gj }t	| |}t
|||| d S )	N)r   r   r)   r)   r)   r)   r   r   g      ?g       @g      g      @c           	         s   | | dk}| }| | }|| }| | }|| }d|  } d d| |   d |   d |  ||<  d || d   d d|    d d|   ||< |S )Nr)   r         r   )	r%   r&   t1t2x1y1x2y2zvaluesr   r   ipJ   s$   

z1TestLinearNDInterpolation.test_square.<locals>.ipr   r)      r   r   r   broadcast_arrayslinspaceravelTcopyr!   r"   r   )r$   pointsrD   xxyyxizir   rB   r   test_square<   s   	z%TestLinearNDInterpolation.test_squarec                 C   sL   t jg dt jd}t j|jd t jd}tj||dd|}t|| d S )Nr   )rS   )rS      )rT   rT   )g      @r9   r   r   Trescaler   r#   r   r   r   test_smoketest_rescalei   s   
z0TestLinearNDInterpolation.test_smoketest_rescalec                 C   s   t jg dt jd}t jg dt jd}t t dddd d d f t dddd d d f \}}| }| }t ||gj }t	|||}tj	||dd	|}t
|| d S )
N)r   )r   d   )
   rX   )rY   r   r   r8   r   rY   rE   rX   TrU   rF   )r$   rL   rC   rM   rN   rO   rP   zi_rescaledr   r   r   test_square_rescaler   s   z-TestLinearNDInterpolation.test_square_rescalec                 C   v   t jg dt jd}t j|jd t jd}|d|  }t|}t|j	||}tj|j	|dd|}t
|| d S NrR   r   r   r,   TrU   )r   r   r   r   r    r/   r0   r!   r"   rL   r   r$   r%   r&   r2   r'   
yi_rescaler   r   r   test_tripoints_input_rescale   s   


z6TestLinearNDInterpolation.test_tripoints_input_rescalec                 C      t jg dt jd}t j|jd t jd}|d|  }t|}d}tjt	|d t
j||dd| W d    d S 1 s?w   Y  d S 	NrR   r   r   r,   zORescaling is not supported when passing a Delaunay triangulation as ``points``.)matchTrU   )r   r   r   r   r    r/   r0   pytestr   
ValueErrorr!   r"   r$   r%   r&   r2   rc   r   r   r   test_tri_input_rescale      

"z0TestLinearNDInterpolation.test_tri_input_rescalec                 C   l   t jd t jdd}t jddt jd  }t||}tt|}t	|dd|dd d S N     r:   y              ?r   )
r   randomseedrandr!   r"   pickleloadsdumpsr   r$   r%   r&   rD   ip2r   r   r   test_pickle      z%TestLinearNDInterpolation.test_pickleN)__name__
__module____qualname__r(   r*   r-   r3   rQ   rW   r[   r`   rg   ru   r   r   r   r   r      s    		
-	r   c                   @   s   e Zd Zdd Zdd ZdS )TestEstimateGradients2DGlobalc           	      C   s   t jg dtd}t|}dd dfdd dfdd d	fd
d dfg}t|D ]=\}\}}||d d df |d d df }tj||dd}t|j	d t
|t |d d d f d|  ddd| d q(d S )N)r   )r   r:   r7   )r)   r:   r   g      ?g333333?g?r   c                 S      d|  d S Nr   r)   r   r%   r&   r   r   r   <lambda>       z>TestEstimateGradients2DGlobal.test_smoketest.<locals>.<lambda>r   c                 S      d|  S Nr   r   r   r   r   r   r          r7   c                 S      d| S Nr   r   r   r   r   r      r   r5   c                 S      dd|   d|  S Nr9   L,@r   r   r   r   r   r          )r9   r   r   r)   ư>)tol)   r:   gh㈵>zitem %d)rtolatolerr_msg)r   r   floatr/   r0   	enumerater!   estimate_gradients_2d_globalr   r    r   )	r$   r%   r2   funcsjfuncgradrA   dzr   r   r   r(      s"   





" 
z,TestEstimateGradients2DGlobal.test_smoketestc                 C   st   t td}t j|jd }t|}t }|	t
jd t
j||dd W d    d S 1 s3w   Y  d S )Nzestimate_gradients_hang.npyr   z$Gradient estimation did not converger)   )maxiter)r   loadr   rm   ro   r    r/   r0   r   filterr!   GradientEstimationWarningr   )r$   rL   rC   r2   supr   r   r   test_regression_2359   s   
"z2TestEstimateGradients2DGlobal.test_regression_2359N)rw   rx   ry   r(   r   r   r   r   r   rz      s    rz   c                   @   s^   e Zd ZdddZdd Zdd	 Zd
d Zdd Zdd Zdd Z	dd Z
dd Zdd ZdS )TestCloughTocher2DInterpolatorNr   Fc                 K   sP  t jd |d u rt jg dtd}|s.tj|||d d df |d d df d|d}n&tj|d d df |d d df f||d d df |d d df d|d}t jdd	}|sb||}	n||d d df |d d df }	||d d df |d d df }
zt|	|
fi | W d S  t	y   t
d
t|	|
  t
d|j  w )Nrk   )r   r5   r7   r6   r{   r|   )r   g?r   r   r)   r   )r   rV   2   r:   z_check_accuracy: abs(a-b):zip.grad:)r   rm   rn   r   r   r!   CloughTocher2DInterpolatorro   r   AssertionErrorprintabsr   )r$   r   r%   r   	alternaterV   kwrD   pabr   r   r   _check_accuracy   s2   
&" 
""z.TestCloughTocher2DInterpolator._check_accuracyc              
   C   s   dd dd dd dd g}t |D ]8\}}| j|dddd| d	 | j|dddd
d| d | j|dddd| d
d | j|dddd
d
d| d qd S )Nc                 S   r}   r~   r   r   r   r   r   r      r   zFTestCloughTocher2DInterpolator.test_linear_smoketest.<locals>.<lambda>c                 S   r   r   r   r   r   r   r   r      r   c                 S   r   r   r   r   r   r   r   r      r   c                 S   r   r   r   r   r   r   r   r      r   gvIh%<=gHz>Function %dr   r   r   r   TzFunction (alternate) %d)r   r   r   r   r   zFunction (rescaled) %dr   r   r   r   rV   z!Function (alternate, rescaled) %d)r   r   r   r   rV   r   r   r   r$   r   r   r   r   r   r   test_linear_smoketest   s*   z4TestCloughTocher2DInterpolator.test_linear_smoketestc              	   C   sd   dd dd dd dd g}t |D ]\}}| j|dddd	| d
 | j|dddd	| dd qd S )Nc                 S      | d S Nr:   r   r   r   r   r   r   	  r   zITestCloughTocher2DInterpolator.test_quadratic_smoketest.<locals>.<lambda>c                 S      |d S r   r   r   r   r   r   r   
  r   c                 S      | d |d  S r   r   r   r   r   r   r         c                 S      | | S Nr   r   r   r   r   r     r   &.>g)\(?r   r   r   Tr   r   r   r   r   r   test_quadratic_smoketest  s   z7TestCloughTocher2DInterpolator.test_quadratic_smoketestc                 C   r.   r+   )
r   r   r   r   r    r/   r0   r!   r   r   r1   r   r   r   r3     r4   z-TestCloughTocher2DInterpolator.test_tri_inputc                 C   ra   rb   )r   r   r   r   r    r/   r0   rd   r   re   r!   r   rf   r   r   r   rg      rh   z5TestCloughTocher2DInterpolator.test_tri_input_rescalec                 C   r\   r]   )r   r   r   r   r    r/   r0   r!   r   rL   r   r^   r   r   r   r`   -  s   

z;TestCloughTocher2DInterpolator.test_tripoints_input_rescalec              
   C   s   dd dd dd dd dd g}t jd t jt jg dtd	t jd
df }t|D ]\}}| j||dddd| d | j||dddd| dd q.d S )Nc                 S   r   r   r   r   r   r   r   r   <  r   z;TestCloughTocher2DInterpolator.test_dense.<locals>.<lambda>c                 S   r   r   r   r   r   r   r   r   =  r   c                 S   r   r   r   r   r   r   r   r   >  r   c                 S   r   r   r   r   r   r   r   r   ?  r   c                 S   s(   t dt j |  t dt j |  S r   )r   cospisinr   r   r   r   r   @  s   ( i  )r   r5   r7   r6   r   i  r:   r   g{Gzt?g{Gz?r   )r%   r   r   r   r   T)r%   r   r   r   r   rV   )	r   rm   rn   r_r   r   ro   r   r   )r$   r   gridr   r   r   r   r   
test_dense9  s$   z)TestCloughTocher2DInterpolator.test_densec                 C   s.   t jdd}t jd}tttj|| d S )Nrl   r9   )r   rm   randnassert_raisesre   r!   r   )r$   r%   r&   r   r   r   test_wrong_ndimM  s   z.TestCloughTocher2DInterpolator.test_wrong_ndimc                 C   ri   rj   )
r   rm   rn   ro   r!   r   rp   rq   rr   r   rs   r   r   r   ru   R  rv   z*TestCloughTocher2DInterpolator.test_picklec                 C   sV  t dddt dd fg}t g d}t||}d|jd< d	}t dt | dt | g}t dt t jd |  dt t jd |  g}||}||}t	|| t j
d
 t j
dd}	t j
d}
|	|jj|
d d d f  }|	||
 }|	||
 }t||}d|jd< ||}||}t	|| t	|| d S )Nr   r7   r   r9   r:   )r)   r   r   r   .r   r)   )r   r   sqrtr!   r   r   r   r   r   r   rm   rn   r   dotrJ   )r$   rL   rC   rD   alphap1p2v1v2Ar   w1w2r   r   r   test_boundary_tri_symmetry]  s,   
"6


z9TestCloughTocher2DInterpolator.test_boundary_tri_symmetry)Nr   FF)rw   rx   ry   r   r   r   r3   rg   r`   r   r   ru   r   r   r   r   r   r      s    
r   )r   numpyr   numpy.testingr   r   r   r   rd   r   r   scipy.interpolate.interpndinterpolater!   scipy.spatial._qhullspatial_qhullr/   rp   r   r   rz   r   r   r   r   r   <module>   s     %