o
    ¢Äi	
  ć                   @   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   ©Zdtype)é   é   é   é   r   é   é   r   Zint64T)	ŚnpŚarrayŚintr   r   r   ŚindicesZastypeZindptr)ŚAŚgraphŚpermŚcorrect_perm© r   śd/home/kim/smarthome/.venv/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   Z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   Z	ones_liker   ŚT)r   r%   r&   r$   ŚBŚCr   r   r   Śtest_graph_structural_rank2   s   ž
’r,   )Śnumpyr   Znumpy.testingr   Zscipy.sparse.csgraphr   r   Zscipy.sparser   r   r   r   r'   r,   r   r   r   r   Ś<module>   s    