
    gi                     \    d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZm	Z	m
Z
  G d d      Zy)    N)random)IS_PYPYassert_assert_array_equalassert_raisesc                   :   e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zej                  j!                  ej$                  j&                  dk(  d      ej                  j!                  ed      ej                  j+                  dej.                  ej0                  ej2                  ej4                  ej6                  ej8                  ej:                  ej<                  ej>                  ej@                  jB                  g
      de"ddfd                     Z#y)TestRegressionc                 .   t        j                  ddd      D ]{  }t        j                  j	                  |dd      }t        t        j                  |t         j                   kD        xr& t        j                  |t         j                  k               } y )Ng      g      @      2   )nplinspacer   mtrandvonmisesr   allpi)selfmurs      Z/var/www/html/audio_env/lib/python3.12/site-packages/numpy/random/tests/test_regression.pytest_VonMises_rangez"TestRegression.test_VonMises_range   si     ++c2q) 	?B&&r1b1ABFF1v:&=266!ruu*+=>	?    c           
         t        t        j                  t        j                  j	                  dddd      dk               t        t        j                  t        j                  j	                  dddd      dkD               dg}t
        j                  d	kD  }|r$t
        j                  d
k7  r|j                  d       |D ]*  }t        t        j                  j                  | dkD         , y )N         
   size   r   ) r"   r"   l        win32)   r$   r$   )	r   r   r   r   hypergeometricsysmaxsizeplatformappend)r   args	is_64bitsargs       r   test_hypergeometric_rangez(TestRegression.test_hypergeometric_range   s    ryy//2r/CaGHIryy//Ar/CaGHI .
 KK%'	0KK9: 	8CBII,,c2Q67	8r   c                 L   d}t         j                  j                  d       t         j                  j                  d|      }t        j                  |dk(        |z  }d|dd}t        |d	kD  |       t        j                  |d
k(        |z  }d|dd}t        |dk  |       y )N  r   g?r   r   zFrequency was fz, should be > 0.45g?   z, should be < 0.23gq=
ףp?)r   r   seed	logseriessumr   )r   Nrvsnfreqmsgs        r   test_logseries_convergencez)TestRegression.test_logseries_convergence$   s    
		qyy""3Q"/ vvdai 1$tAh&89tS!vvdai 1$tAh&89tS!r   c                 @   g dg dg dg dfD ]  }t         j                  j                  d      }t        |      }|j	                  |       t        j
                  |d   |d   |d   |d	   gt        
      }t        t        j
                  |t        
      |        y )N)r   r1   r   N)r   r   r1   r1   r   r   N)r   r<   r=   N)r;   r1   r   Ni90  r   r   r   r1   dtype)r   r   RandomStatelistshufflearrayobjectr   )r   trngshuffledexpecteds        r   test_shuffle_mixed_dimensionz+TestRegression.test_shuffle_mixed_dimension4   s    !0+&( 	KA ))''.CAwHKK!xx1qtQqT1Q4 8GHrxx?J	Kr   c           
      V   t         j                  j                         }t        j                  g d      }t	        d      D ]f  }t         j                  j                  |       |j                  d       t        |j                  ddt        j                  d      dz        |       h y )N)
r         r1   r   	   r!   rL   r   r   r   i  r   g      $@)r    p)	r   r   r@   rC   ranger2   r   choiceones)r   mresis       r   test_call_within_randomstatez+TestRegression.test_call_within_randomstate@   sy    II!!#hh56q 	PAIINN1FF4Lqxxrwwr{S7HxI3O		Pr   c                 .   t         j                  j                  dgdggd       t         j                  j                  dgdggt        j                  d             t         j                  j                  dgdggt        j                  d             y )Nr   r   r   )r   r   multivariate_normalint_int64)r   s    r   #test_multivariate_normal_size_typesz2TestRegression.test_multivariate_normal_size_typesJ   sq     			%%qcQC5q%9
		%%qcQC5rwwqz%B
		%%qcQC5rxx{%Cr   c                     t         j                  j                  d       t         j                  j                  ddd      }t	        t        j
                  t        j                  |             d       y )NiIg-C6?d   r   zNans in np.random.beta)r   r   r2   betar   anyisnan)r   xs     r   test_beta_small_parametersz)TestRegression.test_beta_small_parametersR   sK     			z"IINN66N4BFF288A;'')ABr   c                    t         j                  j                  d       g d}g d}t         j                  t         j                  t         j
                  fD ]  }t        j                  ||      t        |      z  }t         j                  j                  ||      }t        ||v        t        t        t         j                  j                  ||dz          y )N  )r   r1   r   )r!   r!   r1   r>   )rN   g?)r   r   r2   float16float32float64rC   r4   rP   r   r   
