U
|
P ±dÅ ã @ sZ d dl Z d dlZd dlmZmZmZ d dlZd dlmZ G dd dZ dd Z
|
dd Z dS )
|
é N)Úassert_Úassert_array_equalÚ assert_raises)Úrandomc @ s° e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zejje d ¡jd k d d d!d" Zd#d$ Zd%d& Zd'S )(ÚTestRegressionc C sL t ddd¡D ]8}t |dd¡}tt |t j k¡oBt |t jk¡ qd S )Ng Àg @é é é2 )ÚnpZlinspacer Zvonmisesr ÚallÚpi)ÚselfÚmuÚr© r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/random/tests/test_randomstate_regression.pyÚtest_VonMises_range s z"TestRegression.test_VonMises_rangec C s t t tjddddddk ¡ t t tjddddddk¡ dg}tjd k}|rhtjd
|
krh| d ¡ |D ]}t tj| dk qld S ) Né é é é
|
©Úsizeé r )éþÿ r r ì Ú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 d¡ t 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 ÚseedZ logseriesr
|
Úsumr )r ÚNZrvsnÚfreqÚmsgr r r Útest_logseries_convergence& s
|
z)TestRegression.test_logseries_convergencec C s dddd gdddd gdddd gdddd gfD ]Z}t d¡ t|}t |¡ tj|d |d |d |d gtd }ttj|td | q,d S )
|
Nr r' r )r r )r' r' )r r é90 r ©Údtype)r r( ÚlistÚshuffler
|
ÚarrayÚobjectr )r ÚtZshuffledÚexpectedr r r Ú test_shuffle_mixed_dimension6 s
|
|
|
|
ý
|
|
&z+TestRegression.test_shuffle_mixed_dimensionc C sl t ¡ }t ddddddddddg
|
¡}tdD ]8}t |¡ | d ¡ t|jd
|
d
|
t d
|
¡d d | q.d S ) Nr é é r' r é r r iá r g $@)r Úp) r Ú RandomStater
|
r3 Úranger( r ÚchoiceZones)r ÚmÚresÚir r r Ú test_call_within_randomstateB 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_typesL s z2TestRegression.test_multivariate_normal_size_typesc C s6 t d¡ t jdddd}tt t |¡¡ d d S )NiÒIg-C ëâ6?éd r zNans in random.beta)r r( Úbetar r
|
ÚanyÚisnan)r Úxr r r Útest_beta_small_parametersT s
|
z)TestRegression.test_beta_small_parametersc C s~ t d¡ dddg}dddg}tjtjtjfD ]J}tj||dt| }t j||d}t ||k t
|
t t j||d d q.d S ) NéÒ r r' r r r/ )r; gÍÌÌÌÌÌì?) r r( r
|
Úfloat16Úfloat32Úfloat64r3 r) r> r r Ú
|
ValueError)r ÚaÚcountsÚdtZprobsÚcr r r Ú"test_choice_sum_of_probs_tolerance[ s
|
|
|
z1TestRegression.test_choice_sum_of_probs_tolerancec C sD t d¡ t ddg¡}tdD ]}t |¡ q dd l}| ¡ d S )NrM rR Aè aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarG r )r r( r
|
r3 r= r2 ÚgcÚcollect©r rR Ú_rW r r r Ú1test_shuffle_of_array_of_different_length_stringsh s
|
z@TestRegression.test_shuffle_of_array_of_different_length_stringsc C sT t d¡ tjt d¡t d¡gtd}tdD ]}t |¡ q0dd l}| ¡ d S )NrM r r r/ r% r )
|
r r( r
|
r3 Úaranger4 r= r2 rW rX rY r r r Ú test_shuffle_of_array_of_objectsw s
|
z/TestRegression.test_shuffle_of_array_of_objectsc C s¼ G dd dt j}t d¡ t d¡ |¡}t |¡}t|t dddg¡ t|t d¡ |¡ G dd d}t d¡ | }t |¡}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)rR rE r r r Ú __array__ s z=TestRegression.test_permutation_subclass.<locals>.M.__array__N)r^ r_ r` r
|
r\ rR ra r r r r ÚM s
|
rb r r )
|
r
|
Zndarrayr r( r\ ÚviewZ permutationr r3 ra )r r* ÚorigÚpermrb r? r r r Útest_permutation_subclass
s
|
|
|
|
z(TestRegression.test_permutation_subclassc C s@ t jdkrdnd}tjdd tjddd|d W 5 Q R X d S )
|
NÚbigz<i4z>i4znon-native byteorder is not©Úmatchr éÈ r )r r0 )r Ú byteorderÚpytestZdeprecated_callr Úrandint)r Zother_byteord_dtr r r Útest_warns_byteorder s z#TestRegression.test_warns_byteorderc C s: t j d¡}t jjdd}| dd¡| dd¡ks6td S )NiÍ[)r( r rG )r
|
r r< rm ÚAssertionError)r Zrs1Zrs2r r r Ú"test_named_argument_initialization¡ s z1TestRegression.test_named_argument_initializationc C s¨ t jjddgd dd}|jt t¡ks,tt jjddgd ddd}|jt t¡ksZtt jjddd}|jt t¡ks~tt jjdddd}|jt t¡ks¤td S ) Nr g¹?r' )r; r F)r; Úreplacer r )rq r )r
|
r r> r0 Úintro )r rU r r r Útest_choice_retun_dtype§ s z&TestRegression.test_choice_retun_dtypeÚlr z Cannot test with 32-bit C long)Úreasonc C sH t d¡ tjdddddddd d
|
d g
|
d d }t jddd}t|| d S )Nr l ¬
|
þ l /*/ l u&-n l ÀLC0 l C34 l ûRD7 l Ãvú l g}Ê1 i ¤tll Ó^C? rD r/ r r r )r r( r
|
r3 rm r )r r6 Úactualr r r Útest_randint_117² s
|
þþz TestRegression.test_randint_117c C s6 t j d¡ tt ddddddg¡dddddg d S )Nr. r r ç Ð?g à?g è?)r
|
r r( r ÚbinomialrE r r r Útest_p_zero_stream½ s ÿz!TestRegression.test_p_zero_streamc C sd t j d¡ t ddddddddddg
|
ddddddddddg
|
g¡}ttjdgdggd d
|
d | d S ) Nií_ r r r r' r r r rx )r' r r )r
|
r r( r3 r ry )r r6 r r r Útest_n_zero_streamÄ s ÿÿz!TestRegression.test_n_zero_streamN)r^ r_ r` r r$ r- r7 rB rF rL rV r[ r] rf rn rp rs rl ÚmarkZskipifr
|
ZiinfoÚmaxrw rz r{ r r r r r s(
|
ÿ
|
r c C s4 t dg ¡jdkstt jdg ddjdks0td S )Nr )r r )r9 r r r )r9 r r r )r Ú multinomialÚshapero r r r r Útest_multinomial_emptyÎ s r c C s* t jtdd t dd¡ W 5 Q R X d S )Nzpvals must be a 1-drh r g333333Ó?)rl ZraisesÚ TypeErrorr r~ r r r r Útest_multinomial_1d_pvalÕ s r ) r rl Z numpy.testingr r r Únumpyr
|
r r r r r r r r Ú<module> s B
|
