
    G/PhV                     P    d dl Z d dlmZmZmZ d dlmZ d dlZ G d d          ZdS )    N)assert_assert_array_equalassert_raises)randomc                   P    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dS )TestRegressionc                    t          j        ddd          D ]u}t          j                            |dd          }t          t          j        |t           j         k              o!t          j        |t           j        k                         vd S )Ng      g      @      2   )nplinspacer   mtrandvonmisesr   allpi)selfmurs      b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numpy/random/tests/test_regression.pytest_VonMises_rangez"TestRegression.test_VonMises_range   s|     +c2q)) 	? 	?B&&r1b11ABF1v:&&=26!ru*+=+=>>>>	? 	?    c           
         t          t          j        t          j                            dddd          dk                          t          t          j        t          j                            dddd          dk                         dg}t
          j        d	k    }|r%t
          j        d
k    r|                    d           |D ]&}t          t          j        j        | dk               'd S )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    ry//2r/CCaGHHIIIry//Ar/CCaGHHIII .
 K%'	 	;00KK9::: 	8 	8CBI,c2Q67777	8 	8r   c                 b   d}t           j                            d           t           j                            d|          }t          j        |dk              |z  }d|dd}t          |d	k    |           t          j        |d
k              |z  }d|dd}t          |dk     |           d S )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y""3Q"// vdai  1$9t9999tS!!!vdai  1$9t9999tS!!!!!r   c                 l   g dg dg dg dfD ]}t           j                            d           t          |          }t          j        |           t          j        |d         |d         |d         |d	         gt          
          }t          t          j        |t          
          |           d S )N)r   r0   r   N)r   r   r0   r0   r   r   N)r   r;   r<   N)r:   r0   r   Ni90  r   r   r   r0   dtype)r   r   r1   listshufflearrayobjectr   )r   tshuffledexpecteds       r   test_shuffle_mixed_dimensionz+TestRegression.test_shuffle_mixed_dimension2   s    !//000+++&&&( 	K 	KA INN5!!!AwwHN8$$$x1qtQqT1Q4 8GGGHrx???JJJJ	K 	Kr   c           
      p   t           j                                        }t          j        g d          }t	          d          D ]q}t           j                            |           |                    d           t          |                    ddt          j        d          dz            |           rd S )N)
r         r0   r   	   r    rI   r   r   r   i  r   g      $@)r   p)	r   r   RandomStaterA   ranger1   r   choiceones)r   mresis       r   test_call_within_randomstatez+TestRegression.test_call_within_randomstate>   s    I!!##h55566q 	N 	NAINN1FF4LLLqxxrwr{{3xGGMMMM		N 	Nr   c                 ,   t           j                            dgdggd           t           j                            dgdggt          j        d                     t           j                            dgdggt          j        d                     d S )Nr   r   r   )r   r   multivariate_normalint_int64)r   s    r   #test_multivariate_normal_size_typesz2TestRegression.test_multivariate_normal_size_typesH   s     		%%qcQC5q%999
	%%qcQC5rwqzz%BBB
	%%qcQC5rx{{%CCCCCr   c                     t           j                            d           t           j                            ddd          }t	          t          j        t          j        |                     d           d S )NiIg-C6?d   r   zNans in np.random.beta)r   r   r1   betar   anyisnan)r   xs     r   test_beta_small_parametersz)TestRegression.test_beta_small_parametersP   s]     		z"""INN66N44BF28A;;''')ABBBBBr   c                    t           j                            d           g d}g d}t           j        t           j        t           j        fD ]}t          j        ||          t          |          z  }t           j                            ||          }t          ||v            t          t          t           j        j        ||dz             d S )N  )r   r0   r   )r    r    r0   r=   )rK   g?)r   r   r1   float16float32float64rA   r3   rN   r   r   
ValueError)r   acountsdtprobscs         r   "test_choice_sum_of_probs_tolerancez1TestRegression.test_choice_sum_of_probs_toleranceW   s     		tII*bj"*4 	H 	HBHV2...V<E	  e ,,AAFOOO*bi&6U3YGGGGG		H 	Hr   c                    t           j                            d           t          j        ddg          }t	          d          D ]!}t           j                            |           "dd l}|                                 d S )Nra   rf     aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarZ   r   )r   r   r1   rA   rM   r@   gccollectr   rf   _rn   s       r   1test_shuffle_of_array_of_different_length_stringsz@TestRegression.test_shuffle_of_array_of_different_length_stringsd   sv     		tHc:&''s 	! 	!AIa     				


r   c                 X   t           j                            d           t          j        t          j        d          t          j        d          gt
                    }t          d          D ]!}t           j                            |           "dd l}|	                                 d S )Nra   r   r    r=   r.   r   )
r   r   r1   rA   arangerB   rM   r@   rn   ro   rp   s       r    test_shuffle_of_array_of_objectsz/TestRegression.test_shuffle_of_array_of_objectss   s     		tHbillBIaLL1@@@t 	! 	!AIa     				


r   c                     G d dt           j                  }t           j                            d           t          j        d                              |          }t           j                            |          }t          |t          j        g d                     t          |t          j        d                              |                      G d d          }t           j                            d            |            }t           j                            |          }t          |t          j        g d                     t          |	                                t          j        d	                     d S )
Nc                       e Zd ZdS )3TestRegression.test_permutation_subclass.<locals>.NN)__name__
__module____qualname__ r   r   r4   rx      s        Dr   r4   r   r   )r   r0   r   c                   6    e Zd Z ej        d          ZddZdS )3TestRegression.test_permutation_subclass.<locals>.Mr
   Nc                     | j         S )N)rf   )r   r>   copys      r   	__array__z=TestRegression.test_permutation_subclass.<locals>.M.__array__   s	    vr   )NN)ry   rz   r{   r   rt   rf   r   r|   r   r   Mr~      s7        	!A     r   r   )r0   r   r    r   r   r
   )
r   ndarrayr   r1   rt   viewpermutationr   rA   r   )r   r4   origpermr   rP   s         r   test_permutation_subclassz(TestRegression.test_permutation_subclass   sZ   	 	 	 	 	
 	 	 	 		qy||  ##y$$T**4)))!4!455541!2!21!5!5666	 	 	 	 	 	 	 	 		qACCy$$Q''4///!:!:;;;1;;==")A,,77777r   N)ry   rz   r{   r   r,   r8   rF   rS   rX   r_   rk   rr   ru   r   r|   r   r   r   r   	   s        ? ? ?8 8 8 " " " 
K 
K 
KN N ND D DC C CH H H    8 8 8 8 8r   r   )	r%   numpy.testingr   r   r   numpyr   r   r   r|   r   r   <module>r      s    



                   L8 L8 L8 L8 L8 L8 L8 L8 L8 L8r   