
    fi                     x    d dl Z d dlZd Zd Zd Zd Zd Z G d d          Zd	 Z	d
 Z
d Zd Zd Zd Zd ZdS )    Nc                      d} d}t          d          D ]e}t          j        | |          }d |                                D             }t          j        |d          sJ t          j        |d          sJ fd S )Nd   g333333?
   c              3       K   | ]	\  }}|V  
d S N .0nds      /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/networkx/algorithms/tests/test_graphical.py	<genexpr>z.test_valid_degree_sequence1.<locals>.<genexpr>   &      ((TQq((((((    egmethodhh)rangenxerdos_renyi_graphdegreeis_graphical)r   piGdegs        r   test_valid_degree_sequence1r      s    AA2YY 1 1 A&&((QXXZZ(((s4000000s40000000	1 1r   c                      d} t          d          D ]e}t          j        | d          }d |                                D             }t          j        |d          sJ t          j        |d          sJ fd S )Nr   r      c              3       K   | ]	\  }}|V  
d S r   r   r	   s      r   r   z.test_valid_degree_sequence2.<locals>.<genexpr>   r   r   r   r   r   )r   r   barabasi_albert_graphr   r   )r   r   r   r   s       r   test_valid_degree_sequence2r#      s    A2YY 1 1$Q**((QXXZZ(((s4000000s40000000	1 1r   c                     t          j        t          j        t          j        g d           t          j        t          j        t          j        dgd           t          j        t          j        t          j        dgd           d S )Nfooredr   r   pytestraisesr   NetworkXExceptionr   r   r   r   test_string_inputr+      s]    
M"&UCCC
M"&%$GGG
M"&%$GGGGGr   c                      t          j        t          j        t          j        dgd           t          j        t          j        t          j        dgd           d S )Ng      R@r   r   r'   r   r   r   test_non_integer_inputr-      sB    
M"&$FFF
M"&$FFFFFr   c                  f    t          j        dgd          rJ t          j        dgd          rJ d S )Nr   r   r   r   r   r   r   test_negative_inputr1   $   s<    tT*****tT*******r   c                   *    e Zd Zed             Zd ZdS )	TestAtlasc                 N    ddl ma t                                          | _        d S )Nr   )atlas)networkx.generatorsr5   graph_atlas_gGAG)clss    r   setup_classzTestAtlas.setup_class*   s-     	.-----%%''r   c                     | j         D ]P}d |                                D             }t          j        |d          sJ t          j        |d          sJ Qd S )Nc              3       K   | ]	\  }}|V  
d S r   r   r	   s      r   r   z'TestAtlas.test_atlas.<locals>.<genexpr>3   s&      00A1000000r   r   r   r   )r8   r   r   r   )selfgraphr   s      r   
test_atlaszTestAtlas.test_atlas1   sq    X 	5 	5E00000C?3t444444?3t4444444	5 	5r   N)__name__
__module____qualname__classmethodr:   r?   r   r   r   r3   r3   )   s<        ( ( [(5 5 5 5 5r   r3   c                  >   g d} t          j        | d          sJ t          j        | d          sJ g d} t          j        | d          sJ t          j        | d          sJ g d} t          j        | d          sJ t          j        | d          sJ d S )N      rG   rG   rG      rH   rH   r    r    r    r   r   r   )r   rG   rG   rG   rG   rH   rH   rH   rH   rH   rH   
r    r    r    r    r    rH   rH   rH   rG      r0   zs    r   test_small_graph_truerM   8   s    )))A?1T******?1T*********A?1T******?1T******&&&A?1T******?1T********r   c                  >   g d} t          j        | d          rJ t          j        | d          rJ g d} t          j        | d          rJ t          j        | d          rJ g d} t          j        | d          rJ t          j        | d          rJ d S )N  rG   rG   rG   rG   rH   rH   rH   r    r    r    r   r   r      rF   rJ   rJ   rH   r    r    r    r    r    r    r    r    r    rH   rH   rH   rG   rJ   r0   rK   s    r   test_small_graph_falserT   D   s    ,,,Aq......q......   Aq......q......)))Aq......q........r   c                     d\  } }d|z  }t          |          D ]r}t          j        | ||dz   z  d d          }d |                                D             }d |                                D             }t          j        ||          sJ sd S )N)r   r   g      ?r    Tc              3       K   | ]	\  }}|V  
d S r   r   r	   s      r   r   z0test_directed_degree_sequence.<locals>.<genexpr>V   s&      ++TQq++++++r   c              3       K   | ]	\  }}|V  
d S r   r   r	   s      r   r   z0test_directed_degree_sequence.<locals>.<genexpr>W   s&      --da------r   )r   r   r   	in_degree
out_degreeis_digraphical)r   rr   r   r   dindouts          r   test_directed_degree_sequencer^   P   s    DAqaA1XX , , AQKt<<++Q[[]]+++--allnn--- d++++++	, ,r   c                     g d} g d}t          j        ||           sJ g d} g d}t          j        ||           rJ g d} g d}t          j        ||           sJ g d}t          j        ||           rJ g d}t          j        ||           rJ g d	x}} t          j        ||           rJ g d
x}} t          j        ||           rJ d S )NrE   )rG   rG   rG   rG   rG   rH   rH   rH   rH   rH   r    rO   )
g   f   ra   ra   ra   ra   ra   ra   ra   ra   rI   )
rH   rH   rH   rH   rH   rH   rH   rH   r    r    )rH   rH   rH   rH   rH   rH   rH   rH   r    r    r    )rH   rH   rH   rH   rH   rH   rH   r    r    rJ   r    r    皙?r    r    r    rerr    )r   rZ   )r]   r\   s     r   test_small_directed_sequencesrg   [   s-   ,,,D
+
+
+CS$'''''///D
<
<
<C d+++++)))D
(
(
(CS$'''''
+
+
+C d+++++
,
,
,C d+++++C$ d+++++!!!!C$ d+++++++r   c                  >   g d} t          j        |           rJ g d} t          j        |           sJ g d} t          j        |           rJ g d} t          j        |           rJ g d} t          j        |           rJ g d} t          j        |           rJ d S )N
rP   rG   rG   rG   rG   rH   rH   rH   r    r    rQ   )rR   rF   rJ   rH   r    r    r    rS   rc   re   )r   is_multigraphicalseqs    r   test_multi_sequencern   t   s    
+
+
+C#C(((((
"
"
"C$$$$$
#
#
#C#C(((((
+
+
+C#C(((((
..C#C(((((


