U
    Af                     @   sD   d dl Z d dlmZmZmZmZ dd Zdd Zdd Zd	d
 Z	dS )    N)powerlaw_sequencerandom_weighted_sampleweighted_choicezipf_rvc                  C   s(   t ddd} t d} t| dks$td S )N
      seed)r   lenAssertionError)seq r   M/tmp/pip-unpacked-wheel-0wvcsx6h/networkx/utils/tests/test_random_sequence.pytest_degree_sequences   s    r   c                  C   sV   t dddd} t ddd} t d} t| s2ttttt d tjtt ddd d S )Ngffffff@   r   )xminr	   g      ?r   )r   )r   typer   intpytestraises
ValueError)rr   r   r   test_zipf_rv   s    r   c                  C   sN   ddd} t | ddd}t | d}t|t|  ks:tttt | d d S )Nr      abr   r   r      )r   sortedkeysr   r   r   r   )mappingsr   r   r   test_random_weighted_sample   s
    

r"   c                  C   s.   ddd} t | dd}t | }|dks*td S )Nr   r   r   r   r   r   )r   r   )r    cr   r   r   test_random_weighted_choice"   s    
r$   )
r   Znetworkx.utilsr   r   r   r   r   r   r"   r$   r   r   r   r   <module>   s
   	