o
    EbP                     @   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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ddg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?q;G d(d) d)Z,G d*d+ d+Z-G d,d- d-Z.G d.d/ d/Z/d0d1 Z0G d2d3 d3Z1G d4d5 d5Z2G d6d7 d7Z3dS )8    N)assert_equalassert_almost_equalassert_assert_allcloseassert_array_equal)raises)cKDTree)Voronoic                 C      t t| S Ntuplesortedx r   @/usr/lib/python3/dist-packages/scipy/spatial/tests/test_qhull.pysorted_tuple      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rd}|d| g}t|t||D ]}|||||   q+d| |f }|tvsEJ ||ft|< dS )	z<
    Generate incremental datasets from basic data sets
    r$   Nr,   r5   zQJ Ppr6      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   
rG   )r$   r0      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/   r0   )qhullZ_Qhullr   arraycopyZ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   rN   r	   )rS   rA   r   r   r   test_issue_8051   s   @zTest_Qhull.test_issue_8051N)__name__
__module____qualname__rU   rV   r   r   r   r   rI      s    &rI   c                   @   sh   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 ]&}||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   rN   doublerM   Delaunayr   verticesfind_simplexZtsearch)rS   rA   tripirE   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  }
||d  }t 	|	| |
| }|t 
t || }|t |d  9 }t ||| }t|| | q-d S )N)r"   r&   r%   )g]2ʐ?g>mj?)g~T!?g^Rz?r\   gB?g?r   r$   r,   )r   rN   rd   rM   re   Zlift_pointsZplane_distance	enumeraterf   Zcrosssqrtdotsignr   )rS   r   ri   rh   zZpzdistrE   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 )Nr[   r\   r/   r,   r$   r   )r   rN   rd   rM   re   r   convex_hullrS   rA   rh   r   r   r   test_convex_hull   s   
(zTestUtilities.test_convex_hullc                 C   @   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$   rK   +=zVolume of cube is incorrectrtolrc         @zArea of cube is incorrectr   rN   rM   
ConvexHullr   volumearearS   rA   hullr   r   r   test_volume_area   s   



zTestUtilities.test_volume_areac                 C   ry   )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 incorrectr|   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 ]U}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	 q5d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   r}   )	r   randomrM   r   r;   
add_pointsr   r   r   )rS   Z	nr_pointsZdimrA   Zinc_hullZinc_restart_hullrj   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 sMw   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 d	S )z@Check that a triangulation has reasonable barycentric transformsr$   rK   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)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   )rA   rf   rB   sumr   finfofloateps	transformZerrstateisnanallabsr   nonzerorN   r;   ZnsimplexZisfiniterm   rg   any)rS   rh   rc   	unit_cubeunit_cube_tolrf   ZscZ	centroidsr   r   cokr   rE   Zat_boundaryr   r   r   _check_barycentric_transforms  s2   0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__rP   rM   re   r   r   r   r   r   )rS   r   rA   rh   Z	bad_countr   r   r   &test_degenerate_barycentric_transformsL  s   
z4TestUtilities.test_degenerate_barycentric_transformsc           	   
   C   s2  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      r1   r/   )r,   r/   r0   r1   r,   r2   r   r$   zndim=%dT)rc   r   r!   ra   r(   )rc   r   r   )r   r   r   r   r;   Zlinspacec_r   r   ravelZbroadcast_arraysZix_TrM   re   r   r   seedrandr:   r_)	rS   r   ZnpointsrB   r   Zgridrc   rh   mr   r   r    test_more_barycentric_transforms]  s0   *
:
z.TestUtilities.test_more_barycentric_transformsN)r   Fr   )rW   rX   rY   __doc__rk   ru   rx   r   r   r   r   r   pytestmarkZslowr   r   r   r   r   rZ      s    
-rZ   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r%|| | qq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   rE   r   r   r   r     s    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 != %rrb   )r;   rA   r:   	simplicesaddZvertex_neighbor_verticesr   )rS   rh   expectedsr   r   Zgotr   r   r   _check  s   

z!TestVertexNeighborVertices._checkc                 C   ,   t jg dt jd}t|}| | d S )N)r"   r#   r%   r\   r   rN   rd   rM   re   r   rw   r   r   r   test_triangle     
z(TestVertexNeighborVertices.test_trianglec                 C   r   )Nr[   r\   r   rw   r   r   r   test_rectangle  r   z)TestVertexNeighborVertices.test_rectanglec                 C   r   )N)r"   r#   r&   r%   r'   )g?r(   r\   r   rw   r   r   r   test_complicated  s
   