C#C(((((((r   c                  
   g d} t          j        |           sJ g d} t          j        |           rJ g d} t          j        |           rJ g d} t          j        |           rJ g d} t          j        |           rJ d S )Nri   rO   )
rP   rG   rG   rG   rG   rH   rH   rb   r    r    rc   re   )r   is_pseudographicalrl   s    r   test_pseudo_sequencerq      s    
+
+
+C %%%%%
.
.
.C$S)))))
,
,
,C$S)))))
..C$S)))))


C$S)))))))r   c                  F   t          j        d          } |                     g d| j                  }t	          j        |d          sJ t	          j        |d          sJ |                     g d| j                  }t	          j        |d          sJ t	          j        |d          sJ |                     g d| j                  }t          j        t          j        t          j        |d           t          j        t          j        t          j        |d           d S )Nnumpy)r    rH   rH   rH   r    )dtyper   r   )rd   rH   rH   rH   r    )	r(   importorskiparrayint64r   r   float64r)   r*   )npdss     r   test_numpy_degree_sequencer{      s   		W	%	%B	///	2	2B?2t$$$$$?2t$$$$$	///	4	4B?2t$$$$$?2t$$$$$	###2:	6	6B
M"&TBBB
M"&TBBBBBr   )r(   networkxr   r   r#   r+   r-   r1   r3   rM   rT   r^   rg   rn   rq   r{   r   r   r   <module>r}      s       1 1 11 1 1H H HG G G
+ + +
5 5 5 5 5 5 5 5	+ 	+ 	+	/ 	/ 	/, , ,, , ,2) ) )(* * *"
C 
C 
C 
C 
Cr   