U
    i2e                     @   s@   d dl Zd dlmZ d dlmZ d dlmZ dd Z	dd Z
dS )    N)
MouseEvent)Line3DCollectionc                  C   s   t  } | jdd}d|_d|_d|_d|_ddddd	g}||||}| j	  |
 }|j|}d
D ]}||_d|_| j	  tdD ]d}td| jf||d d f  }||\}	}
|	dkstt|
d dkst|
d d |kstqqjd S )N3dZ
projectionr   iT            )ii   Zbutton_press_eventind)pltfigureadd_subplotZrollZelevazimstalescatterZcanvasZ	draw_idleZget_offsetsZ	transDataZ	transformranger   containsAssertionErrorlen)figaxxZscatter_collectionZscatter_offsetZscatter_locationr   ieventr   r    r   I/tmp/pip-unpacked-wheel-qhyxngld/mpl_toolkits/mplot3d/tests/test_art3d.py'test_scatter_3d_projection_conservation	   s.    


r   c                  C   s   t jdddgddt jdddgddfg} t jddgddt jddgddt jddgddg}t }|jdd}|t|  |j|ddi t  d S )	Ng        float)Zdtypeg      ?r   r   ZvisibleF)	npZfromiterr   r   r   Zadd_collectionr   r   Zdraw)lcZpcr   r   r   r   r   test_zordered_error,   s    r!   )Znumpyr   Zmatplotlib.pyplotZpyplotr   Zmatplotlib.backend_basesr   Zmpl_toolkits.mplot3d.art3dr   r   r!   r   r   r   r   <module>   s
   #