a
    .'aO                  T   @   s`  d dl Z d dlZd dlZd dlmZmZmZmZm	Z	 d dl
Z
d dl
mZ d dlm  mZ d dlmZ d dlmZ d dlZdd Zdd	 Zefd
dZejd g dZeddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddggQZeddgdd gddgd dgd d gd dgddeejj  gdd gddgg	Z!ej"dded d gd dgdd gddgggZ#e#ej"ddded d gd dgdd gddgg g Z$e%eej"ddej"ddej"ddej"ddej"dd ej"dd!ej"dd"ee!d#
Z&e#dfe$dfd$Z'd%d& Z(e&D ]Z)d'D ]Z*e(e)e* qܐqG d(d) d)Z+G d*d+ d+Z,G d,d- d-Z-G d.d/ d/Z.d0d1 Z/G d2d3 d3Z0G d4d5 d5Z1G d6d7 d7Z2dS )8    N)assert_equalassert_almost_equalassert_assert_allcloseassert_array_equal)raises)cKDTree)Voronoic                 C   s   t t| S Ntuplesortedx r   =lib/python3.9/site-packages/scipy/spatial/tests/test_qhull.pysorted_tuple   s    r   c                 C   s   t t| S r
   )r   npuniquer   r   r   r   sorted_unique_tuple   s    r   c                 C   sb   t | tjr|  } t |tjr(| }tt|| } |   tt||}|  t| | d S r
   )
isinstancer   Zndarraytolistlistmapsortr   )abtplr   r   r   !assert_unordered_tuple_list_equal   s    r     )r   r   r      r"   r   r"   r"         ?r&   )r&   g      ?gQ	gzGgQgHzG        gHzG?gQ?gzG@gQ	@r"   
            ?                     )
some-pointsz	random-2dz	random-3dz	random-4dz	random-5dz	random-6dz	random-7dz	random-8dpathological-1pathological-2)zbug-2850z
bug-2850-2c           	      C   s   t |  }|jd }d}|d }| dkr,d}n| dkr8d}|d| g}t|t||D ]}|||||   qVd| |f }|tvsJ ||ft|< dS )	z<
    Generate incremental datasets from basic data sets
    r"   Nr*   r3   zQJ Ppr4      z%s-chunk-%d)DATASETSshaperangelenappendINCREMENTAL_DATASETS)	name	chunksizepointsndimoptsZnminchunksjnew_namer   r   r   _add_inc_dataa   s    
rE   )r"   r.      c                   @   s   e Zd Zdd Zdd ZdS )
Test_Qhullc                 C   s  t dtddgddgddgddgddggd}t| }t dtddgddgddgddggd}t| }t| }|  t| }|  tt	|j tt	|j t
|d |d  t|d |d td t|d	 |d	 td t|d
 |d
 td t|d |d  t
|d |d  t|d |d td t|d	 |d	 td t|d
 |d
 td t|d |d  |  tt	|j |  tt	|j d S )N   vr   r"         ?r&   s   Qzg       @r   r*   r-   r.   )qhullZ_Qhullr   arraycopydeepcopyZget_voronoi_diagramcloseassert_raisesRuntimeErrorr   r   r   r   )selfr   ZxdyZydZxd2Zyd2r   r   r   test_swapping   s<    & zTest_Qhull.test_swappingc                 C   sL   t ddgddgddgddgddgddgddgddgddgg	}t| d S )Nr   r"   r*   )r   rL   r	   )rR   r?   r   r   r   test_issue_8051   s    @zTest_Qhull.test_issue_8051N)__name__
__module____qualname__rT   rU   r   r   r   r   rG      s   &rG   c                   @   sb   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dddZ
dd Zejjdd ZdS )TestUtilitiesz-
    Check that utility functions work.

    c                 C   s   t jg dt jd}t|}t|jg dg dg dD ]L}||d d }t||d d|f d t||d d }t|| q:d S )	Nr    r!   r$   r#   Zdtyper"   r-   r*   r-   r"   r   ))r+   r+   r"   )      ?r^   r   )333333?皙?r"   r*   z%rerr_msg)	r   rL   doublerK   Delaunayr   verticesfind_simplexZtsearch)rR   r?   tripirC   r   r   r   test_find_simplex   s    
