o 6˜¿aMã@sBddlZddlmZmZmZddlmZddlZGdd„dƒZdS)éN)Úassert_Úassert_array_equalÚ assert_raises)Úrandomc@sdeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dS)ÚTestRegressioncCsNt ddd¡D]}tj |dd¡}tt |tj k¡o"t |tjk¡ƒqdS)NgÀg@ééé2)ÚnpÚlinspacerÚmtrandÚvonmisesrÚallÚpi)ÚselfÚmuÚr©rúD/usr/lib/python3/dist-packages/numpy/random/tests/test_regression.pyÚtest_VonMises_range s(þz"TestRegression.test_VonMises_rangec CsŽtt tjjddddddk¡ƒtt tjjddddddk¡ƒdg}tjd k}|r6tjd kr6| d ¡|D] }ttjj|Ždkƒq8dS) Nééé é ©Úsizeér)éþÿrrlÚwin32)ìþÿÿrr) rr rrÚhypergeometricÚsysÚmaxsizeÚplatformÚappend)rÚargsÚ is_64bitsÚargrrrÚtest_hypergeometric_ranges""ÿ  ÿz(TestRegression.test_hypergeometric_rangecCsˆd}tj d¡tjjd|d}t |dk¡t|ƒ}d|d›d}t|d k|ƒt |d k¡t|ƒ}d|d›d }t|d k|ƒdS) Néèrgš™™™™™é?rrzFrequency was Úfz, should be > 0.45gÍÌÌÌÌÌÜ?éz, should be < 0.23gq= ×£pÍ?)r rÚseedÚ logseriesÚsumÚfloatr)rÚNÚrvsnÚfreqÚmsgrrrÚtest_logseries_convergence"s z)TestRegression.test_logseries_convergencecCs~gd¢gd¢gd¢gd¢fD].}tj d¡t|ƒ}t |¡tj|d|d|d|d gtd }ttj|td |ƒqdS) N)rr+rN)©rr©r+r+©rrN)rr6r7N)r5r+rNi90rrrr+©Údtype)r rr,ÚlistÚshuffleÚarrayÚobjectr)rÚtÚshuffledÚexpectedrrrÚtest_shuffle_mixed_dimension2sý  &øz+TestRegression.test_shuffle_mixed_dimensioncCs`tj ¡}t gd¢¡}tdƒD]}tj |¡| d¡t|jddt d¡dd|ƒqdS)N) réér+ré rrCrrriárg$@)rÚp) r rÚ RandomStater<Úranger,rÚchoiceÚones)rÚmÚresÚirrrÚtest_call_within_randomstate>s    "üz+TestRegression.test_call_within_randomstatecCsXtjjdgdggddtjjdgdggt d¡dtjjdgdggt d¡ddS)Nrrr)r rÚmultivariate_normalÚint_Úint64©rrrrÚ#test_multivariate_normal_size_typesHs"z2TestRegression.test_multivariate_normal_size_typescCs:tj d¡tjjdddd}tt t |¡¡ dƒdS)NiÒ–Ig-Cëâ6?édrzNans in np.random.beta)r rr,ÚbetarÚanyÚisnan)rÚxrrrÚtest_beta_small_parametersPs z)TestRegression.test_beta_small_parameterscCs€tj d¡gd¢}gd¢}tjtjtjfD]'}tj||dt|ƒ}tjj||d}t ||vƒt t tjj||ddqdS)NéÒ)rr+r)rrr+r8)rEgÍÌÌÌÌÌì?) r rr,Úfloat16Úfloat32Úfloat64r<r.rHrrÚ ValueError)rÚaÚcountsÚdtÚprobsÚcrrrÚ"test_choice_sum_of_probs_toleranceWs  üz1TestRegression.test_choice_sum_of_probs_tolerancecCsHtj d¡t ddg¡}tdƒD]}tj |¡qddl}| ¡dS)NrYr^ÁèaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarSr)r rr,r<rGr;ÚgcÚcollect©rr^Ú_rerrrÚ1test_shuffle_of_array_of_different_length_stringsds   z@TestRegression.test_shuffle_of_array_of_different_length_stringscCsXtj d¡tjt d¡t d¡gtd}tdƒD]}tj |¡qddl}|  ¡dS)NrYrrr8r)r) r rr,r<Úaranger=rGr;rerfrgrrrÚ test_shuffle_of_array_of_objectsss   z/TestRegression.test_shuffle_of_array_of_objectscCs¼Gdd„dtjƒ}tj d¡t d¡ |¡}tj |¡}t|t gd¢¡ƒt|t d¡ |¡ƒGdd„dƒ}tj d¡|ƒ}tj |¡}t|t gd¢¡ƒt|  ¡t d ¡ƒdS) Nc@s eZdZdS)z3TestRegression.test_permutation_subclass..NN)Ú__name__Ú __module__Ú __qualname__rrrrr0‚sr0rr)rr+rc@seZdZe d¡Zdd„ZdS)z3TestRegression.test_permutation_subclass..MrcSs|jS)N)r^rQrrrÚ __array__Žsz=TestRegression.test_permutation_subclass..M.__array__N)rlrmrnr rjr^rorrrrÚM‹s  rp)r+rrrrr) r Úndarrayrr,rjÚviewÚ permutationrr<ro)rr0ÚorigÚpermrprJrrrÚtest_permutation_subclasss    z(TestRegression.test_permutation_subclassN)rlrmrnrr(r4rArMrRrXrcrirkrvrrrrr s    r) r!Ú numpy.testingrrrÚnumpyrr rrrrrÚs