U
    ?h
                     @   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   s\   t ddddddg} t jj| dd}|d	d
dks8tt jj| ddd }|d	ksXtd S )Nr      r      r   r   r      r      r
      F
normalizedg?gUUUUUU?r   nxGraphrichclubrich_club_coefficientAssertionError)GrcZrc0 r   Y/var/www/html/venv/lib/python3.8/site-packages/networkx/algorithms/tests/test_richclub.pytest_richclub   s
    r   c                  C   s>   t ddddddg} t jj| ddd	}|d
d
dks:td S )Nr   r   r   r   r	   r   r   r   )Qseed      ?r   r   ZrcNormr   r   r   test_richclub_seed   s    r   c                  C   s<   t ddddddg} t jj| dd}|d	d	dks8td S )
Nr   r   r   r   r	   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s0td S )	Nr   
   Fr   g @ P?g@ `?g2<`?)r   r   r   )r   Zbalanced_treer   r   r   )Tr   r   r   r   test_richclub2   s    r"   c                  C   sH   t  } t j| dd}|ddddddd	d	d	d	d
d
dddddksDtd S )NFr   g)?g?gyy?g?gII?g10?      ?gUUUUUU?        )r   r   r   r   r
   r            	   r                   )r   Zkarate_club_graphr   r   r   r   r   r   r   test_richclub3&   s(    r/   c                  C   sL   t  } | ddddddddd	g	 t j| d
d}|dddddksHtd S )Nr   r   )r   r   )r   r
   r   )r   r(   )r%   r(   )r&   r(   )r'   r(   Fr   g?r#   r$   )r   r   r   r   )r   r   Zadd_edges_fromr   r   r.   r   r   r   test_richclub4>   s    r0   c               	   C   s.   t tj t } t|  W 5 Q R X d S N)pytestraisesr   NetworkXNotImplementedZDiGraphr   r   r   r   r   test_richclub_exceptionG   s    r6   c               	   C   s.   t tj t } t|  W 5 Q R X d S r1   )r2   r3   r   r4   Z
MultiGraphr   r5   r   r   r   test_rich_club_exception2M   s    r7   c               	   C   sH   t  } | dd | dd tjtdd t |  W 5 Q R X d S )Nr   r   zDrich_club_coefficient is not implemented for graphs with self loops.)match)r   r   Zadd_edger2   r3   	Exceptionr   r5   r   r   r   test_rich_club_selfloopS   s    r:   )r2   Znetworkxr   r   r   r   r"   r/   r0   r6   r7   r:   r   r   r   r   <module>   s   

	