
    Kh?                     \    S SK r S SKrS SKJrJrJr  S SKrS SKJr   " S S5      r	S r
S rg)    N)assert_assert_array_equalassert_raises)randomc                       \ 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\R&                  R)                  \R,                  " S5      R.                  S:  SS9S 5       rS rS rSrg)TestRegression   c                 0   [         R                  " SSS5       Hz  n[        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)selfmurs      `/var/www/html/env/lib/python3.13/site-packages/numpy/random/tests/test_randomstate_regression.pytest_VonMises_range"TestRegression.test_VonMises_range   s^     ++c2q)BAr*ABFF1v:&=266!ruu*+=> *    c                    [        [        R                  " [        R                  " SSSSS9S:  5      5        [        [        R                  " [        R                  " SSSSS9S:  5      5        S/n[
        R                  S	:  nU(       a%  [
        R                  S
:w  a  UR                  S5        U H"  n[        [        R                  " U6 S:  5        M$     g )N         
   size   r   ) r"   r"           win32)   r%   r%   )	r   r   r   r   hypergeometricsysmaxsizeplatformappend)r   args	is_64bitsargs       r   test_hypergeometric_range(TestRegression.test_hypergeometric_range   s    v,,QBR@1DEFv,,RBR@1DEF .
 KK%'	0KK9:CF))3/!34 r   c                 &   Sn[         R                  " S5        [         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  r   g?r   r   zFrequency was fz, should be > 0.45g?   z, should be < 0.23gq=
ףp?)r   seed	logseriesr   sumr   )r   Nrvsnfreqmsgs        r   test_logseries_convergence)TestRegression.test_logseries_convergence&   s    A!, vvdai 1$tAh&89tS!vvdai 1$tAh&89tS!r   c                 4   / SQ/ SQ/ SQ/ SQ4 H  n[         R                  " S5        [        U5      n[         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   r3   r   N)r   r   r3   r3   r   r   N)r   r?   r@   N)r>   r3   r   N90  r   r   r   r3   dtype)r   r4   listshuffler   arrayobjectr   )r   tshuffledexpecteds       r   test_shuffle_mixed_dimension+TestRegression.test_shuffle_mixed_dimension6   s{    !0+&(A KKAwHNN8$xx1qtQqT1Q4 8GHrxx?J(r   c           
      4   [         R                  " 5       n[        R                  " / SQ5      n[	        S5       H\  n[         R
                  " U5        UR                  S5        [        UR                  SS[        R                  " S5      S-  S9U5        M^     g )N)
r         r3   r   	   r!   rO   r   r   r   i  r   g      $@)r    p)	r   RandomStater   rF   ranger4   r   choiceones)r   mresis       r   test_call_within_randomstate+TestRegression.test_call_within_randomstateB   sh     hh56qAKKNFF4Lqxxrwwr{3xGM	 r   c                     [         R                  " S/S//SS9  [         R                  " S/S//[        R                  " S5      S9  [         R                  " S/S//[        R                  " S5      S9  g )Nr   r   r   )r   multivariate_normalr   int_int64r   s    r   #test_multivariate_normal_size_types2TestRegression.test_multivariate_normal_size_typesL   s\     	""A3!A6""A3!BGGAJ?""A3!BHHQK@r   c                     [         R                  " S5        [         R                  " SSSS9n[        [        R
                  " [        R                  " U5      5      (       + S5        g )NiIg-C6?d   r   zNans in random.beta)r   r4   betar   r   anyisnan)r   xs     r   test_beta_small_parameters)TestRegression.test_beta_small_parametersT   sA     	JKKS1BFF288A;'')>?r   c                 n   [         R                  " S5        / SQn/ SQn[        R                  [        R                  [        R
                  4 Hd  n[        R                  " X#S9[        U5      -  n[         R                  " XS9n[        XQ;   5        [        [        [         R                  XS-  S9  Mf     g )N  )r   r3   r   )r!   r!   r3   rB   )rQ   g?)r   r4   r   float16float32float64rF   r6   rT   r   r   
ValueError)r   acountsdtprobscs         r   "test_choice_sum_of_probs_tolerance1TestRegression.test_choice_sum_of_probs_tolerance[   sy     	D**bjj"**4BHHV.V<Ea)AAFO*fmmQ)D	 5r   c                     [         R                  " S5        [        R                  " SS/5      n[	        S5       H  n[         R
                  " U5        M     SS KnUR                  5         g )Nrk   rp     aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarc   r   )r   r4   r   rF   rS   rE   gccollectr   rp   _ry   s       r   1test_shuffle_of_array_of_different_length_strings@TestRegression.test_shuffle_of_array_of_different_length_stringsh   sK     	DHHc:&'sANN1  	


r   c                 .   [         R                  " S5        [        R                  " [        R                  " S5      [        R                  " S5      /[
        S9n[        S5       H  n[         R                  " U5        M     SS KnUR                  5         g )Nrk   r   r!   rB   r1   r   )
r   r4   r   rF   arangerG   rS   rE   ry   rz   r{   s       r    test_shuffle_of_array_of_objects/TestRegression.test_shuffle_of_array_of_objectsw   s]     	DHHbiilBIIaL1@tANN1  	


r   c                     " S S[         R                  5      n[        R                  " S5        [         R                  " S5      R                  U5      n[        R                  " U5      n[        U[         R                  " / SQ5      5        [        U[         R                  " S5      R                  U5      5         " S S5      n[        R                  " S5        U" 5       n[        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   r7   r      s    r   r7   r   r   )r   r3   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)rp   )r   rC   copys      r   	__array__=TestRegression.test_permutation_subclass.<locals>.M.__array__   s    vvr   r   )NN)	r   r   r   r   r   r   rp   r   r   r   r   r   Mr      s    		!Ar   r   )r3   r   r!   r   r   r   )
r   ndarrayr   r4   r   viewpermutationr   rF   r   )r   r7   origpermr   rV   s         r   test_permutation_subclass(TestRegression.test_permutation_subclass   s    	

 	 	Ayy|  #!!$'4)!4541!2!21!56	 	 	AC!!!$4/!:;1;;="))A,7r   c                     [         R                  S:X  a  SOSn[        R                  " SS9   [        R
                  " SSSUS	9  S S S 5        g ! , (       d  f       g = f)
Nbigz<i4z>i4znon-native byteorder is notmatchr      r   )r    rC   )r'   	byteorderpytestdeprecated_callr   randint)r   other_byteord_dts     r   test_warns_byteorder#TestRegression.test_warns_byteorder   sD    $'MMU$:5##*GHNN1c2BC IHHs   A
Ac                     [         R                  R                  S5      n[         R                  R                  SS9nUR                  SS5      UR                  SS5      :X  d   eg )Ni[)r4   r   rc   )r   r   rR   r   )r   rs1rs2s      r   "test_named_argument_initialization1TestRegression.test_named_argument_initialization   sQ    ii##I.ii###3{{1c"ckk!S&9999r   c                    [         R                  R                  SS/S-  SS9nUR                  [         R                  " [         R                  5      :X  d   e[         R                  R                  SS/S-  SSS9nUR                  [         R                  " [         R                  5      :X  d   e[         R                  R                  SSS9nUR                  [         R                  " [         R                  5      :X  d   e[         R                  R                  SSSS9nUR                  [         R                  " [         R                  5      :X  d   eg )	Nr   g?r3   )rQ   r    F)rQ   replacer    r   )r   r    )r   r   rT   rC   long)r   rt   s     r   test_choice_retun_dtype&TestRegression.test_choice_retun_dtype   s    IIRB473ww"((277++++IIRB47EBww"((277++++IIRa(ww"((277++++IIRQ7ww"((277++++r   lr#   zCannot test with 32-bit C long)reasonc                     [         R                  " S5        [        R                  " / SQSS9n[         R                  " SSS9n[        X!5        g )Nr   )
l   
 l   /*/ l   u&-n l   LC0 l   C34 l   RD7 l   v l   g}1 i	tll   ^C? r^   rB   r#   r   r   )r   r4   r   rF   r   r   )r   rJ   actuals      r   test_randint_117TestRegression.test_randint_117   s@     	A88 5<CE B/6,r   c                     [         R                  R                  S5        [        [        R                  " S/ SQ5      / SQ5        g )NrA   r   )r         ?g      ?g      ?r   )r   r   r   r   r   )r   r   r4   r   binomialr_   s    r   test_p_zero_stream!TestRegression.test_p_zero_stream   s-     			u6??1.EF*	,r   c                     [         R                  R                  S5        [         R                  " / SQ/ SQ/5      n[	        [        R
                  " S/S//SSS9U5        g )	Ni_ )
r   r   r   r   r   r   r   r   r   r   )
r   r!   r3   r   r   r   r   r   r   r   r   r   r   )r3   r   r   )r   r   r4   rF   r   r   )r   rJ   s     r   test_n_zero_stream!TestRegression.test_n_zero_stream   sO     			w88;;= >6??QC";7K#	%r   r   N)r   r   r   r   r   r.   r;   rK   rY   r`   rh   ru   r}   r   r   r   r   r   r   markskipifr   iinfomaxr   r   r   r   r   r   r   r   r      s    ?5 " 
KNA@E8,D:	, [[))E1?  A-A-,%r   r   c                      [         R                  " S/ 5      R                  S:X  d   e[         R                  " S/ SS9R                  S:X  d   eg )Nr   )r   r   )rO   r   r   r   )rO   r   r   r   )r   multinomialshaper   r   r   test_multinomial_emptyr      sH     b"%++t333a)4::lJJJr   c                      [         R                  " [        SS9   [        R                  " SS5        S S S 5        g ! , (       d  f       g = f)Nzpvals must be a 1-dr   r   g333333?)r   raises	TypeErrorr   r   r   r   r   test_multinomial_1d_pvalr      s-    	y(=	>2s# 
?	>	>s	   ;
A	)r'   r   numpy.testingr   r   r   numpyr   r   r   r   r   r   r   r   <module>r      s5    
    ~% ~%BK$r   