o
    3ήc
                     @   s\   d dl Z d dlZdd Zdd Zdd Zdd	 Zd
d Zdd Zdd Z	dd Z
dd ZdS )    Nc                  C   sT   t g d} t jj| dd}|dddksJ t jj| ddd }|dks(J d S )Nr      r      )r   r   )r      )r      r      F
normalizedg?gUUUUUU?r   r   nxGraphrichclubrich_club_coefficient)Grcrc0 r   N/tmp/pip-target-vg8gfxp4/lib/python/networkx/algorithms/tests/test_richclub.pytest_richclub   s
   r   c                  C   s6   t g d} t jj| ddd}|dddksJ d S )Nr   r   r   )Qseed      ?r   r   r   rcNormr   r   r   test_richclub_seed   s   r   c                  C   s4   t g d} t jj| dd}|dddksJ d S )Nr   r   )r   r   r   r   r   r   r   r   test_richclub_normalized   s   r   c                  C   s4   t dd} t jj| dd}|ddddksJ d S )	Nr   
   Fr   g @ P?g@ `?g2<`?)r   r   r   )r   balanced_treer   r   )Tr   r   r   r   test_richclub2   s   r"   c                  C   s   t  } t j| dd}|i ddddddd	d
ddddddddddddddddddddddddksAJ d S )NFr   r   g)?r   g?r   gyy?r   g?r   gII?r
   g10?         ?      	   r   gUUUUUU?                       )r   karate_club_graphr   r   r   r   r   r   test_richclub3&   sF   	
r0   c                  C   s>   t  } | g d t j| dd}|dddddksJ d S )N)	r   r   )r   r   )r   r   r	   )r
   r'   )r#   r'   )r%   r'   )r&   r'   Fr   g?r$   r*   )r   r   r   r   )r   r   add_edges_fromr   r/   r   r   r   test_richclub4>   s   r2   c                  C   D   t tj t } t|  W d    d S 1 sw   Y  d S N)pytestraisesr   NetworkXNotImplementedDiGraphr   r   r   r   r   test_richclub_exceptionG      "r:   c                  C   r3   r4   )r5   r6   r   r7   
MultiGraphr   r9   r   r   r   test_rich_club_exception2M   r;   r=   c                  C   s^   t  } | dd | dd tjtdd t |  W d    d S 1 s(w   Y  d S )Nr   r   zDrich_club_coefficient is not implemented for graphs with self loops.)match)r   r   add_edger5   r6   	Exceptionr   r9   r   r   r   test_rich_club_selfloopS   s   "rA   )r5   networkxr   r   r   r   r"   r0   r2   r:   r=   rA   r   r   r   r   <module>   s    

	