a
    R¹§e5
  ã                   @   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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   Úint64T)	ÚnpÚarrayÚintr   r   r   ÚindicesZastypeZindptr)ÚAÚgraphÚpermÚcorrect_perm© r   ú^/var/www/sunrise/env/lib/python3.9/site-packages/scipy/sparse/csgraph/tests/test_reordering.pyÚ test_graph_reverse_cuthill_mckee   s&    
ùù	

r   c                  C   sb   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   Ztocsrr   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   