+
    ei  c                   sJ    ^ RI t ^ RIt^ RIHtHt ^ RIHtHt  ! R R4      t	R# )    N)MT19937	Generator)assert_assert_array_equalc                   s   a  ] tR t^t o R tR tR tR tR tR t	R t
R tR	 tR
 tR tR tR tR tR tR tR tR tR tRtV tR# )TestRegressionc                s8    \        \        R 4      4      V n        R# )l   (vn4Ti N)r   r   mt19937self   &\/usr/lib64/python3.14/site-packages/numpy/random/tests/test_generator_mt19937_regressions.pysetup_methodZTestRegression.setup_method
   s     )>!?@    c                s>   \         P                  ! RR ^4       F  pV P                  P                  V^^24      p\	        \         P
                  ! V\         P                  ) 8  4      ;'       d(    \         P
                  ! V\         P                  8*  4      4       K  	  R# )g      @Ng      )npZlinspacer   Zvonmisesr   allZpi)r   ZmuZr   &  r
   test_vonmises_rangeZ"TestRegression.test_vonmises_range   sg     ++c2q)B%%b!R0ABFF1v:&==266!ruu*+=> *r   c                sR   \        \        P                  ! V P                  P	                  ^^^^
R7      ^8  4      4       \        \        P                  ! V P                  P	                  ^^^^
R7      ^ 8  4      4       Rp\        V P                  P                  ! V!  ^ 8  4       R# )   sizeN) r   r   )r   r   r   r   Zhypergeometric)r   Zargs   & r
   test_hypergeometric_rangeZ(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   R pV P                   P                  RVR7      p\        P                  ! V^8H  4      V,          pRVR R2p\	        VR8  V4       \        P                  ! V^8H  4      V,          pRVR R2p\	        VR8  V4       R	# )
  g?r   zFrequency was Zfz, should be > 0.45g?z, should be < 0.23gq=
ףp?N)r   Z	logseriesr   sumr   )r   NZrvsnZfreqZmsg   &    r
   test_logseries_convergenceZ)TestRegression.test_logseries_convergence   s    ||%%c%2 vvdai 1$tAh&89tS!vvdai 1$tAh&89tS!r   c                sv   . RO. RO. RO. RO3 F  p\        \        R4      4      p\        P                  ! V\        R7      pVP                  V4       \        P                  ! V^,          V^ ,          V^,          V^,          .\        R7      p\        \        P                  ! V\        R7      V4       K  	  R# )   Ni90  dtype)r      r   N)r   r   r   r   r   r   N)r   r!   r"   N)r    r   r   N)r   r   r   arrayobjectshuffler   )r   Ztr   Zshuffledexpectedr   r
   test_shuffle_mixed_dimensionZ+TestRegression.test_shuffle_mixed_dimension,   s    !0+&(A  /Gxx0HOOH%xx1qtQqT1Q4 8GHrxx?J(r   c           
     s   \         P                  ! . RO4      p\        ^4       Fd  p\        \	        V4      4      p\        \	        R4      4      p\        VP                  ^
^
\         P                  ! ^
4      R,          R7      V4       Kf  	  R# )r   i  g      $@)r   pN)
r      r    r   i   r   r   r)   r      )r   r#   ranger   r   r   choiceones)r   ZresZir   mr   r
   test_call_within_randomstateZ+TestRegression.test_call_within_randomstate8   sa    hh56qA
+G'$-(Aqxxrwwr{S7HxI3O	 r   c                s   V P                   P                  ^ .^ ..^R7       V P                   P                  ^ .^ ..\        P                  ! ^4      R7       V P                   P                  ^ .^ ..\        P                  ! ^4      R7       R# )r    r   N)r   Zmultivariate_normalr   Zint_int64r   r	   r
   #test_multivariate_normal_size_typesZ2TestRegression.test_multivariate_normal_size_typesA   sq     	((!se!(<((!se"''!*(E((!se"((1+(Fr   c                s    V P                   P                  R R ^dR7      p\        \        P                  ! \        P
                  ! V4      4      '       * R4       R# )g-C6?r   zNans in mt19937.betaN)r   betar   r   anyisnan)r   xr   r
   test_beta_small_parametersZ)TestRegression.test_beta_small_parametersI   s>     LLff37BFF288A;'')?@r   c                s>    V P                   P                  R R4       R# )gn5gWw'&l7N)r   r2   r   r	   r
   test_beta_very_small_parametersZ.TestRegression.test_beta_very_small_parametersO   s    %'r   c                s   \         P                  ! R 4      P                  pV P                  P	                  V^ ,          V^(,          ^2R7      p\         P
                  ! \         P                  ! V4      4      '       d   Q hR# )g      ?r   N)r   Zfinfotinyr   r2   r3   r4   )r   r8   r5   r   r
   'test_beta_ridiculously_small_parametersZ6TestRegression.test_beta_ridiculously_small_parametersS   sZ     xx}!!LLdRi<66"((1+&&&&&r   c                s    R pR pRpV P                   P                  WVR7      p\        P                  ! V^ 8H  4      pRpRV,          Tu;8  d   RV,          8  g   Q h Q hR# )g{Gzd?i@B r   gSis@gffffff?g?N)r   r2   r   count_nonzero)r   aZbnr5   ZnzerosZexpected_freq   &      r
   !test_beta_expected_zero_frequencyZ0TestRegression.test_beta_expected_zero_frequencyZ   sj     LLa+!!!q&) *m#fCtm/CCCCCCr   c                s   . ROp. ROp\         P                  \         P                  \         P                  3 F  p\         P                  ! W#R7      \        V4      ,          pV P                  P                  WR7      p\        WQ9   4       \        P                  ! \        4      ;_uu_ 4        V P                  P                  WR,          R7       RRR4       K  	  R#   + '       g   i     K  ; i)r   r   )r(   g?N)r   r   r   )r*   r*   r   )r   Zfloat16float32Zfloat64r#   r   r   r,   r   pytestZraisesZ
