o
    ˷e                     @   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 \/var/www/ideatree/venv/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   
MultiGraphr   r   r   r   r   r   r)   r   r   r   test_attr_matrix_multigraph+   s<   r0   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   onesfloatfill_diagonalr   r   todenser   )r   r   mtxr*   r   r   r   test_attr_sparse_matrixO   s   

r:   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 )Nr1   r   r   r   r   r   r	   r    r!   r#   r&   r'   )r2   r3   r   r(   r   r4   r   r   r   r   r8   r)   r   r   r    test_attr_sparse_matrix_directed]   s   
r;   )
r2   r3   r   networkxr   r   r+   r0   r:   r;   r   r   r   r   <module>   s    
$