
    Pi              
           d dl Z d dlZd Ze j                            dg ddfd gddfdgddfdd gdd	fg d
ddfg          d             Zd Zd Zd Z	d Z
dS )    Nc                      t           j                            dd          } t           j                            d          } t          |           dk    sJ d S )N
      seed)nxutilspowerlaw_sequencelen)seqs    }/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/networkx/utils/tests/test_random_sequence.pytest_degree_sequencesr      sH    
(
$
$Ra
$
0
0C
(
$
$R
(
(Cs88r>>>>>>    )deg_seqvalidreasonFzmust have one more nodeT    zmust have strictly positive)   r   r   r   c                 f    t           j                            |           \  }}||k    sJ ||v sJ d S )N)r   r	   is_valid_tree_degree_sequence)r   r   r   vrs        r   test_valid_degree_sequencer      s;     811'::DAq::::Q;;;;;;r   c                     t           j                            ddd          } t           j                            ddd          } t           j                            d          } t          |           sJ t                      t          j        t          t           j        j        d           t          j        t          t           j        j        dd           d S )Ngffffff@r   r   )xminr   g      ?r   )r   )r   r	   zipf_rvtypeintpytestraises
ValueError)r   s    r   test_zipf_rvr#      s    
11--A
a##A
A77C7
M*bh.444
M*bh.::::::r   c                  R   ddd} t           j                            | dd          }t           j                            | d          }t          |          t          |                                           k    sJ t          j        t          t           j        j        | d           d S )Nr      abr   r   r   r   )r   r	   random_weighted_samplesortedkeysr    r!   r"   )mappingss     r   test_random_weighted_sampler.   %   s    R  G
''';;A
''33A!99w||~~......
M*bh=wJJJJJr   c                      ddd} t           j                            | d          }t           j                            |           }|dk    sJ d S )Nr   r   r&   r   r   r'   )r   r	   weighted_choice)r,   cs     r   test_random_weighted_choicer2   -   sO    QG
  q 11A
  ))A888888r   c                  d    t           j                            dgdz            d         dk    sJ d S )Ng?d   g      ?)r   r	   cumulative_distribution r   r   "test_random_sequence_low_precisionr8   4   s4    8++SECK88<CCCCCCr   )r    networkxr   r   markparametrizer   r#   r.   r2   r8   r7   r   r   <module>r<      s           "	U-.
dB
e./
Q56	tR 	 	 	 	; ; ;K K K  D D D D Dr   