zTestUtilities.test_find_simplexc                 C   s   t jg dt jd}t jddgt jd}t|}||}||}||}t|jD ]\}}||d  }	||d  }
||d  }t 	|	| |
| }|t 
t || }|t |d  9 }t ||| }t|| | qZd S )N)r    r$   r#   )g]2ʐ?g>mj?)g~T!?g^Rz?r[   gB?g?r   r"   r*   )r   rL   rc   rK   rd   Zlift_pointsZplane_distance	enumeratere   ZcrosssqrtdotZsignr   )rR   r   rh   rg   zZpzdistrC   vZx1Zx2Zx3ndr   r   r   test_plane_distance   s"    




z!TestUtilities.test_plane_distancec                 C   sF   t jg dt jd}t|}t|jddgddgddgddgg d S )NrZ   r[   r-   r*   r"   r   )r   rL   rc   rK   rd   r   convex_hullrR   r?   rg   r   r   r   test_convex_hull   s    
zTestUtilities.test_convex_hullc                 C   s@   t g d}t|}t|jdddd t|jdddd d S )N)r   r   r   )r   r"   r   )r"   r   r   )r"   r"   r   )r   r   r"   )r   r"   r"   )r"   r   r"   )r"   r"   r"   rI   +=zVolume of cube is incorrectrtolrb         @zArea of cube is incorrectr   rL   rK   
ConvexHullr   volumearearR   r?   hullr   r   r   test_volume_area   s    


