o
    3ήci                     @   sb   d dl Z e d d dlZd dlZd dlmZmZmZmZ dZ	dd Z
dd Zd	d
 Zdd ZdS )    Nnumpy)lattice_referenceomegarandom_referencesigma*   c                  C   s   t jdddtd} t| dtd}t | }t |}||ks J tt j ttt 	  W d    n1 s9w   Y  tt j
 ttt   W d    n1 sWw   Y  t 	d}t|dtd}d S )N2      皙?seed   niterr   )r   r   )      )nxconnected_watts_strogatz_graphrngr   average_clusteringpytestraisesNetworkXErrornextGraphNetworkXNotImplementedDiGraph)GGrCCrHHl r$   P/tmp/pip-target-vg8gfxp4/lib/python/networkx/algorithms/tests/test_smallworld.pytest_random_reference   s   


r&   c                  C   s   t jdddtd} t| dtd}t | }t |}||ks J tt jtt   tt j	tt 
  t d}t|dd}d S )Nr   r	   r   r   r   r   )r   )r   r   r   r   average_shortest_path_lengthr   r   r   r   r   r   )r   GlLLlr"   r#   r$   r$   r%   test_lattice_reference   s   


r+   c                  C   sT   t jdddtd} t jdddtd}t| ddtd}t|ddtd}||k s(J d S )Nr   r	   r
   r   r   r   r   nrandr   )r   r   r   r   )Gsr   sigmassigmarr$   r$   r%   
test_sigma+   s
   r1   c                  C   s   t jdddtd} t jdddtd}t jdddtd}t| ddtd}t|ddtd}t|ddtd}||k r;||k s=J t dd}t  }t |}t j|d	d
}	|||||	f}|D ]}
d|
  kridkslJ  J q\d S )Nr   r	   r   r   r   r
   r,      r   )r-   )r   r   r   r   barbell_graphkarate_club_graph)r(   r   r.   omegalomegaromegas	G_barbellG_karateomega_barbellomega_karateor$   r$   r%   
test_omega3   s   
r>   )r   importorskiprandomnetworkxr   r   r   r   r   r   r&   r+   r1   r>   r$   r$   r$   r%   <module>   s    
