o
    ]ZŽh  ã                   @   sF   d dl Z e  d¡Zd dlZdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Z	dS )é    NÚnumpyc                     s¦   t  ¡ ‰ ˆ jddddd ˆ jddddd ˆ jdddd ˆ jdddd ‡ fdd„} ‡ fd	d
„}t jˆ || d}tj |d t dgg¡¡ |d dgksQJ ‚d S )Nr   é   é   ©Ú	thicknessÚweighté   ©r   c                    s   ˆ j |   dd¡d S )NÚsizeç      à?r   )ÚnodesÚget)Úu©ÚG© úT/var/www/auris/lib/python3.10/site-packages/networkx/linalg/tests/test_attrmatrix.pyÚ	node_attr   s   z#test_attr_matrix.<locals>.node_attrc                    s   ˆ |  |   dd¡S )Nr   r   )r   )r   Úvr   r   r   Ú	edge_attr   s   z#test_attr_matrix.<locals>.edge_attr)r   r   g      @g      ø?)ÚnxÚGraphÚadd_edgeÚattr_matrixÚnpÚtestingÚassert_equalÚarray)r   r   ÚMr   r   r   Útest_attr_matrix   s   r   c                  C   s’   t  ¡ } | jddddd | jddddd | jdddd | jdddd t j| g d¢d}t g d	¢g d
¢g d¢g¡}tj |t |¡¡ d S )Nr   r   r   r   r   r	   ©r   r   r   ©Úrc_order©ç        ç      ð?r%   ©r$   r$   r%   ©r$   r$   r$   )r   ÚDiGraphr   r   r   r   r   r   ©r   r   Údatar   r   r   Útest_attr_matrix_directed   s   þÿr+   c                  C   s,  t  ¡ } | jddddd | jddddd | jddddd | jdddd | jdddd t j| g d¢d}t g d	¢g d
¢g d¢g¡}tj |t |¡¡ t j| dg d¢d}t g d¢g d¢g d¢g¡}tj |t |¡¡ t j| dg d¢d}t g d¢g d¢g d¢g¡}tj |t |¡¡ d S )Nr   r   r   r   r   r	   r    r!   )r$   ç      @r%   )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
MultiGraphr   r   r   r   r   r   r)   r   r   r   Útest_attr_matrix_multigraph+   s<   þÿþÿþÿr/   c                  C   sž   t  d¡ t ¡ } | jddddd | jdddd | jdddd t | ¡}|d }t dt¡}t 	|d¡ tj
 | ¡ t |¡¡ |d g d	¢ksMJ ‚d S )
NÚscipyr   r   r   r   r   r	   )r   r   r    )ÚpytestÚimportorskipr   r   r   Úattr_sparse_matrixr   ZonesÚfloatZfill_diagonalr   r   Útodenser   )r   r   Zmtxr*   r   r   r   Útest_attr_sparse_matrixO   s   

r6   c                  C   s    t  d¡ t ¡ } | jddddd | jddddd | jdddd | jdddd tj| g d¢d	}t g d
¢g d¢g d¢g¡}tj 	| 
¡ t |¡¡ d S )Nr0   r   r   r   r   r   r	   r    r!   r#   r&   r'   )r1   r2   r   r(   r   r3   r   r   r   r   r5   r)   r   r   r   Ú test_attr_sparse_matrix_directed]   s   
þÿr7   )
r1   r2   r   Znetworkxr   r   r+   r/   r6   r7   r   r   r   r   Ú<module>   s    
$