
    h                     ,    S r SSKrSSKr " S S5      rg)z@Unit tests for the :mod:`networkx.generators.stochastic` module.    Nc                   B    \ rS rSrSrS rS rS rS rS r	S r
S	 rS
rg)TestStochasticGraph   z?Unit tests for the :func:`~networkx.stochastic_graph` function.c                 0   [         R                  " 5       nUR                  SS5        UR                  SS5        [         R                  " U5      n[         R                  " X5      (       d   e[        UR                  SS95      SSSS04SSSS04/:X  d   eg )Nr         Tdataweight      ?)nxDiGraphadd_edgestochastic_graphis_isomorphicsortededgesselfGSs      [/var/www/html/env/lib/python3.13/site-packages/networkx/generators/tests/test_stochastic.pytest_default_weights(TestStochasticGraph.test_default_weights   s    JJL	

1a	

1a"%%%%agg4g()Hc?#Hc?#.
 
 	
 
    c                     [         R                  " 5       nUR                  SSSS9  UR                  SSSS9  [         R                  " USS9  [	        UR                  SS95      SSS	S
04SSS	S
04/:X  d   eg)z<Tests for an in-place reweighting of the edges of the graph.r   r   r   r   F)copyTr	   r   r   Nr   r   r   r   r   r   )r   r   s     r   test_in_place!TestStochasticGraph.test_in_place   s    JJL	

1a
"	

1a
"
AE*agg4g()Hc?#Hc?#.
 
 	
 
r   c                     [         R                  " 5       nUR                  SSSS9  UR                  SSSS9  [         R                  " U5      n[	        UR                  SS95      SSSS04SSSS04/:X  d   eg )	Nr   r   r   r   Tr	   r   r   r   r   s      r   test_arbitrary_weights*TestStochasticGraph.test_arbitrary_weights!   s    JJL	

1a
"	

1a
""agg4g()Hc?#Hc?#.
 
 	
 
r   c                     [         R                  " 5       nUR                  / SQ5        [         R                  " U5      nSS0n[	        UR                  SS95      SSU4SSU4SSU4SSU4/:X  d   eg )	N)r   r   r&   r   r   r'   r   g      ?Tr	   r   r   r   )r   MultiDiGraphadd_edges_fromr   r   r   )r   r   r   ds       r   test_multidigraph%TestStochasticGraph.test_multidigraph+   s    OO	9:"tagg4g()1I1I1I1I	.
 
 	
 
r   c                     [         R                  " 5       nUR                  SSSS9  UR                  SSSS9  [         R                  " U5      n[	        UR                  SS95      SSSS04SSSS04/:X  d   eg)	z3Smoke test: ensure ZeroDivisionError is not raised.r   r   r   r   Tr	   r   Nr   r   s      r   test_zero_weights%TestStochasticGraph.test_zero_weights7   s    JJL	

1a
"	

1a
""agg4g()Ha=!Ha=!.
 
 	
 
r   c                     [         R                  " [        R                  5         [        R                  " [        R
                  " 5       5        S S S 5        g ! , (       d  f       g = fN)pytestraisesr   NetworkXNotImplementedr   Graphr   s    r   test_graph_disallowed)TestStochasticGraph.test_graph_disallowedB   s4    ]]2445
+ 655   *A
A&c                     [         R                  " [        R                  5         [        R                  " [        R
                  " 5       5        S S S 5        g ! , (       d  f       g = fr1   )r2   r3   r   r4   r   
MultiGraphr6   s    r   test_multigraph_disallowed.TestStochasticGraph.test_multigraph_disallowedF   s4    ]]24450 655r9    N)__name__
__module____qualname____firstlineno____doc__r   r    r#   r+   r.   r7   r<   __static_attributes__r>   r   r   r   r      s(    I	
	



	
,1r   r   )rC   r2   networkxr   r   r>   r   r   <module>rF      s    F  @1 @1r   