zTestUtilities.test_volume_areac                 C   s@   t g d}t|}t|jdddd t|jdddd d S )N)
)gQ4?gΜR@?gX&lL5?)gfNz?gjM?g{D흚j?)ga?8\?g?gw5K?)g,lnq?g6=?gU?)gw컝?g`?ga?)g+q?g\hϱ?gLg?)gGe?gE9;?gd:L?)gȊKB?gBYM?g?)g~EHy?gUdk?gmVq"?)gqp?ge?g\*L?gY/?Hz>z(Volume of random polyhedron is incorrectry   g\4?z&Area of random polyhedron is incorrectr|   r   r   r   r   test_random_volume_area   s    


z%TestUtilities.test_random_volume_areac                 C   s  d}d}t j||f}tj|d|d ddf dd}tj|d|d ddf dd}t|d |D ]}t|d|d ddf }||||d ddf  |j|||d ddf dd t|j|jdd	 t|j|jdd	 t|j|jdd	 t|j|jdd	 qjdS )
zwTest that incremental mode gives the same volume/area as
        non-incremental mode and incremental mode with restart   r-   Nr"   TincrementalZrestartr   rz   )	r   randomrK   r}   r9   
add_pointsr   r~   r   )rR   Z	nr_pointsZdimr?   Zinc_hullZinc_restart_hullri   r   r   r   r   )test_incremental_volume_area_random_input  s    """z7TestUtilities.test_incremental_volume_area_random_input Fr   c                    s  |j |j }d|jd  }|jdd| }dd }ttj}	||j|}
tj	dd> t
|
jddt|
| | dk jddB }W d	   n1 s0    Y  t| d
|t| f  |d	d	d	dd	d	f |d	d	dd	d	d	f   t fddt|jD }t|jd	d	ddf |t|	k B }t| d
|t| f  ||}|dkt
|jd	d	ddf B }t| d
|t| f  |r||kjdd}||d| kjddO }|dk|B }t| d
|t| f  d	S )z@Check that a triangulation has reasonable barycentric transformsr"   rI   Zaxisc                 S   sB   | d d dd d f }| d d d dd d f }t d||| S )Nr(   z
ijk,ik->ij)r   Zeinsum)Ztrr   rZTinvr   r   r   barycentric_transform*  s    zJTestUtilities._check_barycentric_transforms.<locals>.barycentric_transformignore)Zinvalidg?Nz%s %sr(   c              	      s*   g | ]"}t j |d d d d f qS r
   )r   linalgZdet).0kqr   r   
<listcomp>9  s   z?TestUtilities._check_barycentric_transforms.<locals>.<listcomp>r   )r?   re   r@   sumr   finfofloateps	transformZerrstateisnanallabsr   nonzerorL   r9   ZnsimplexZisfiniterl   rf   any)rR   rg   rb   	unit_cubeunit_cube_tolre   ZscZ	centroidsr   r   cokr~   rC   Zat_boundaryr   r   r   _check_barycentric_transforms  s.    L2(
"z+TestUtilities._check_barycentric_transformsc                 C   sr   t tjtjtdd}|d }|  t	|}t 
|jd d ddf  }t|dk | | | d S )Ndatazdegenerate_pointset.npzr   r      )r   loadospathjoindirname__file__rO   rK   rd   r   r   r   r   r   )rR   r   r?   rg   Z	bad_countr   r   r   &test_degenerate_barycentric_transformsL  s    
z4TestUtilities.test_degenerate_barycentric_transformsc           	   
   C   s4  t tj}ddddd}tddD ]}t dd	|| }t jttt j	t j
t j|g|    j}d
| }t|}| j||dd t jd t j|jd dk }||d d f  d| t jj||d d f j d  7  < t|}| j||dd| d tt j||f }| j||dd| d q$d S )NF      r/   r-   )r*   r-   r.   r/   r*   r0   r   r"   zndim=%dT)rb   r   r   r`   r&   )rb   r   r   )r   r   r   r   r9   Zlinspacec_r   r   ravelZbroadcast_arraysZix_TrK   rd   r   r   seedrandr8   r_)	rR   r   Znpointsr@   r   Zgridrb   rg   mr   r   r    test_more_barycentric_transforms]  s.    *
:
z.TestUtilities.test_more_barycentric_transformsN)r   Fr   )rV   rW   rX   __doc__rj   rs   rv   r   r   r   r   r   pytestmarkZslowr   r   r   r   r   rY      s      
-rY   c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestVertexNeighborVerticesc                    s   dd t |jjd D }|jD ].}|D ]$}|D ]}||kr0|| | q0q(q |j\  fddt |jjd D }t||d||f d d S )Nc                 S   s   g | ]
}t  qS r   )setr   rC   r   r   r   r         z5TestVertexNeighborVertices._check.<locals>.<listcomp>r   c              
      s.   g | ]&}t tt | |d    qS r"   )r   r   intr   indicesZindptrr   r   r     s   z%r != %rra   )r9   r?   r8   	simplicesaddZvertex_neighbor_verticesr   )rR   rg   expectedsr   r   Zgotr   r   r   _check  s    

z!TestVertexNeighborVertices._checkc                 C   s,   t jg dt jd}t|}| | d S )N)r    r!   r#   r[   r   rL   rc   rK   rd   r   ru   r   r   r   test_triangle  s    
z(TestVertexNeighborVertices.test_trianglec                 C   s,   t jg dt jd}t|}| | d S )NrZ   r[   r   ru   r   r   r   test_rectangle  s    
z)TestVertexNeighborVertices.test_rectanglec                 C   s,   t jg dt jd}t|}| | d S )N)r    r!   r$   r#   r%   )g?r&   r[   r   ru   r   r   r   test_complicated  s
    

z+TestVertexNeighborVertices.test_complicatedN)rV   rW   rX   r   r   r   r   r   r   r   r   r     s   r   c                   @   sr   e Zd 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ejdeedd ZdS )TestDelaunayz*
    Check that triangulation works.

    c                 C   s   t jd}tttj| d S Nr"   )r   ma
