U
    ?h?                     @   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| |  qd 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   ]/var/www/html/venv/lib/python3.8/site-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|| 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   minimum)XZXcsrZXdenser   r   r   test_multiple_edges,   s    *.r   )numpyr   Znumpy.testingr   Zscipy.sparser   Zscipy.sparse.csgraphr   r   r   r   r   r   r   r   r   <module>   s   