ValueError)r   acountsdtprobscs         r   "test_choice_sum_of_probs_tolerancez1TestRegression.test_choice_sum_of_probs_toleranceY   s     			t**bjj"**4 	JBHHV2.V<E		  e ,AAFO*bii&6&6US[I		Jr   c                     t         j                  j                  d       t        j                  ddg      }t	        d      D ]!  }t         j                  j                  |       # dd l}|j                          y )Nrc   rh     aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar\   r   )r   r   r2   rC   rO   rB   gccollectr   rh   _rp   s       r   1test_shuffle_of_array_of_different_length_stringsz@TestRegression.test_shuffle_of_array_of_different_length_stringsf   sZ     			tHHc:&'s 	!AIIa 	! 	


r   c                 N   t         j                  j                  d       t        j                  t        j                  d      t        j                  d      gt
              }t        d      D ]!  }t         j                  j                  |       # dd l}|j                          y )Nrc   r   r!   r>   r/   r   )
r   r   r2   rC   arangerD   rO   rB   rp   rq   rr   s       r    test_shuffle_of_array_of_objectsz/TestRegression.test_shuffle_of_array_of_objectsu   sl     			tHHbiilBIIaL1@t 	!AIIa 	! 	


r   c                     G d dt         j                        }t         j                  j                  d      }t        j                  d      j                  |      }|j                  |      }t        |t        j                  g d             t        |t        j                  d      j                  |              G d d      }t         j                  j                  d      } |       }|j                  |      }t        |t        j                  g d             t        |j                         t        j                  d	             y )
Nc                       e Zd Zy)3TestRegression.test_permutation_subclass.<locals>.NN)__name__
__module____qualname__ r   r   r5   rz      s    r   r5   r   r   )r   r1   r   c                   8    e Zd Z ej                  d      ZddZy)3TestRegression.test_permutation_subclass.<locals>.Mr   Nc                     | j                   S )N)rh   )r   r?   copys      r   	__array__z=TestRegression.test_permutation_subclass.<locals>.M.__array__   s    vvr   )NN)r{   r|   r}   r   rv   rh   r   r~   r   r   Mr      s    		!Ar   r   )r1   r   r!   r   r   r   )
r   ndarrayr   r@   rv   viewpermutationr   rC   r   )r   r5   rF   origpermr   rR   s          r   test_permutation_subclassz(TestRegression.test_permutation_subclass   s    	

 	 ii##A&yy|  #t$4)!4541!2!21!56	 	 ii##A&Cq!4/!:;1;;="))A,7r   r1   zPython running -OO)reasonzPyPy does not modify tp_docclsreturnNc                     t        |d      sJ 	 t        j                  |       y # t        $ r2 t	        j
                  d|j                   d|j                          Y y w xY w)N__text_signature__zinvalid signature: .)hasattrinspect	signaturerg   pytestfailr|   r}   )r   r   s     r   test_inspect_signaturez%TestRegression.test_inspect_signature   s`    $ s0111	Sc" 	SKK-cnn-=Qs?O?O>PQR	Ss   & 8A! A!)$r{   r|   r}   r   r-   r9   rI   rU   rZ   ra   rm   rt   rw   r   r   markskipifr&   flagsoptimizer   parametrizer   	GeneratorMT19937PCG64	PCG64DXSMPhiloxr@   SFC64BitGeneratorSeedSequencebit_generatorSeedlessSeedSequencetyper   r~   r   r   r	   r	      s   ?8 " 
KPDCJ8, [[		**a/8LM[[(EF[[NNLLMMLL  55	
S$ S4 S G N"Sr   r	   )r   r&   r   numpyr   r   numpy.testingr   r   r   r   r	   r~   r   r   <module>r      s(     
    M MdS dSr   