masked_allrP   
ValueErrorrK   rd   rR   Zmasked_arrayr   r   r   test_masked_array_fails  s    z$TestDelaunay.test_masked_array_failsc                 C   s0   t jddddt jfgt jd}tttj| d S )Nr    r!   r$   r"   r[   )r   rL   nanrc   rP   r   rK   rd   rR   Zpoints_with_nanr   r   r   test_array_with_nans_fails  s    z'TestDelaunay.test_array_with_nans_failsc                 C   s   t ddD ]}t|d |f}t |D ]}d|||f< q(d|dd d f< t|}|j  t|jtj|d t	dd d d f  t|j
dtj|d t	dd d d f   q
d S )Nr*   r2   r"   rI   r(   r[   )r9   r   zerosrK   rd   re   r   r   aranger   	neighbors)rR   Zndr?   rC   rg   r   r   r   test_nd_simplex  s    

&zTestDelaunay.test_nd_simplexc                 C   sR   t jg dt jd}t|}t|jg dg dg t|jg dg dg d S )NrZ   r[   r\   r]   )r(   r(   r"   )r(   r(   r   )r   rL   rc   rK   rd   r   re   r   ru   r   r   r   test_2d_square  s    
zTestDelaunay.test_2d_squarec                 C   sp   t jg dt jd}t jg dt jd}t j||f }t j||f }tt j||f  tt j||f  d S )N)r   r"   r   r"   r[   )r   r   r"   r"   )r   rL   Zfloat64r   rK   rd   r   )rR   r   rS   ZxpZypr   r   r   test_duplicate_points  s    z"TestDelaunay.test_duplicate_pointsc                 C   s   t d }t|}t|j|j  |  t|j|j  |  t d }t|}t|j|j  |  t|j|j  |  d S )Nr4   r5   )r7   rK   rd   r   r?   re   maxminru   r   r   r   test_pathological  s    

zTestDelaunay.test_pathologicalc                 C   sN   t jdd}t j||f }tj|dd}tt |j	 t 
t| d S )Nr)   r*   z	QJ Qbb Pp)qhull_options)r   r   r   r   rK   rd   r   r   r   r   r   r:   ru   r   r   r   test_joggle  s    zTestDelaunay.test_jogglec                 C   s   t jdd}t j||f }t|}ttt |j	
 t|d k tt|jt|d k ttt |jd d df t|d k tt |jdk d S )Nr)   r*   r   )r   r   r   r   rK   rd   r   r:   r   r   r   Zcoplanarr   Zvertex_to_simplexru   r   r   r   test_coplanar  s    
$,zTestDelaunay.test_coplanarc                 C   s4   g d}t j|dd}tddg}t|j| d S )Nr    r!   r#   r%   )皙?r   Tfurthest_site)r"   r.   r   )r.   r*   r   )rK   rd   r   rL   r   r   )rR   r?   rg   r   r   r   r   test_furthest_site  s    zTestDelaunay.test_furthest_siter=   c           	      C   s  t | \}}tj|dd}tj|d d|d}|dd  D ]}|| q:t|}tj|d d|d}t|dkr|jtj|dd  dddd |drtt	|j
 t|jd  tt	|j
 t|jd  nt|j
|j
td t|j
|j
td d S )	Nr   r   Tr   r   r"   r   ZpathologicalrJ   )r<   r   concatenaterK   rd   r   r:   
startswithr   r   r   r   r   r8   r   r   )	rR   r=   rB   rA   r?   objchunkobj2obj3r   r   r   test_incremental  s8    



