U
|
P ±d ã @ sL d dl mZmZ d dlZd dlZd dlmZmZ ee Z G dd dZ
|
dS )é )Úassert_Úassert_array_equalN)Ú GeneratorÚMT19937c @ sl e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd ZdS )ÚTestRegressionc C sL t ddd¡D ]8}t |dd¡}tt |t j k¡oBt |t jk¡ qd S )Ng Àg @é é é2 )ÚnpZlinspaceÚmt19937Zvonmisesr ÚallÚpi)ÚselfÚmuÚr© r úld:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/random/tests/test_generator_mt19937_regressions.pyÚtest_vonmises_range s z"TestRegression.test_vonmises_rangec C sZ t t tjddddddk ¡ t t tjddddddk¡ d}t tj| dk d S ) Né é é é
|
©Úsizeé r )éþÿ r r )r r
|
r r Zhypergeometric)r Úargsr r r Útest_hypergeometric_range s z(TestRegression.test_hypergeometric_rangec C s~ d}t td}|jd|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 Z logseriesr
|
Úsumr )r ÚNr ZrvsnÚfreqÚmsgr r r Útest_logseries_convergence s z)TestRegression.test_logseries_convergencec C s dddd gdddd gdddd gdddd gfD ]b}t td}tj|td}| |¡ tj|d |d |d |d gtd}ttj|td| q,d S )
|
Nr r r )r r )r r )r r i90 ©Zdtyper )r r r
|
ÚarrayÚobjectÚshuffler )r Útr ZshuffledÚexpectedr r r Ú test_shuffle_mixed_dimension+ s
|
|
|
|
ý
|
&z+TestRegression.test_shuffle_mixed_dimensionc C sh t ddddddddddg
|
¡}tdD ]<}tt|}ttd}t|jddt d¡d d
|
| q&d S ) Nr é r r r r iá r g $@)r Úp)r
|
r' Úranger r r ÚchoiceZones)r ÚresÚir Úmr r r Ú test_call_within_randomstate7 s
|
z+TestRegression.test_call_within_randomstatec C sR t jdgdggdd t jdgdggt d¡d t jdgdggt d¡d d S )Nr r r )r Zmultivariate_normalr
|
Úint_Úint64©r r r r Ú#test_multivariate_normal_size_types@ s z2TestRegression.test_multivariate_normal_size_typesc C s8 t td}|jdddd}tt t |¡¡ d d S )NiÒIg-C ëâ6?éd r zNans in mt19937.beta)r r Úbetar r
|
ÚanyÚisnan)r r Úxr r r Útest_beta_small_parametersH s z)TestRegression.test_beta_small_parametersc
|
C s t td}dddg}dddg}tjtjtjfD ]\}tj||dt| }|j||d}t ||k t
|
t ¡ |j||d d W 5 Q R X q0d S ) NéÒ r r r r r& )r. gÍÌÌÌÌÌì?) r r r
|
Úfloat16Úfloat32Úfloat64r' r! r0 r ÚpytestZraisesÚ
|
ValueError)r r ÚaÚcountsÚdtZprobsÚcr r r Ú"test_choice_sum_of_probs_toleranceO s
|
|
z1TestRegression.test_choice_sum_of_probs_tolerancec C sF t td}t ddg¡}tdD ]}| |¡ q"dd l}| ¡ d S )Nr? rE Aè aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar9 r )r r r
|
r' r/ r) ÚgcÚcollect©r r rE Ú_rJ r r r Ú1test_shuffle_of_array_of_different_length_strings] s z@TestRegression.test_shuffle_of_array_of_different_length_stringsc C sV t td}tjt d¡t d¡gtd}tdD ]}| |¡ q2dd l}| ¡ d S )Nr? r r r& r r )
|
r r r
|
r' Úaranger( r/ r) rJ rK rL r r r Ú test_shuffle_of_array_of_objectsl s z/TestRegression.test_shuffle_of_array_of_objectsc C sÀ G dd dt j}ttd}t d¡ |¡}| |¡}t|t dddg¡ t|t d¡ |¡ G dd d}ttd}| }| |¡}t|t d ddddg¡ 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)rE r7 r r r Ú __array__ s z=TestRegression.test_permutation_subclass.<locals>.M.__array__N)rQ rR rS r
|
rO rE rT r r r r ÚM s
|
rU r r )
|
r
|
Zndarrayr r rO ÚviewZ permutationr r' rT )r r" r ÚorigÚpermrU r3 r r r Útest_permutation_subclassz s
|
|
z(TestRegression.test_permutation_subclassc C sT t d¡dksttt dg¡d t jdgdd}tjdgtjd}t|| d S )Ng Úfloatr& )r Zstandard_gammaÚAssertionErrorr r
|
r' rA )r Úactualr+ r r r Ú test_gamma_0 s
|
zTestRegression.test_gamma_0N)rQ rR rS r r r% r, r4 r8 r> rI rN rP rY r] r r r r r s r ) Z numpy.testingr r Únumpyr
|
rC Z numpy.randomr r r r r r r r Ú<module> s
|
|
|
