U
|
P ±dÔ ã @ sB d dl Z d dlmZmZmZ d dlmZ d dlZG dd dZdS )é N)Úassert_Úassert_array_equalÚ assert_raises)Úrandomc @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )ÚTestRegressionc C sN t ddd¡D ]:}tj |dd¡}tt |t j k¡oDt |t jk¡ qd S )Ng Àg @é é é2 )ÚnpZlinspacer ZmtrandZvonmisesr ÚallÚpi)ÚselfÚmuÚr© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/random/tests/test_regression.pyÚtest_VonMises_range s z"TestRegression.test_VonMises_rangec C s t t tjjddddddk ¡ t t tjjddddddk¡ dg}tjd k}|rltjd
|
krl| d ¡ |D ]}t tjj| dk qpd S ) Né é é é
|
©Úsizeé r )éþÿ r r l Úwin32)ì þÿÿr r ) r r
|
r r ZhypergeometricÚsysÚmaxsizeÚplatformÚappend)r ÚargsZ is_64bitsÚargr r r Útest_hypergeometric_range s ""ÿ
|
|
z(TestRegression.test_hypergeometric_rangec C s d}t j d¡ t jjd|d}t |dk¡| }d|dd}t|d k| t |d
|
k¡| }d|dd }t|d k | d S ) Néè r gé?r r zFrequency was Úfz, should be > 0.45gÍÌÌÌÌÌÜ?é z, should be < 0.23gq=
|
×£pÍ?)r
|
r ÚseedZ logseriesÚsumr )r ÚNZrvsnÚfreqÚmsgr r r Útest_logseries_convergence" s z)TestRegression.test_logseries_convergencec C s dddd gdddd gdddd gdddd gfD ]\}t j d¡ t|}t |¡ t j|d |d |d |d gtd }tt j|td | q,d S )
|
Nr r& r )r r )r& r& )r r i90 r ©Zdtype)r
|
r r' ÚlistÚshuffleÚarrayÚobjectr )r ÚtZshuffledÚexpectedr r r Ú test_shuffle_mixed_dimension2 s
|
|
|
|
ý
|
&z+TestRegression.test_shuffle_mixed_dimensionc C sp t j ¡ }t ddddddddddg
|
¡}tdD ]:}t j |¡ | d ¡ t|jd
|
d
|
t d
|
¡d d | q0d S ) Nr é é r& r é r r iá r g $@)r Úp) r
|
r Z RandomStater0 Úranger' r ÚchoiceZones)r ÚmÚresÚir r r Ú test_call_within_randomstate> s
|
|
z+TestRegression.test_call_within_randomstatec C sX t jjdgdggdd t jjdgdggt d¡d t jjdgdggt d¡d d S )Nr r r )r
|
r Zmultivariate_normalÚint_Úint64©r r r r Ú#test_multivariate_normal_size_typesH s z2TestRegression.test_multivariate_normal_size_typesc C s: t j d¡ t jjdddd}tt t |¡¡ d d S )NiÒIg-C ëâ6?éd r zNans in np.random.beta)r
|
r r' Úbetar ÚanyÚisnan)r Úxr r r Útest_beta_small_parametersP s z)TestRegression.test_beta_small_parametersc C s t j d¡ dddg}dddg}t jt jt jfD ]N}t j||dt| }t jj||d}t ||k t
|
t t jj||d d q0d S ) NéÒ r r& r r r- )r8 gÍÌÌÌÌÌì?) r
|
r r' Úfloat16Úfloat32Úfloat64r0 r( r: r r Ú
|
ValueError)r ÚaÚcountsÚdtZprobsÚcr r r Ú"test_choice_sum_of_probs_toleranceW s
|
|
z1TestRegression.test_choice_sum_of_probs_tolerancec C sH t j d¡ t ddg¡}tdD ]}t j |¡ q"dd l}| ¡ d S )NrI rN Aè aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarC r )r
|
r r' r0 r9 r/ ÚgcÚcollect©r rN Ú_rS r r r Ú1test_shuffle_of_array_of_different_length_stringsd s z@TestRegression.test_shuffle_of_array_of_different_length_stringsc C sX t j d¡ t jt d¡t d¡gtd}tdD ]}t j |¡ q2dd l}| ¡ d S )NrI r r r- r$ r )
|
r
|
r r' r0 Úaranger1 r9 r/ rS rT rU r r r Ú test_shuffle_of_array_of_objectss s z/TestRegression.test_shuffle_of_array_of_objectsc C sÄ G dd dt j}t j d¡ t d¡ |¡}t j |¡}t|t dddg¡ t|t d¡ |¡ G dd d}t j d¡ | }t j |¡}t|t ddd ddg¡ t| ¡ t d
|
¡ d S ) Nc @ s e Zd ZdS )z3TestRegression.test_permutation_subclass.<locals>.NN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r) s r) r r r r& c @ s e Zd Ze d¡Zdd ZdS )z3TestRegression.test_permutation_subclass.<locals>.Mr c S s | j S )N)rN rA r r r Ú __array__ s z=TestRegression.test_permutation_subclass.<locals>.M.__array__N)rZ r[ r\ r
|
rX rN r] r r r r ÚM s
|
r^ r r )
|
r
|
Zndarrayr r' rX ÚviewZ permutationr r0 r] )r r) ÚorigÚpermr^ r; r r r Útest_permutation_subclass s z(TestRegression.test_permutation_subclassN)rZ r[ r\ r r# r, r4 r> rB rH rR rW rY rb r r r r r s
|
r ) r Z numpy.testingr r r Únumpyr r
|
r r r r r Ú<module> s
|