zTestDelaunay.test_incrementalN)rV   rW   rX   r   r   r   r   r   r   r   r   r   r   r   r   parametrizer   r<   r   r   r   r   r   r     s   r   c                 C   s  t tt|}t tt|}||kr| jd dkrdttj }|D ]4\}}|D ]\}}| | | |  }|tj	| }t
|d  |d g}	t|	| | | |  }
t|	| | | |  }tt|
|	dsqXtt||	dsqXt|| | | |  }
t|| | | |  }t|| | | |  }|
| k sX|
|| krXqX|| k sX||| krrqX qJqXtdqJd S t|| d S )Nr"   r*   i  r   zcomparison fails)r   r   r   r8   r   r   r   r   r   ZnormrL   rm   allcloseAssertionErrorr   )r?   Zfacets_1Zfacets_2r   r   r   ZapZbptrq   Zc1Zc2Zc3r   r   r   assert_hulls_equal#  s4    
r   c                   @   s  e Zd Zdd Zdd Zejdee	dd Z
ejdeedd	 Zd
d Zdd Zejdddgdd Zejdddgejdeddggejg dedfeddggejg dedfeddggejg dedfedd gd d ggejg d!edfedd"ggejg dedfgd#d$ Zejdddgd%d& Zejdddgd'd( Zejdddgd)d* Zd+S ),TestConvexHullc                 C   s   t jd}tttj| d S r   )r   r   r   rP   r   rK   r}   r   r   r   r   r   [  s    z&TestConvexHull.test_masked_array_failsc                 C   s.   t jdddt jfgt jd}tttj| d S )Nr    r$   r*   r[   )r   rL   r   rc   rP   r   rK   r}   r   r   r   r   r   _  s    z)TestConvexHull.test_array_with_nans_failsr=   c                 C   sn   t | }t|}t|}t||j|j |jd dkrVtt	
|jt	|j ntt	
|j|j d S )Nr"   r*   )r7   rK   rd   r}   r   rt   r   r8   r   r   r   r   re   )rR   r=   r?   rg   r   r   r   r   test_hull_consistency_tric  s    

z(TestConvexHull.test_hull_consistency_tric           	      C   s   t | \}}tj|dd}tj|d dd}|dd  D ]}|| q8t|}tj|d dd}t|dkr|jtj|dd  dddd t||j|j t||j|j d S )Nr   r   Tr   r"   r   )	r<   r   r   rK   r}   r   r:   r   r   )	rR   r=   rB   _r?   r   r   r   r   r   r   r   r   t  s    
zTestConvexHull.test_incrementalc                 C   s   t jd t jdd}t|}tt |jt 	|j
 |j|j
 j\}}t ||  ||  }tt t t |dk d S )Nr   r,   r*   r   )r   r   r   r   rK   r}   r   r   r   r   re   r?   r   Zarctan2meanr   r   ZdiffZunwrap)rR   r?   r   r   rS   Zangler   r   r   test_vertices_2d  s    
zTestConvexHull.test_vertices_2dc                 C   s<   t g d}t|}t|jddd t|jddd d S )Nrw   rI   rx   r   r{   r|   ru   r   r   r   r     s    
zTestConvexHull.test_volume_arear   FTc                 C   s^   t ddgddgddgddgddgg}tj||dd}t jg dtd}|j}t|| d S )	Nr`   皙?r_   333333?QG4r?   r   r   )FTFFr[   r   rL   rK   r}   boolgoodr   rR   r   r?   r   r   actualr   r   r   test_good2d  s    
zTestConvexHull.test_good2d
visibilityr   zQG-4znew_gen, expectedr_   gffffff?)FFFFFr[   gffffff)FTFFFg=
ףp=?)FFFTTr&   r   )FFTFFg433333?c                 C   sh   t ddgddgddgddgddgg}tj|d|d}|| |j}d|v rZt |}t|| d S )Nr`   r   r_   r   Tr   -)r   rL   rK   r}   r   r  invertr   )rR   Znew_genr   r  r?   r   r  r   r   r   test_good2d_incremental_changes  s    '


z.TestConvexHull.test_good2d_incremental_changesc                 C   sr   t ddgddgddgddgddgg}tj||d}|j}|d u sHJ |rn|t d |j}|d u snJ d S )Nr`   r   r_   r   )r?   r   r"   r*   )r   rL   rK   r}   r  r   r   )rR   r   r?   r   r  r   r   r   test_good2d_no_option  s    
z$TestConvexHull.test_good2d_no_optionc                 C   s^   t ddgddgddgddgddgg}tj||dd}t jg dtd}|j}t|| d S )Nr`   r   r_   r   r   )FFFFr[   r   r  r   r   r   test_good2d_inside  s    
z!TestConvexHull.test_good2d_insidec                 C   sZ   t g dg dg dg dg dg}tj||dd}t jg dtd	}t|j| d S )
Nr'   r'   r'   )gN7?g+xٿgg>@?)gFE%'?gWۧ=g/J?)g6r?gtĭ4gŷ)gw6y?gQgb?ZQG0r   )TFFFr[   )r   rL   rK   r}   r   r   r  )rR   r   r?   r   r   r   r   r   test_good3d  s    
zTestConvexHull.test_good3dN)rV   rW   rX   r   r   r   r   r   r   r7   r   r<   r   r   r   r  r   rL   r   r  r
  r  r  r   r   r   r   r   Z  sL   

	


