
    SBg                     H    S SK JrJr  S SKrS SKrS SKJrJr   " S S5      r	g)    )assert_assert_array_equalN)	GeneratorMT19937c                       \ rS rSrS rS rS rS rS rS r	S r
S	 rS
 rS rS rS rS rS rS rS rS rS rS rSrg)TestRegression   c                 6    [        [        S5      5      U l        g )Nl   (vn4Ti )r   r   mt19937selfs    W/usr/lib/python3/dist-packages/numpy/random/tests/test_generator_mt19937_regressions.pysetup_methodTestRegression.setup_method	   s     )>!?@    c                 :   [         R                  " SSS5       H  nU R                  R                  USS5      n[	        [         R
                  " U[         R                  * :  5      =(       a'    [         R
                  " U[         R                  :*  5      5        M     g )Ng      g      @      2   )nplinspacer   vonmisesr   allpi)r   murs      r   test_vonmises_range"TestRegression.test_vonmises_range   sd     ++c2q)B%%b!R0ABFF1v:&=266!ruu*+=> *r   c                 B   [        [        R                  " U R                  R	                  SSSSS9S:  5      5        [        [        R                  " U R                  R	                  SSSSS9S:  5      5        Sn[        U R                  R                  " U6 S:  5        g )	N         
   size   r   ) r'   r'   )r   r   r   r   hypergeometric)r   argss     r   test_hypergeometric_range(TestRegression.test_hypergeometric_range   s    t||221b"22FJKLt||222q"22FJKL 1++T2Q67r   c                    SnU R                   R                  SUS9n[        R                  " US:H  5      U-  nSUS S3n[	        US:  U5        [        R                  " US	:H  5      U-  nSUS S
3n[	        US:  U5        g )N  g?r$   r   zFrequency was fz, should be > 0.45g?   z, should be < 0.23gq=
ףp?)r   	logseriesr   sumr   )r   Nrvsnfreqmsgs        r   test_logseries_convergence)TestRegression.test_logseries_convergence   s    ||%%c%2 vvdai 1$tAh&89tS!vvdai 1$tAh&89tS!r   c                 B   / SQ/ SQ/ SQ/ SQ4 H  n[        [        S5      5      n[        R                  " U[        S9nUR                  U5        [        R                  " US   US   US	   US
   /[        S9n[        [        R                  " U[        S9U5        M     g )N)r   r/   r    N)r   r   r/   r/   r    r    N)r   r:   r;   N)r9   r/   r    Ni90  dtyper/   r   r    r   )r   r   r   arrayobjectshuffler   )r   tr   shuffledexpecteds        r   test_shuffle_mixed_dimension+TestRegression.test_shuffle_mixed_dimension+   s    !0+&(A  /Gxx0HOOH%xx1qtQqT1Q4 8GHrxx?J(r   c           
         [         R                  " / SQ5      n[        S5       H]  n[        [	        U5      5      n[        [	        S5      5      n[        UR                  SS[         R                  " S5      S-  S9U5        M_     g )N)
r      r   r   r   r    r    rG   r   r&   r    i  r#   g      $@)r%   p)r   r>   ranger   r   r   choiceones)r   resir   ms        r   test_call_within_randomstate+TestRegression.test_call_within_randomstate7   s`    hh56qA
+G'$-(Aqxxrwwr{3xGM	 r   c                    U R                   R                  S/S//SS9  U R                   R                  S/S//[        R                  " S5      S9  U R                   R                  S/S//[        R                  " S5      S9  g )Nr   r   r$   )r   multivariate_normalr   int_int64r   s    r   #test_multivariate_normal_size_types2TestRegression.test_multivariate_normal_size_types@   sq     	((!se!(<((!se"''!*(E((!se"((1+(Fr   c                     U R                   R                  SSSS9n[        [        R                  " [        R
                  " U5      5      (       + S5        g )Ng-C6?d   r$   zNans in mt19937.beta)r   betar   r   anyisnan)r   xs     r   test_beta_small_parameters)TestRegression.test_beta_small_parametersH   s>     LLff37BFF288A;'')?@r   c                 <    U R                   R                  SS5        g )Ngn5gWw'&l7)r   rY   r   s    r   test_beta_very_small_parameters.TestRegression.test_beta_very_small_parametersN   s    %'r   c                     [         R                  " S5      R                  nU R                  R	                  US-  US-  SS9n[         R
                  " [         R                  " U5      5      (       a   eg )Ng      ?    (   r   r$   )r   finfotinyr   rY   rZ   r[   )r   rf   r\   s      r   'test_beta_ridiculously_small_parameters6TestRegression.test_beta_ridiculously_small_parametersR   sZ     xx}!!LLd2gtBwR866"((1+&&&&&r   c                     SnSnSnU R                   R                  XUS9n[        R                  " US:H  5      nSnSU-  Us=:  a
  SU-  :  d   e   eg )Ng{Gzd?i@B r$   r   gSis@gffffff?g?)r   rY   r   count_nonzero)r   abnr\   nzerosexpected_freqs          r   !test_beta_expected_zero_frequency0TestRegression.test_beta_expected_zero_frequencyY   sj     LLa+!!!q&) *M!F?T--??????r   c                    / SQn/ SQn[         R                  [         R                  [         R                  4 H  n[         R                  " X#S9[        U5      -  nU R                  R                  XS9n[        XQ;   5        [        R                  " [        5         U R                  R                  XS-  S9  S S S 5        M     g ! , (       d  f       M  = f)N)r   r/   r    )r&   r&   r/   r<   )rH   g?)r   float16float32float64r>   r1   r   rJ   r   pytestraises
ValueError)r   rk   countsdtprobscs         r   "test_choice_sum_of_probs_tolerance1TestRegression.test_choice_sum_of_probs_tolerancep   s     **bjj"**4BHHV.V<E##A#/AAFOz*##As#3 +*	 5 +*s   C
C	c                     [         R                  " SS/5      n[        S5       H  nU R                  R	                  U5        M      SS KnUR                  5         g )Nrk     aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarX   r   )r   r>   rI   r   r@   gccollectr   rk   _r   s       r   1test_shuffle_of_array_of_different_length_strings@TestRegression.test_shuffle_of_array_of_different_length_strings}   sE    
 HHc:&'sALL  #  	


r   c                    [         R                  " [         R                  " S5      [         R                  " S5      /[        S9n[	        S5       H  nU R
                  R                  U5        M      SS KnUR                  5         g )Nr   r&   r<   r-   r   )	r   r>   aranger?   rI   r   r@   r   r   r   s       r    test_shuffle_of_array_of_objects/TestRegression.test_shuffle_of_array_of_objects   sW     HHbiilBIIaL1@tALL  #  	


r   c                 x    " S S[         R                  5      n[        [        S5      5      n[         R                  " S5      R                  U5      nUR                  U5      n[        U[         R                  " / SQ5      5        [        U[         R                  " S5      R                  U5      5         " S S5      n[        [        S5      5      nU" 5       nUR                  U5      n[        U[         R                  " / SQ5      5        [        UR                  5       [         R                  " S	5      5        g )
Nc                       \ rS rSrSrg)3TestRegression.test_permutation_subclass.<locals>.N    N)__name__
__module____qualname____firstlineno____static_attributes__r   r   r   r2   r      s    r   r2   r   r    )r/   r   r   c                   B    \ rS rSr\R
                  " S5      rSS jrSrg)3TestRegression.test_permutation_subclass.<locals>.M   r   Nc                     U R                   $ )N)rk   )r   r=   copys      r   	__array__=TestRegression.test_permutation_subclass.<locals>.M.__array__   s    vvr   r   )NN)	r   r   r   r   r   r   rk   r   r   r   r   r   Mr      s    		!Ar   r   )r&   r   r    r   r/   r   )
r   ndarrayr   r   r   viewpermutationr   r>   r   )r   r2   r   origpermr   rN   s          r   test_permutation_subclass(TestRegression.test_permutation_subclass   s    	

 	 GAJ'yy|  #""4(4)!4541!2!21!56	 	 GAJ'C""1%4/!:;1;;="))A,7r   c                 &   U R                   R                  S5      S:X  d   e[        U R                   R                  S/5      S5        U R                   R                  S/SS9n[        R                  " S/[        R
                  S9n[        X5        g )Ng        floatr<   )r   standard_gammar   r   r>   rt   )r   actualrC   s      r   test_gamma_0TestRegression.test_gamma_0   ss    ||**3/36664<<66u=sC,,cU',B88RD

36,r   c                     [        U R                  R                  SSS9[        R                  " [        R
                  5      R                  5        g )NgKH9r    )rH   r%   )r   r   	geometricr   iinforT   maxr   s    r   test_geometric_tiny_prob'TestRegression.test_geometric_tiny_prob   s8     	4<<11E1B88BHH-11	3r   c                     SnU R                   R                  SUS9n[        U[        R                  " U[        R
                  S95        g )NrG   i'  r$   r<   )r   zipfr   r   rK   rT   r   rm   samples      r   test_zipf_large_parameter(TestRegression.test_zipf_large_parameter   s;     ""5q"162771BHH#=>r   c                     SnU R                   R                  SUS9n[        R                  " US:  5      US-  :  d   eg )Ni g    ?r$   l            r/   )r   r   r   rj   r   s      r   test_zipf_a_near_1!TestRegression.test_zipf_a_near_1   sE     ""?";
 /!A#555r   )r   N)r   r   r   r   r   r   r*   r6   rD   rO   rU   r]   r`   rg   rp   r}   r   r   r   r   r   r   r   r   r   r   r   r   r      sg    A?8"
KNGA('@.48.-3?	6r   r   )
numpy.testingr   r   numpyr   rv   numpy.randomr   r   r   r   r   r   <module>r      s    7   +G6 G6r   