o 6aM@sBddlZddlmZmZmZddlmZddlZGdddZdS)N)assert_assert_array_equal assert_raises)randomc@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestRegressioncCsNtdddD]}tj|dd}tt|tj ko"t|tjkqdS)Ngg@2)nplinspacermtrandvonmisesrallpi)selfmurrD/usr/lib/python3/dist-packages/numpy/random/tests/test_regression.pytest_VonMises_range s(z"TestRegression.test_VonMises_rangec Cstttjjddddddktttjjddddddkdg}tjd k}|r6tjd kr6|d |D] }ttjj|dkq8dS) N sizer)rrlwin32)rr) rr rrhypergeometricsysmaxsizeplatformappend)rargs is_64bitsargrrrtest_hypergeometric_ranges""  z(TestRegression.test_hypergeometric_rangecCsd}tjdtjjd|d}t|dkt|}d|dd}t|d k|t|d kt|}d|dd }t|d k|dS) Nrg?rrzFrequency was fz, should be > 0.45g?z, should be < 0.23gq= ףp?)r rseed logseriessumfloatr)rNrvsnfreqmsgrrrtest_logseries_convergence"s z)TestRegression.test_logseries_convergencecCs~gdgdgdgdfD].}tjdt|}t|tj|d|d|d|d gtd }ttj|td |qdS) N)rr+rN)rrr+r+rrN)rr6r7N)r5r+rNi90rrrr+dtype)r rr,listshufflearrayobjectr)rtshuffledexpectedrrrtest_shuffle_mixed_dimension2s  &z+TestRegression.test_shuffle_mixed_dimensioncCs`tj}tgd}tdD]}tj||dt|jddtddd|qdS)N) rr+r rrCrrrirg$@)rp) r r RandomStater<ranger,rchoiceones)rmresirrrtest_call_within_randomstate>s    "z+TestRegression.test_call_within_randomstatecCsXtjjdgdggddtjjdgdggtddtjjdgdggtdddS)Nrrr)r rmultivariate_normalint_int64rrrr#test_multivariate_normal_size_typesHs"z2TestRegression.test_multivariate_normal_size_typescCs:tjdtjjdddd}ttt| ddS)NiIg-C6?drzNans in np.random.beta)r rr,betaranyisnan)rxrrrtest_beta_small_parametersPs z)TestRegression.test_beta_small_parameterscCstjdgd}gd}tjtjtjfD]'}tj||dt|}tjj||d}t ||vt t tjj||ddqdS)N)rr+r)rrr+r8)rEg?) r rr,float16float32float64r<r.rHrr ValueError)racountsdtprobscrrr"test_choice_sum_of_probs_toleranceWs  z1TestRegression.test_choice_sum_of_probs_tolerancecCsHtjdtddg}tdD]}tj|qddl}|dS)NrYr^aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaarSr)r rr,r<rGr;gccollectrr^_rerrr1test_shuffle_of_array_of_different_length_stringsds   z@TestRegression.test_shuffle_of_array_of_different_length_stringscCsXtjdtjtdtdgtd}tdD]}tj|qddl}| dS)NrYrrr8r)r) r rr,r<aranger=rGr;rerfrgrrr test_shuffle_of_array_of_objectsss   z/TestRegression.test_shuffle_of_array_of_objectscCsGdddtj}tjdtd|}tj|}t|tgdt|td|Gddd}tjd|}tj|}t|tgdt| td dS) Nc@s eZdZdS)z3TestRegression.test_permutation_subclass..NN)__name__ __module__ __qualname__rrrrr0sr0rr)rr+rc@seZdZedZddZdS)z3TestRegression.test_permutation_subclass..MrcSs|jS)N)r^rQrrr __array__sz=TestRegression.test_permutation_subclass..M.__array__N)rlrmrnr rjr^rorrrrMs  rp)r+rrrrr) r ndarrayrr,rjview permutationrr<ro)rr0origpermrprJrrrtest_permutation_subclasss    z(TestRegression.test_permutation_subclassN)rlrmrnrr(r4rArMrRrXrcrirkrvrrrrr s    r) r! numpy.testingrrrnumpyrr rrrrrs