U
    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 )    NZ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std S )Nr         	thicknessweight   r   c                    s    j |  ddd S )Nsize      ?r   )Znodesget)uG I/tmp/pip-unpacked-wheel-j2te_plx/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AssertionError)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| dddgd}tdd	d	gddd	gdddgg}tj|t| d S )
Nr   r   r   r   r   r   rc_order              ?)r   DiGraphr   r   r   r   r   r   r   r   datar   r   r   test_attr_matrix_directed   s    r&   c                  C   sD  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| dddgd}tdd	d
gd	dd
gd
d
dgg}tj|t| t j| ddddgd}tddd
gddd
gd
d
dgg}tj|t| t j| ddddgd}tdd	dgd	dd	gdd	dgg}tj|t| d S )Nr   r   r   r   r   r   r   r!   g      @r"   r   )r   r    g      "@r   g       @)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 dddgkstd S )	Nscipyr   r   r   r   r   r   )r   r   )pytestimportorskipr   r   r   attr_sparse_matrixr   ZonesfloatZfill_diagonalr   r   todenser   r   )r   r   Z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| dddgd}td	d
d
gd	d	d
gd	d	d	g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    test_attr_sparse_matrix_directed]   s    
r/   )
r)   r*   r   Znetworkxr   r   r&   r'   r.   r/   r   r   r   r   <module>   s   
$