ValueError)r   r;   ZcountsZdtZprobsZcs   &     r
   "test_choice_sum_of_probs_toleranceZ1TestRegression.test_choice_sum_of_probs_toleranceq   s     **bjj"**4BHHV.V<E##A#/AAFOz**##A#5 +*	 5 +**s   -$CC0c                s    \         P                  ! R R.4      p\        ^d4       F  pV P                  P	                  V4       K   	  ^ RIpVP                  4        R# )r;   NA  aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)r   r#   r+   r   r%   gccollectr   r;   Z_rB      &   r
   1test_shuffle_of_array_of_different_length_stringsZ@TestRegression.test_shuffle_of_array_of_different_length_strings~   sE    
 HHc:&'sALL  #  	


r   c                s   \         P                  ! \         P                  ! ^4      \         P                  ! ^4      .\        R7      p\	        R4       F  pV P
                  P                  V4       K   	  ^ RIpVP                  4        R# )r   r   r   N)	r   r#   aranger$   r+   r   r%   rB   rC   rD   rE   r
    test_shuffle_of_array_of_objectsZ/TestRegression.test_shuffle_of_array_of_objects   sW     HHbiilBIIaL1@tALL  #  	


r   c                sz    ! R  R\         P                  4      p\        \        ^4      4      p\         P                  ! ^4      P                  V4      pVP                  V4      p\        V\         P                  ! . RO4      4       \        V\         P                  ! ^4      P                  V4      4        ! R R4      p\        \        ^4      4      pV! 4       pVP                  V4      p\        V\         P                  ! . RO4      4       \        VP                  4       \         P                  ! ^4      4       R# )c                   s    ] tR t^tRtR# )3TestRegression.test_permutation_subclass.<locals>.N N)__name__
__module____qualname____firstlineno____static_attributes__rJ   r   r
   r   rI      s    r   r   c                   sN   a  ] tR t^t o ]P
                  ! ^4      tRR ltRtV t	R# )3TestRegression.test_permutation_subclass.<locals>.MNc                s    V P                   # )N)r;   )r   r   Zcopys   &&&r
   	__array__Z=TestRegression.test_permutation_subclass.<locals>.M.__array__   s    vvr   rJ   )NN)
rK   rL   rM   rN   r   rG   r;   rQ   rO   __classdictcell__Z__classdict__   @r
   MrP      s     		!A r   rU   N)r   r    r   )r*   r   r   r    r   )
r   Zndarrayr   r   rG   ZviewZpermutationr   r#   rQ   )r   r   r   ZorigZpermrU   r.   r=   r
   test_permutation_subclassZ(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                s6   V P                   P                  R 4      R 8X  g   Q h\        V P                   P                  R .4      R 4       V P                   P                  R .RR7      p\        P                  ! R .\        P
                  R7      p\        W4       R# )g        Zfloatr   N)r   Zstandard_gammar   r   r#   r?   )r   Zactualr&   r   r
   test_gamma_0ZTestRegression.test_gamma_0   ss    ||**3/36664<<66u=sC,,cU',B88RD

36,r   c                s    \        V P                  P                  R ^R7      \        P                  ! \        P
                  4      P                  4       R# )gKH9)r(   r   N)r   r   Z	geometricr   Ziinfor0   Zmaxr   r	   r
   test_geometric_tiny_probZ'TestRegression.test_geometric_tiny_prob   s8     	4<<11E1B88BHH-11	3r   c                s    ^pV P                   P                  RVR7      p\        V\        P                  ! V\        P
                  R7      4       R# )r)   i'  r   r   N)r   zipfr   r   r-   r0   r   r<   Zsampler   r
   test_zipf_large_parameterZ(TestRegression.test_zipf_large_parameter   s;     ""5q"162771BHH#=>r   c                s    R pV P                   P                  RVR7      p\        P                  ! VR8  4      V^,          8  g   Q hR# )i g    ?r   Nl            )r   rY   r   r:   rZ   r   r
   test_zipf_a_near_1Z!TestRegression.test_zipf_a_near_1   sE     ""?";
 /!a%777r   )r   N)rK   rL   rM   rN   r   r   r   r   r'   r/   r1   r6   r7   r9   r>   rA   rF   rH   rV   rW   rX   r[   r\   rO   rR   rS   rT   r
   r   r      so     A?8"
KPGA('D.68.-3?	8 	8r   r   )
r@   Znumpyr   Znumpy.randomr   r   Znumpy.testingr   r   r   rJ   r   r
   <module>r]      s      + 5G8 G8r   