U
    ?h5
                     @   sT   d dl Zd dlmZ d dlmZmZ d dlmZm	Z	m
Z
 dd Zdd Zd	d
 ZdS )    N)assert_equal)reverse_cuthill_mckeestructural_rank)
csc_matrix
csr_matrix
coo_matrixc                  C   s  t jddddddddgddddddddgddddddddgddddddddgddddddddgddddddddgddddddddgddddddddggtd} t| }t|}t ddddddd	dg}t|| |jd
|_|jd
|_t|d}t|| d S )N   r   Zdtype                  Zint64T)	nparrayintr   r   r   indicesZastypeZindptr)Agraphpermcorrect_perm r   \/var/www/html/venv/lib/python3.8/site-packages/scipy/sparse/csgraph/tests/test_reordering.py test_graph_reverse_cuthill_mckee   s&    	

r   c               A   C   sr  t jdtd} t ddddddddddddddddddddddddd	d	d	d
d
d
d
ddddddddddddddddddddddddddddddddg?}t dddddddddddd
dddddd	dddd
dddd	dddd
dddddddddddddddddddddddddd	dddd
dddg?}t| ||ff }t|}t dddd	ddddddd
dddddg}t|| d S )N?   r	   r   r   r   r   r   r   r
   r      	   
                  )r   Zonesr   r   r   Ztocsrr   r   )datarowscolsr   r   r   r   r   r   )test_graph_reverse_cuthill_mckee_ordering   s                                                                                                         r'   c                  C   s  t dddgdddgdddgg} tt| d tdddddddddddddddddddddddg}tdddddddddddddddddddddddg}t|}t|||ffd	d
}tt|d t ddddgddddgg}tt|d tt|jd d S )Nr   r   r   r   r   r   r
   r   )r   r   )shape)r   r   r   r   r   Z	ones_liker   T)r   r%   r&   r$   BCr   r   r   test_graph_structural_rank2   s    
88

r,   )numpyr   Znumpy.testingr   Zscipy.sparse.csgraphr   r   Zscipy.sparser   r   r   r   r'   r,   r   r   r   r   <module>   s   