z+TestVertexNeighborVertices.test_complicatedN)rW   rX   rY   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      t jd}tttj| d S Nr$   )r   ma
masked_allrQ   
ValueErrorrM   re   rS   Zmasked_arrayr   r   r   test_masked_array_fails     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   rN   nanrd   rQ   r   rM   re   rS   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 ]R}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,   r4   r$   rK   r*   r\   )r;   r   zerosrM   re   rf   r   r   aranger   	neighbors)rS   ZndrA   rE   rh   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 )Nr[   r\   r]   r^   )r*   r*   r$   )r*   r*   r   )r   rN   rd   rM   re   r   rf   r   rw   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   rN   Zfloat64r   rM   re   r   )rS   r   rT   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 )Nr6   r7   )r9   rM   re   r   rA   rf   maxminrw   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   rM   re   r   r   r   r   r   r<   rw   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   rM   re   r   r<   r   r   r   Zcoplanarr   Zvertex_to_simplexrw   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$   r0   r   )r0   r,   r   )rM   re   r   rN   r   r   )rS   rA   rh   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rJ|jtj|dd  dddd |drrt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pathologicalrL   )r>   r   concatenaterM   re   r   r<   
startswithr   r   r   r   r   r:   r    r   )	rS   r?   rD   rC   rA   objchunkobj2obj3r   r   r   test_incremental  s8   




zTestDelaunay.test_incrementalN)rW   rX   rY   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 ]\}}|D ]\}}| | | |  }|tj	| }t
|d  |d g}	t|	| | | |  }
t|	| | | |  }tt|
|	dslq)tt||	dswq)t|| | | |  }
t|| | | |  }t|| | | |  }|
| k s|
|| krq)|| k s||| krq) ntdq#d S t|| d S )Nr$   r,   i  r   zcomparison fails)r   r   r   r:   r   r   r   r   r   ZnormrN   rn   allcloseAssertionErrorr   )rA   Zfacets_1Zfacets_2r   r   r   ZapZbptrs   Zc1Zc2Zc3r   r   r   assert_hulls_equal#  s6   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   r   r   )r   r   r   rQ   r   rM   r   r   r   r   r   r   [  r   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   rN   r   rd   rQ   r   rM   r   r   r   r   r   r   _  s   z)TestConvexHull.test_array_with_nans_failsr?   c                 C   sp   t | }t|}t|}t||j|j |jd dkr,tt	
|jt	|j d S tt	
|j|j d S )Nr$   r,   )r9   rM   re   r   r   rv   r   r:   r   r   r   r   rf   )rS   r?   rA   rh   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 ]}|| qt|}tj|d dd}t|dkrH|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   rM   r   r   r<   r   r   )	rS   r?   rD   _rA   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   rM   r   r   r   r   r   rf   rA   r   Zarctan2meanr   r   ZdiffZunwrap)rS   rA   r   r   rT   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 )Nrz   rK   r{   r   r~   r   rw   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 )	Nra   皙?r`   333333?QG4rA   r   r   )FTFFr\   r   rN   rM   r   boolgoodr   rS   r   rA   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 r-t |}t|| d S )Nra   r  r`   r  Tr  -)r   rN   rM   r   r   r  invertr   )rS   Znew_genr   r  rA   r   r
  r   r   r   test_good2d_incremental_changes  s   
'