r   c                   @   sh   e Zd Zdd Zdd Zdd Zejde	e
dd	 Zd
d Zdd Zejde	edd ZdS )TestVoronoic                 C   s   t jd}tttj| d S r   )r   r   r   rP   r   rK   r	   r   r   r   r   r     s    z#TestVoronoi.test_masked_array_failsc                 C   s   g d}d}|  || d S )N)	r    r!   )r   r*   r#   r$   r	  )r*   r   )r*   r"   )r*   r*   a	  
        2
        5 10 1
        -10.101 -10.101
           0.5    0.5
           0.5    1.5
           1.5    0.5
           1.5    1.5
        2 0 1
        3 2 0 1
        2 0 2
        3 3 0 1
        4 1 2 4 3
        3 4 0 2
        2 0 3
        3 4 0 3
        2 0 4
        0
        12
        4 0 3 0 1
        4 0 1 0 1
        4 1 4 1 2
        4 1 2 0 2
        4 2 5 0 2
        4 3 4 1 3
        4 3 6 0 3
        4 4 5 2 4
        4 4 7 3 4
        4 5 8 0 4
        4 6 7 0 3
        4 7 8 0 4
        _compare_qvoronoirR   r?   outputr   r   r   test_simple  s     zTestVoronoi.test_simplec                 K   s^  dd |   D }t|d d }ttt|dd|  }t|d d }dd |d| d| |  D }dd |d| | d	 D }d
d |d| | d	 D }	tj|fi |}
dd }t|
j	| t
ttt|
jttt| tttt||tt||	}tttt||
j tt||
j}|  |  t
|| d	S )z:Compare to output from 'qvoronoi o Fv < data' to Voronoi()c                 S   s   g | ]}t tt| qS r   )r   r   r   splitr   r   r   r   r   r   H  r   z1TestVoronoi._compare_qvoronoi.<locals>.<listcomp>r"   r   r-   r*   c                 S   s"   g | ]}d d |dd D qS )c                 S   s   g | ]}t |d  qS r   r   r   rS   r   r   r   r   L  r   <TestVoronoi._compare_qvoronoi.<locals>.<listcomp>.<listcomp>r"   Nr   r  r   r   r   r   L  s   c                 S   s"   g | ]}d d |dd D qS )c                 S   s   g | ]}t |qS r   r  r  r   r   r   r   N  r   r  r"   r-   r   r  r   r   r   r   N  s   Nc                 S   s"   g | ]}d d |dd D qS )c                 S   s   g | ]}t |d  qS r   r  r  r   r   r   r   P  r   r  r-   Nr   r  r   r   r   r   P  s   c                 S   s   t t| S r
   r   r   r   r   r   	sorttupleV  s    z0TestVoronoi._compare_qvoronoi.<locals>.sorttuple)strip
