3
Od5
                 @   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 jddddddd	dg}t|| |jjd
|_|jjd
|_t|d}t|| d S )N   r   )dtype                  Zint64T)	nparrayintr   r   r   indicesZastypeZindptr)Agraphpermcorrect_perm r   ^/var/www/html/virt/lib64/python3.6/site-packages/scipy/sparse/csgraph/tests/test_reordering.py test_graph_reverse_cuthill_mckee   s     

r   c           @   C   sr  t jdtd} t j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 j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j }t|}t j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j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j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j|}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   )shape)r   r   )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,   )Znumpyr   Znumpy.testingr   Zscipy.sparse.csgraphr   r   Zscipy.sparser   r   r   r   r'   r,   r   r   r   r   <module>   s   