z.TestConvexHull.test_good2d_incremental_changesc                 C   sv   t ddgddgddgddgddgg}tj||d}|j}|d u s$J |r7|t d |j}|d u s9J d S d S )Nra   r  r`   r  )rA   r   r$   r,   )r   rN   rM   r   r  r   r   )rS   r   rA   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 )Nra   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   rN   rM   r   r  r   r  )rS   r   rA   r   r   r   r   r   test_good3d  s   
zTestConvexHull.test_good3dN)rW   rX   rY   r   r   r   r   r   r   r9   r   r>   r   r  r   r  r   rN   r  r  r  r  r  r   r   r   r   r   Z  sN    

	


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   r   r   )r   r   r   rQ   r   rM   r	   r   r   r   r   r     r   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rS   rA   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  s    z1TestVoronoi._compare_qvoronoi.<locals>.<listcomp>r$   r   r/   r,   c                 S   "   g | ]}d d |dd D qS )c                 S      g | ]}t |d  qS r   r   r   rT   r   r   r   r   L      <TestVoronoi._compare_qvoronoi.<locals>.<listcomp>.<listcomp>r$   Nr   r  r   r   r   r   L      c                 S   r  )c                 S   s   g | ]}t |qS r   r  r   r   r   r   r   N      r"  r$   r/   r   r  r   r   r   r   N  r#  Nc                 S   r  )c                 S   r  r   r  r   r   r   r   r   P  r!  r"  r/   Nr   r  r   r   r   r   P  r#  c                 S   r
   r   r   r   r   r   r   	sorttupleV  r   z0TestVoronoi._compare_qvoronoi.<locals>.sorttuple)strip
splitlinesr   r   r   r   rM   r	   r   rf   r   r   regionszipridge_pointsr   ridge_verticesr   )rS   rA   r  kwZnvertexrf   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 ]F\}}tt|dks!q|j	| j
dd}d||d  |  }|j|| dd\}	}
t|
|d  |j|| dd\}	}
t|
|d  qd S )Nr   r   gư>r$   )r   )r9   KDTreerM   r	   Z
ridge_dictitemsr   r   asarrayrf   r   Zqueryr   )rS   r?   rA   Ztreer-  ri   rr   Zridge_midpointrt   rq   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   )rS   rA   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 ]}|| q,t|}tj|d d|d}t|dkrY|jtj|dd  dddd tt|j	t|j	 tt|j	t|j	 ||fD ]J ddit
 jD ]\}	}
t
|jD ]\}}t|
|r||	< qq~ fd	d
dd }t| j||j t| j||j qsd S )Nr   r$   r/   r   Tr   r   r*   c              
      s^   t | drttfdd| D S z|  W S  ty. } ztd j|  f |d }~ww )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   rf   )r   eZobjxr4  Z
vertex_mapr   r   r4    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     s    zDTestVoronoi.test_incremental.<locals>.simplified.<locals>.<listcomp>)r   r   r   remover   )r   r/  r   r   r   
simplified  s   
z0TestVoronoi.test_incremental.<locals>.simplified)r>   r:   r   r   rM   r	   r   r<   r   Zpoint_regionrl   rf   r   r(  r+  )rS   r?   rD   rC   rA   r   r   r   r   rj   rr   rE   Zv2r:  r   r8  r   r     sL   
	zTestVoronoi.test_incrementalN)rW   rX   rY   r   r  r  r   r   r   r   r9   r1  r   r2  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   r:   r   r   r  r   r   r   r   )rS   Zarr1Zarr2r}   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)   r?  r)   rK   r)   r?  r)   rK   r?  r(   r)   rK   )r   rN   rM   HalfspaceIntersectionr   intersections)rS   
halfspacesfeasible_pointrA   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 qN|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 qd S )Nr   zselfdual-4d-polytope.txtr$   )r)   r)   r)   r)   )   r0   g      @@rH  r2   r*   )r)   r)   r(         r   )r   r   r   r   r   r   Z
genfromtxthstackrN   rM   rC  r   rD  r:   r   Zdual_volumer<   dual_facetsrn   r=  r   dual_points	itertoolspermutationsr   r   )
rS   fnameZineqsrE  
feas_pointhsZfacetZdistsrA   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 )Nr>  r@  rA  rB  r(   r(   r(   r(   rI  )r   rN   rQ   r   rM   rC  Z
QhullError)rS   rE  rF  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 ]R\}}	||	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 qY|  d S )N)r)   r)   r?  rI  )r)   r?  r)   rI  )r?  r)   r)   rI  )rK   r)   r)   rI  )r)   rK   r)   rI  )r)   r)   rK   rI  )rK   rK   rK   )rK   rK   r?  )rK   r?  rK   )r$   r?  r?  r?  r4   r  Tr   r   r$   )r   rN   rJ  ZvstackrM   rC  rl   Zadd_halfspacesZnewaxisr   rE  rD  r=  rP   )rS   rE  Zextra_normalsZoffsetsZextra_halfspacesrP  Zinc_hsZ
inc_res_hsrj   ZehstotalrQ  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s`J t||jD ]\}}t|t|kstJ qft|j	| d S )N)r?  r)   r)   r)   )rK   r)   r)   r?  )r)   r?  r)   r)   )r)   rK   r)   r?  )r)   r)   r?  r)   )r)   r)   rK   r?  rT  )r   r   )r,   r   r   )r   rW  r   )r   r,   r   )r   r   rW  )r   r   r,   )r,   r0   r   )r0   r,   r$   )r1   r,   r   )r,   r1   r$   )r/   r0   r$   )r0   r/   r   )r1   r/   r$   )r/   r1   r   )
r   rN   rM   rC  r<   rK  r)  r   r   rL  )rS   rE  rR  rQ  Zqhalf_pointsZqhalf_facetsr   r   r   r   r   	test_cubeM  s<   

z$Test_HalfspaceIntersection.test_cubeN)r   )	rW   rX   rY   r=  rG  rS  rU  r   rX  r   r   r   r   r;    s    
*r;  )4r   rO   Znumpyr   Znumpy.testingr   r   r   r   r   r   r   rQ   Zscipy.spatial._qhullZspatialZ_qhullrM   Zscipy.spatialr   r.  r	   rM  r   r   r   r    r   r   rA   rN   Zpathological_data_1r   Zfloat_r   Zpathological_data_2r   Zbug_2850_chunksZbug_2850_chunks_2r0  r9   r>   rG   r?   r@   rI   rZ   r   r   r   r   r  r;  r   r   r   r   <module>   s   				




  $, W! 7 = K