splitlinesr   r   r   r   rK   r	   r   re   r   r   regionszipridge_pointsr   ridge_verticesr   )rR   r?   r  kwZnvertexre   Znregionr  r  r  vorr  Zp1Zp2r   r   r   r  D  s4    "zTestVoronoi._compare_qvoronoir=   c                 C   s   t | }t|}t|}|j D ]\}}tt|dksBq$|j	| j
dd}d||d  |  }|j|| dd\}	}
t|
|d  |j|| dd\}	}
t|
|d  q$d S )Nr   r   gư>r"   )r   )r7   KDTreerK   r	   Z
ridge_dictitemsr   r   asarrayre   r   Zqueryr   )rR   r=   r?   Ztreer!  rh   rp   Zridge_midpointrr   ro   r   r   r   r   test_ridgese  s    
zTestVoronoi.test_ridgesc                 C   s    g d}d}| j ||dd d S )Nr   a'  
        2
        3 5 1
        -10.101 -10.101
        0.6000000000000001    0.5
           0.5 0.6000000000000001
        3 0 2 1
        2 0 1
        2 0 2
        0
        3 0 2 1
        5
        4 0 2 0 2
        4 0 4 1 2
        4 0 1 0 1
        4 1 4 0 1
        4 2 4 0 2
        Tr   r  r  r   r   r   r   ~  s    zTestVoronoi.test_furthest_sitec                 C   s8   g d}t |}t|jd t |dd}t|jd d S )Nr   FTr   )r	   r   r   )rR   r?   r!  r   r   r   test_furthest_site_flag  s
    z#TestVoronoi.test_furthest_site_flagc                    s  t | d d jd dkrd S t | \}}tj|dd}tj|d d|d}|dd  D ]}|| qXt|}tj|d d|d}t|dkr|jtj|dd  dddd tt|j	t|j	 tt|j	t|j	 ||fD ] ddit
 jD ]4\}	}
t
|jD ] \}}t|
|r||	< qq fd	d
dd }t| j||j t| j||j qd S )Nr   r"   r-   r   Tr   r   r(   c              
      sn   t | dr$ttfdd| D S z
|  W S  tyh } z"td j|  f |W Y d }~n
d }~0 0 d S )N__len__c                    s   g | ]} |qS r   r   r  )remapr   r   r     r   z?TestVoronoi.test_incremental.<locals>.remap.<locals>.<listcomp>z,incremental result has spurious vertex at %r)hasattrr   r   KeyErrorr   re   )r   eZobjxr(  Z
vertex_mapr   r   r(    s    


z+TestVoronoi.test_incremental.<locals>.remapc                 S   s:   t tt| }d|v r |d dd |D }|  |S )Nr   c                 S   s   g | ]}t |d kr|qS r   )r:   r  r   r   r   r     r   zDTestVoronoi.test_incremental.<locals>.simplified.<locals>.<listcomp>)r   r   r   remover   )r   r#  r   r   r   
simplified  s    
z0TestVoronoi.test_incremental.<locals>.simplified)r<   r8   r   r   rK   r	   r   r:   r   Zpoint_regionrk   re   r   r  r  )rR   r=   rB   rA   r?   r   r   r   r   ri   rp   rC   Zv2r.  r   r,  r   r     sF    
	zTestVoronoi.test_incrementalN)rV   rW   rX   r   r  r  r   r   r   r   r7   r%  r   r&  r<   r   r   r   r   r   r    s   )!
r  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S )Test_HalfspaceIntersectionr   c                 C   sx   t |j|j tj|jd ftd}|D ]>}tt|| |k jddd }t |jd d||d < q(t|  dS )z2Check that every line in arr1 is only once in arr2r   r[   r"   r   r   TN)	r   r8   r   r   r   r   r   r   r   )rR   Zarr1Zarr2rz   Ztruthsl1Zindexesr   r   r   assert_unordered_allclose  s    "z4Test_HalfspaceIntersection.assert_unordered_allclosec                 C   sn   t g dg dg dg dg}t ddg}t ddgddgddgddgg}t||}t|j| d S )N      r'   r'   r'   r3  r'   rI   r'   r3  r'   rI   r3  r&   r'   rI   )r   rL   rK   HalfspaceIntersectionr   intersections)rR   
halfspacesfeasible_pointr?   r   r   r   r    test_cube_halfspace_intersection  s    
"z;Test_HalfspaceIntersection.test_cube_halfspace_intersectionc           
      C   s<  t jt jtdd}t|}t|d d dd f |d d d df f }tg d}t	
