o
    čEb	  ć                   @   sP   d dl Zd dlmZ d dlmZmZmZmZ dd Z	dd Z
dd	 Zd
d ZdS )é    N)Śassert_array_almost_equal)Śbreadth_first_treeŚdepth_first_treeŚcsgraph_to_denseŚcsgraph_from_densec                  C   s   t  g d¢g d¢g d¢g d¢g d¢g”} t| dd} t  g d¢g d¢g d	¢g d
¢g d
¢g”}dD ]}t| d|}tt|| q0d S )N©r   é   é   r   r   ©r   r   r   r   é   ©r	   r   r   é   r   ©r   r   r   r   r   ©r   r   r   r   r   r   ©Z
null_value©r   r   r   r   r   )r   r   r   r   r   ©r   r   r   r   r   ©TF©ŚnpŚarrayr   r   r   r   ©ŚcsgraphZbfirstŚdirectedZbfirst_test© r   śK/usr/lib/python3/dist-packages/scipy/sparse/csgraph/tests/test_traversal.pyŚtest_graph_breadth_first   ó&   
ü
ü’žr   c                  C   s   t  g d¢g d¢g d¢g d¢g d¢g”} t| dd} t  g d¢g d	¢g d
¢g d¢g d¢g”}dD ]}t| d|}tt|| q0d S )Nr   r
   r   r   r   r   r   )r   r   r   r   r   r   r   )r   r   r   r   r   )r   r   r   r   r   r   ©r   r   r   r   r   r   )r   Zdfirstr   Zdfirst_testr   r   r   Śtest_graph_depth_first   r   r   c                  C   óP   t  dgg”} t| dd} t  dgg”}dD ]}t| d|}tt|| qd S ©Nr   r   r   r   r   r   r   r   Ś&test_graph_breadth_first_trivial_graph/   ó   ’žr"   c                  C   r    r!   r   r   r   r   r   Ś$test_graph_depth_first_trivial_graph;   r#   r$   )Znumpyr   Znumpy.testingr   Zscipy.sparse.csgraphr   r   r   r   r   r   r"   r$   r   r   r   r   Ś<module>   s    