
    fiz                     X    d dl Z d dlZd Zd Zd Zd Zd Zd Zd Z	d	 Z
d
 Zd Zd ZdS )    Nc                      t          j        g d          } t           j                            | d          }|dddk    sJ t           j                            | d          d         }|dk    sJ d S )Nr      r      )r   r   )r      )r      r
      F
normalizedg?UUUUUU?r   r   nxGraphrichclubrich_club_coefficient)Grcrc0s      /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/networkx/algorithms/tests/test_richclub.pytest_richclubr      s    
AAABBA		*	*1	*	?	?BY8,,,,,, +
+
+A%
+
@
@
CC+    c                      t          j        g d          } t           j                            | dd          }|dddk    sJ d S )Nr   r   r   Qseed      ?r   r   r   rcNorms     r   test_richclub_seedr"      sS    
AAABBA[..qAA.>>F%%%%%%%%r   c                      t          j        g d          } t           j                            | dd          }|dddk    sJ d S )Nr   r   *   r   r   r   r   r    s     r   test_richclub_normalizedr%      sS    
AAABBA[..qAB.??F%%%%%%%%r   c                      t          j        dd          } t           j                            | d          }|ddddk    sJ d S )	Nr   
   Fr   g @ P?g@ `?g2<`?r   r   r   )r   balanced_treer   r   )Tr   s     r   test_richclub2r+      sb    
BA		*	*1	*	?	?B!""       r   c                      t          j                    } t          j        | 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    sJ d S )NFr   r   g)?r   g?r   gyy?r	   g?r
   gII?r   g10?         ?      	   r'   gUUUUUU?                       )r   karate_club_graphr   r   r   s     r   test_richclub3r:   &   s   
A		!!	6	6	6B 	<	< 	
; 	
:	
 	
9 	
9 	
9 	
9 	
9 	
8 	G 	G 	C 	C 	C  	C!      r   c                      t          j                    } |                     g d           t          j        | d          }|dddddk    sJ d S )N)	r   r   )r   r	   )r   r
   r   )r   r1   )r-   r1   )r/   r1   )r0   r1   Fr   g?r.   r4   )r   r   r   r	   )r   r   add_edges_fromr   r9   s     r   test_richclub4r=   >   sh    


APPP   
	!!	6	6	6BY8<<<<<<<<r   c                      t          j        t          j                  5  t          j                    } t          j        |            d d d            d S # 1 swxY w Y   d S N)pytestraisesr   NetworkXNotImplementedDiGraphr   r   s    r   test_richclub_exceptionrE   G   s    	r0	1	1 $ $JLL
 ###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $   (AAAc                      t          j        t          j                  5  t          j                    } t          j        |            d d d            d S # 1 swxY w Y   d S r?   )r@   rA   r   rB   
MultiGraphr   rD   s    r   test_rich_club_exception2rI   M   s    	r0	1	1 $ $MOO
 ###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $rF   c                     t          j                    } |                     dd           |                     dd           t          j        t
          d          5  t          j        |            d d d            d S # 1 swxY w Y   d S )Nr   r   zDrich_club_coefficient is not implemented for graphs with self loops.match)r   r   add_edger@   rA   	Exceptionr   rD   s    r   test_rich_club_selflooprO   S   s    


AJJq!JJq!	T
 
 
 $ $ 	 ###	$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s   A==BBc                     t          j                    } t          j        | d          }|i k    sJ t          d          D ]5}|                     |           t          j        | d          }|i k    sJ 6t          j                    } |                     dd           t          j        | d          }|ddik    sJ t          j                    } |                     g d           |                     dd           t          j        | d          }|ddik    sJ |                     dd           t          j        | d          }|ddik    sJ |                     dd           t          j        | d          }|ddd	k    sJ d S )
NFr   r	   r   r   r(   r   r   r   )r   r   r   rangeadd_noderM   add_nodes_fromr   r   is      r   'test_rich_club_leq_3_nodes_unnormalizedrV   ^   s   


A		!!	6	6	6B88881XX  	

1%aE:::Rxxxxx 	

AJJq!		!!	6	6	6B!Q<<<< 	

AYYYJJq!		!!	6	6	6B!Q<<<< JJq!		!!	6	6	6B!U JJq!		!!	6	6	6BQ1r   c                     t          j                    } t          j        t           j        j        d          5  t          j        | d          }d d d            n# 1 swxY w Y   t          d          D ]j}|                     |           t          j        t           j        j        d          5  t          j        | d          }d d d            n# 1 swxY w Y   kd S )NzGraph has fewer than four nodesrK   Tr   r	   )	r   r   r@   rA   	exceptionNetworkXErrorr   rQ   rR   rT   s      r   %test_rich_club_leq_3_nodes_normalizedrZ      sZ   


A	
"/
 
 
 : : %aD999	: : : : : : : : : : : : : : : 1XX > >	

1]L&3
 
 
 	> 	> )!===B		> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	> 	>> >s#   AA #A 2CC	C	)r@   networkxr   r   r"   r%   r+   r:   r=   rE   rI   rO   rV   rZ    r   r   <module>r]      s          & & && & &    0= = =$ $ $$ $ $$ $ $     F> > > > >r   