||}t|jjd t|jd tt|jd |jD ]}tt|d q|d d d	f |d d d d	f | }| |d d d d	f j| j|j tg d
}|D ]$}	tt|j|	kjddd qd S )Nr   zselfdual-4d-polytope.txtr"   )r'   r'   r'   r'   )   r.   g      @@r<  r0   r(   )r'   r'   r&         r   )r   r   r   r   r   r   Z
genfromtxthstackrL   rK   r7  r   r8  r8   r   Zdual_volumer:   dual_facetsrm   r1  r   dual_points	itertoolspermutationsr   r   )
rR   fnameZineqsr9  
feas_pointhsZfacetZdistsr?   pointr   r   r   $test_self_dual_polytope_intersection  s"    
0
*&z?Test_HalfspaceIntersection.test_self_dual_polytope_intersectionc                 C   s   t g dg dg dg dg}t g d}tttj|| t dgdgg}tttj|| t ddgg}tttj|| t ddg}ttjtj|| d S )Nr2  r4  r5  r6  r&   r&   r&   r&   r=  )r   rL   rP   r   rK   r7  Z
QhullError)rR   r9  r:  r   r   r   test_wrong_feasible_point  s    
z4Test_HalfspaceIntersection.test_wrong_feasible_pointc              	   C   sd  t g dg dg dg dg dg dg}t g dg dg d	g d
g}t dggd }t t || f|f}t g d}tj||dd}tj||dd}t|D ]\}}	||	t jd d f  |j|	t jd d f dd t ||d |d d d f f}
t|
|}t	|j
|j
 t	|j
|j
 t	|j|j | |j|j q|  d S )N)r'   r'   r3  r=  )r'   r3  r'   r=  )r3  r'   r'   r=  )rI   r'   r'   r=  )r'   rI   r'   r=  )r'   r'   rI   r=  )rI   rI   rI   )rI   rI   r3  )rI   r3  rI   )r"   r3  r3  r3  r2   r  Tr   r   r"   )r   rL   r>  ZvstackrK   r7  rk   Zadd_halfspacesZnewaxisr   r9  r8  r1  rO   )rR   r9  Zextra_normalsZoffsetsZextra_halfspacesrD  Zinc_hsZ
inc_res_hsri   ZehstotalrE  r   r   r   r   #  s:    

"z+Test_HalfspaceIntersection.test_incrementalc              	   C   s   t g dg dg dg dg dg dg}t g d}t||}t g dg d	g d
g dg dg dg}g dg dg dg dg dg dg dg dg}t|t|jksJ t||jD ]\}}t|t|ksJ qt|j	| d S )N)r3  r'   r'   r'   )rI   r'   r'   r3  )r'   r3  r'   r'   )r'   rI   r'   r3  )r'   r'   r3  r'   )r'   r'   rI   r3  rH  )r   r   )r*   r   r   )r   rK  r   )r   r*   r   )r   r   rK  )r   r   r*   )r*   r.   r   )r.   r*   r"   )r/   r*   r   )r*   r/   r"   )r-   r.   r"   )r.   r-   r   )r/   r-   r"   )r-   r/   r   )
r   rL   rK   r7  r:   r?  r  r   r   r@  )rR   r9  rF  rE  Zqhalf_pointsZqhalf_facetsr   r   r   r   r   	test_cubeM  s<    

z$Test_HalfspaceIntersection.test_cubeN)r   )	rV   rW   rX   r1  r;  rG  rI  r   rL  r   r   r   r   r/    s   
*r/  )3r   rM   Znumpyr   Znumpy.testingr   r   r   r   r   r   r   rP   Zscipy.spatial.qhullZspatialrK   Zscipy.spatialr   r"  r	   rA  r   r   r   r   r   r   r?   rL   Zpathological_data_1r   Zfloat_r   Zpathological_data_2r   Zbug_2850_chunksZbug_2850_chunks_2r$  r7   r<   rE   r=   r>   rG   rY   r   r   r   r   r  r/  r   r   r   r   <module>   s     $, W! 7 = K