o
    h                     @   s   d dl Zd dlZd dlm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ejdddgejdee	gdd ZdS )    N)assert_array_almost_equal)	csr_array)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   
null_valuer   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   TFnparrayr   r   r   r   csgraphbfirstdirectedbfirst_test r   d/var/www/html/scripts/venv/lib/python3.10/site-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   dfirstr   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_graph1      r(   c                  C   r&   r'   r"   r   r   r   r   $test_graph_depth_first_trivial_graph=   r)   r*   r   TF	tree_funcc                 C   sd   t dgtjdgdggtjdfdd}|jjtjksJ | |d|d}tt|ddgddgg d S )Nr	   r   )dtype)r
   r
   )shape)r   )r   r   r   int64indicesr,   r   r   )r+   r   gtreer   r   r   test_int64_indicesI   s   &r2   )numpyr   pytestnumpy.testingr   scipy.sparser   scipy.sparse.csgraphr   r   r   r   r    r%   r(   r*   markparametrizer2   r   r   r   r   <module>   s    