U
    Gvf?  ã                   @   sL   d dl Zd dlmZ d dlmZ d dlmZmZ dd„ Z	dd„ Z
d	d
„ ZdS )é    N)Úassert_array_almost_equal)Ú
csr_matrix)Úcsgraph_from_denseÚcsgraph_to_densec               
   C   sÞ   t j d¡ t j d¡} | dk }| dk }dt jt jfD ]F}|| |< t jdd t| dd}W 5 Q R X d| |< t| | ¡ ƒ q6t jt jfD ]N}d| |< || |< t jdd t| dd}W 5 Q R X d| |< t| | ¡ ƒ qŠd S )	NéÒ  ©é
   r   gš™™™™™Ù?çš™™™™™é?r   Úignore)Úinvalid)Ú
null_value)	ÚnpÚrandomÚseedÚnanÚinfZerrstater   r   Útoarray)ÚGZ
some_nullsZ	all_nullsr   ÚG_csr© r   úO/tmp/pip-unpacked-wheel-96ln3f52/scipy/sparse/csgraph/tests/test_conversions.pyÚtest_csgraph_from_dense   s"    r   c                  C   sf   t j d¡ t j d¡} | dk }t j| |< t| ƒ}ddt j t jfD ]}|| |< t| t||ƒƒ qDd S )Nr   r   r	   r   r   )r   r   r   r   r   r   r   )r   Znullsr   r   r   r   r   Útest_csgraph_to_dense   s    
r   c               
   C   sÜ   t j d¡ t j d¡} t| ƒ}|jdd d… |jd d d…< | ¡ }t|d d …dd d…f | d d …d d d…f | d d …dd d…f  ƒ t|ƒ}t|d d …dd d…f t  | d d …d d d…f | d d …dd d…f ¡ƒ d S )Nr   r   é   é   )	r   r   r   r   Úindicesr   r   r   Zminimum)ÚXZXcsrZXdenser   r   r   Útest_multiple_edges,   s    *ÿ.ÿr   )Znumpyr   Znumpy.testingr   Zscipy.sparser   Zscipy.sparse.csgraphr   r   r   r   r   r   r   r   r   Ú<module>   s   