o
    Eb?                     @   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 ]-}|| |< t jdd t| dd}W d    n1 s8w   Y  d| |< t| |  qt jt jfD ]1}d| |< || |< t jdd t| dd}W d    n1 spw   Y  d| |< t| |  qOd S )	N  
   r   g?皙?r   ignore)Zinvalid)
null_value)	nprandomseednaninfZerrstater   r   toarray)GZ
some_nullsZ	all_nullsr   G_csr r   M/usr/lib/python3/dist-packages/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|| q"d 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    