
    ܛ7i              
           d dl Z d dlZd Ze j                  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
y)    Nc                      t         j                  j                  dd      } t         j                  j                  d      } t        |       dk(  sJ y )N
      seed)nxutilspowerlaw_sequencelen)seqs    f/home/rose/Desktop/poly/venv/lib/python3.12/site-packages/networkx/utils/tests/test_random_sequence.pytest_degree_sequencesr      s?    
((
$
$Ra
$
0C
((
$
$R
(Cs8r>>    )deg_seqvalidreasonFzmust have one more nodeT    zmust have strictly positive)   r   r   r   c                 b    t         j                  j                  |       \  }}||k(  sJ ||v sJ y )N)r   r	   is_valid_tree_degree_sequence)r   r   r   vrs        r   test_valid_degree_sequencer      s4     8811':DAq::Q;;r   c                     t         j                  j                  ddd      } t         j                  j                  ddd      } t         j                  j                  d      } t        |       sJ t               t        j                  t        t         j                  j                  d       t        j                  t        t         j                  j                  dd       y )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7C7
MM*bhh..4
MM*bhh..:r   c                  H   ddd} t         j                  j                  | dd      }t         j                  j                  | d      }t        |      t        | j	                               k(  sJ t        j                  t        t         j                  j                  | d       y )Nr      abr   r   r   r   )r   r	   random_weighted_samplesortedkeysr    r!   r"   )mappingss     r   test_random_weighted_sampler.   %   sv    R G
''';A
''3A!9w||~....
MM*bhh==wJr   c                      ddd} t         j                  j                  | d      }t         j                  j                  |       }|dk(  sJ y )Nr   r   r&   r   r   r'   )r   r	   weighted_choice)r,   cs     r   test_random_weighted_choicer2   -   sD    QG
  q 1A
  )A8O8r   c                  Z    t         j                  j                  dgdz        d   dk(  sJ y )Ng?d   g      ?)r   r	   cumulative_distribution r   r   "test_random_sequence_low_precisionr8   4   s*    88++SECK8<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Dr   