o
    	ńh	
  ć                   @   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_arrayŚ	csr_arrayŚ	coo_arrayc               
   C   s   t jg d¢g d¢g d¢g d¢g d¢g d¢g d¢g d¢gtd} t| }t|}t  g d¢”}t|| |j d	”|_|j d	”|_t|d
}t|| d S )N)é   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   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   ©Śdtype)é   é   é   é   r   é   é   r   Śint64T)	ŚnpŚarrayŚintr   r   r   ŚindicesŚastypeŚindptr)ŚAŚgraphŚpermŚcorrect_perm© r   ś^/var/www/vscode/kcb/lib/python3.10/site-packages/scipy/sparse/csgraph/tests/test_reordering.pyŚ test_graph_reverse_cuthill_mckee   s&   
łł	

r   c                  C   s^   t jdtd} t  g d¢”}t  g d¢”}t| ||ff}t|}t  g d¢”}t|| d S )Né?   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   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   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!   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"   r    r   r'   r   r%   r   r   r!   r#   r   r   )r   Śonesr   r   r   r   r   )ŚdataŚrowsŚcolsr   r   r   r   r   r   Ś)test_graph_reverse_cuthill_mckee_ordering   s   r,   c                  C   s¦   t g d¢g d¢g d¢g} tt| d t g d¢”}t g d¢”}t |”}t|||ffdd}tt|d	 t g d
¢g d¢g}tt|d tt|jd d S )N)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   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    )Śshaper   )r   r   r   r   )r   r   r   r   r   )r   r   r   r   r   Ś	ones_liker   ŚT)r   r*   r+   r)   ŚBŚCr   r   r   Śtest_graph_structural_rank2   s   ž
’r2   )Śnumpyr   Śnumpy.testingr   Śscipy.sparse.csgraphr   r   Śscipy.sparser   r   r   r   r,   r2   r   r   r   r   Ś<module>   s    