B <ÓFdã@szddlmZmZmZddlZddlmZmZmZm Z m Z ddl m Z ddl mZddl ZGdd„deƒZedkrveƒdS) é)ÚdivisionÚabsolute_importÚprint_functionN)ÚTestCaseÚrun_module_suiteÚassert_Úassert_array_equalÚ assert_raises)Úrandom)Úlongc@sdeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dS)ÚTestRegressioncCsRxLt ddd¡D]:}tj |dd¡}tt |tj k¡oFt |tjk¡ƒqWdS)NgÀg@ééé2)ÚnpZlinspacer ZmtrandZvonmisesrÚallZpi)ÚselfZmuÚr©rúU/opt/alt/python37/lib64/python3.7/site-packages/numpy/random/tests/test_regression.pyÚtest_VonMises_range sz"TestRegression.test_VonMises_rangec Cs’tt tjjddddddk¡ƒtt tjjddddddk¡ƒdg}tjd k}|rltjd krl| d ¡x |D]}ttjj|ŽdkƒqrWdS) Nééé é )Úsizeér)iþÿiþÿiþÿlZwin32)lþÿÿlþÿÿlþÿÿ) rrrr ZhypergeometricÚsysÚmaxsizeÚplatformÚappend)rÚargsZ is_64bitsÚargrrrÚtest_hypergeometric_ranges""   z(TestRegression.test_hypergeometric_rangecCs|d}tj d¡tjjd|d}t |dk¡t|ƒ}d|}t|dk|ƒt |dk¡t|ƒ}d |}t|d k|ƒdS) Nièrgš™™™™™é?)rrz"Frequency was %f, should be > 0.45gÍÌÌÌÌÌÜ?éz"Frequency was %f, should be < 0.23gq= ×£pÍ?)rr ÚseedZ logseriesÚsumÚfloatr)rÚNZrvsnZfreqÚmsgrrrÚtest_logseries_convergence#s z)TestRegression.test_logseries_convergencecCsBtj d¡tj d¡}tj d¡tj tdƒ¡}t||ƒdS)NiÒé )rr r%Z permutationr r)rÚaÚbrrrÚtest_permutation_longs3s    z%TestRegression.test_permutation_longscCszxtddddgddddgddddgddddggD]D}tj d¡t|ƒ}t |¡t||d|d|d|dgƒq.WdS) Nrr$r)rr)r$r$)rri90r)rr r%ÚlistÚshuffler)rÚtZshuffledrrrÚtest_shuffle_mixed_dimension:s     z+TestRegression.test_shuffle_mixed_dimensionc Csttj ¡}t ddddddddddg ¡}xFtdƒD]:}tj |¡| d ¡t|jd d t d ¡d d |ƒq2WdS) Nréér$ré rriárg$@)rÚp) rr Z RandomStateÚarrayÚranger%rÚchoiceZones)rÚmZresÚirrrÚtest_call_within_randomstateEs    z+TestRegression.test_call_within_randomstatecCsXtjjdgdggddtjjdgdggt d¡dtjjdgdggt d¡ddS)Nrr)r)rr Zmultivariate_normalZint_Zint64)rrrrÚ#test_multivariate_normal_size_typesOsz2TestRegression.test_multivariate_normal_size_typescCs:tj d¡tjjdddd}tt t |¡¡ dƒdS)NiÒ–Ig-Cëâ6?éd)rzNans in np.random.beta)rr r%ZbetarÚanyZisnan)rÚxrrrÚtest_beta_small_parametersWs z)TestRegression.test_beta_small_parameterscCsˆtj d¡dddg}dddg}xbtjtjtjfD]N}tj||dt|ƒ}tjj||d}t ||kƒt t tjj||ddq2WdS) NiÒrr$rr)Zdtype)r6gÍÌÌÌÌÌì?) rr r%Zfloat16Zfloat32Zfloat64r7r&r9rr Ú ValueError)rr,ZcountsZdtZprobsÚcrrrÚ"test_choice_sum_of_probs_tolerance^s    z1TestRegression.test_choice_sum_of_probs_tolerancecCsLtj d¡t ddg¡}xtdƒD]}tj |¡q$Wddl}| ¡dS)NiÒr,Aèaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar>r)rr r%r7r8r0ÚgcÚcollect)rr,Ú_rErrrÚ1test_shuffle_of_array_of_different_length_stringsks  z@TestRegression.test_shuffle_of_array_of_different_length_stringscCsXtj d¡t t d¡t d¡g¡}xtdƒD]}tj |¡q0Wddl}| ¡dS)NiÒrrièr) rr r%r7Zaranger8r0rErF)rr,rGrErrrÚ test_shuffle_of_array_of_objectszs  z/TestRegression.test_shuffle_of_array_of_objectsN)Ú__name__Ú __module__Ú __qualname__rr#r*r.r2r<r=rArDrHrIrrrrr s   r Ú__main__)Z __future__rrrrZ numpy.testingrrrrr Znumpyr Z numpy.compatr rr rJrrrrÚs  }