Ë
    ¡ehV  ã                   óH   — 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y)é    N)Úassert_Úassert_array_equalÚassert_raises)Úrandomc                   óN   — 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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      ú`/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/numpy/random/tests/test_regression.pyÚtest_VonMises_rangez"TestRegression.test_VonMises_range   si   € ô —+‘+˜c 2 qÓ)ò 	?ˆBÜ—‘×&Ñ& r¨1¨bÓ1ˆAÜ”B—F‘F˜1¤§¡˜v™:Ó&Ò=¬2¯6©6°!´r·u±u±*Ó+=Õ>ñ	?ó    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À   € ä”—‘”r—y‘y×/Ñ/°°2°rÀÐ/ÓCÀaÑGÓHÔIÜ”—‘”r—y‘y×/Ñ/°°A°rÀÐ/ÓCÀaÑGÓHÔIð .ð
ˆô —K‘K %Ñ'ˆ	ÙœŸ™¨Ò0à�K‰KÐ9Ô:Øò 	8ˆCÜ”B—I‘I×,Ñ,¨cÐ2°QÑ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ÔÜ�y‰y×"Ñ" 3¨QÐ"Ó/ˆô �v‰v�d˜a‘iÓ  1Ñ$ˆØ˜t A˜hÐ&8Ð9ˆÜ��t‘˜SÔ!ä�v‰v�d˜a‘iÓ  1Ñ$ˆØ˜t A˜hÐ&8Ð9ˆÜ��t‘˜SÕ!r   c                 óH  — g d¢g d¢g d¢g d¢fD ]‘  }t         j                  j                  d«       t        |«      }t        j                  |«       t        j
                  |d   |d   |d   |d	   gt        ¬
«      }t        t        j
                  |t        ¬
«      |«       Œ“ y )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Ú+Ú&ð(ò 	KˆAô �I‰I�N‰N˜5Ô!Ü˜A“wˆHÜ�N‰N˜8Ô$Ü—x‘x  1¡ q¨¡t¨Q¨q©T°1°Q±4Ð 8ÄÔGˆHÜœrŸx™x¨¼Ô?ÀÕ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   é   é   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>   sx   € ä�I‰I×!Ñ!Ó#ˆÜ�h‰hÒ5Ó6ˆÜ�q“ò 	NˆAÜ�I‰I�N‰N˜1ÔØ�F‰F�4ŒLä˜qŸx™x¨°´r·w±w¸r³{À3±˜xÓGÈÕMñ		N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_typesH   sq   € ô 	�	‰	×%Ñ% q c¨Q¨C¨5°qÐ%Ô9Ü
�	‰	×%Ñ% q c¨Q¨C¨5´r·w±w¸q³zÐ%ÔBÜ
�	‰	×%Ñ% q c¨Q¨C¨5´r·x±xÀ³{Ð%Õ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   r1   Úbetar   ÚanyÚisnan)r   Úxs     r   Útest_beta_small_parametersz)TestRegression.test_beta_small_parametersP   sK   € ô 	�	‰	�‰�zÔ"Ü�I‰I�N‰N˜6 6°ˆNÓ4ˆÜ”B—F‘Fœ2Ÿ8™8 A›;Ó'Ð'Ð)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   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ÔÚˆÚˆÜ—*‘*œbŸj™j¬"¯*©*Ð4ò 	HˆBÜ—H‘H˜V¨2Ô.´°V³Ñ<ˆEÜ—	‘	× Ñ   eÐ Ó,ˆAÜ�A˜�FŒOÜœ*¤b§i¡i×&6Ñ&6¸¸UÀ3¹YÖGñ		Hr   c                 óö   — t         j                  j                  d«       t        j                  ddg«      }t	        d«      D ]!  }t         j                  j                  |«       Œ# dd l}|j                  «        y )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   sZ   € ô 	�	‰	�‰�tÔä�H‰H�c˜:Ð&Ó'ˆä�s“ò 	!ˆAÜ�I‰I×Ñ˜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 )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   sl   € ô 	�	‰	�‰�tÔÜ�H‰H”b—i‘i “l¤B§I¡I¨a£LÐ1¼Ô@ˆä�t“ò 	!ˆAÜ�I‰I×Ñ˜aÕ ð	!ó 	Ø
�
‰
�r   c                 óÒ  —  G d„ dt         j                  «      }t         j                  j                  d«       t        j                  d«      j                  |«      }t         j                  j                  |«      }t        |t        j                  g d¢«      «       t        |t        j                  d«      j                  |«      «        G d„ d«      }t         j                  j                  d«        |«       }t         j                  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   r4   rx   ‚   s   „ Ør   r4   r   r   )r   r0   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)rf   )r   r>   Úcopys      r   Ú	__array__z=TestRegression.test_permutation_subclass.<locals>.M.__array__Ž   s   € Ø—v‘v�r   )NN)ry   rz   r{   r   rt   rf   r�   r|   r   r   ÚMr~   ‹   s   „ Ø�—	‘	˜!“ˆ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�   sâ   € ô	”—
‘
ô 	ô 	�	‰	�‰�qÔÜ�y‰y˜‹|× Ñ  Ó#ˆÜ�y‰y×$Ñ$ TÓ*ˆÜ˜4¤§¡ª)Ó!4Ô5Ü˜4¤§¡¨1£×!2Ñ!2°1Ó!5Ô6÷	ñ 	ô 	�	‰	�‰�qÔÙ‹CˆÜ�y‰y×$Ñ$ QÓ'ˆÜ˜4¤§¡ª/Ó!:Ô;Ü˜1Ÿ;™;›=¬"¯)©)°A«,Õ7r   N)ry   rz   r{   r   r,   r8   rF   rS   rX   r_   rk   rr   ru   rˆ   r|   r   r   r   r   	   s>   „ ò?ò8ò "ò 
KòNòDòCòHòòó8r   r   )	r%   Únumpy.testingr   r   r   Únumpyr   r   r   r|   r   r   ú<module>r‹      s&   ðÛ 
÷ñ õ Û ÷L8ò L8r   