U Kv f@sNddlZddlmZddlmZddlmZmZddlm Z GdddZ dS)N) assert_raises) StepFunctionmonotone_fn_inverter) ECDFDiscretec@s<eZdZddZddZddZddZd d Zd d Zd S)TestDistributionscCs`td}td}t||}|tddgddgddgg}t|dd gd d gd dggdS) N皙 @@@@rnparangerarraynptassert_almost_equalselfxyfvalsrM/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/distributions/tests/test_ecdf.pytest_StepFunction s     z#TestDistributions.test_StepFunctioncCsHtd}td}ttt||td}td}ttt||dS)Nr)rr)rrr ValueErrorrzeros)rrrrrrtest_StepFunctionBadShapes     z+TestDistributions.test_StepFunctionBadShapecCsdtd}td}t||dd}|tddgddgdd gg}t|d d gd d gd d ggdS)NrrightZsiderr r r r r rrrrrrrrrtest_StepFunctionValueSideRights    z1TestDistributions.test_StepFunctionValueSideRightc Csddddddddddg }dddd d d d d ddg }t||}t|dddddgddd d dgt||dd}t|dddddgdd d ddgdS)Nrrr rr%r&)rrr)rrrrf2rrrtest_StepFunctionRepeatedValues s  $z1TestDistributions.test_StepFunctionRepeatedValuesc Cslddddddddd d g }d d }|t|}t||}t|j|ddd t|j|ddd dS)Nr*r+r,r-r.r/r0r1r2r3cSsd|S)Ng?r)rrrr*z=TestDistributions.test_monotone_fn_inverter..)rrrrassert_array_equalrr)rrfnrrrrrtest_monotone_fn_inverter(s  z+TestDistributions.test_monotone_fn_invertercCsddddg}t|}t|jtj dddgt|jddddgtdddddg}tddddgd dddgd }t|j|jt|j|jdS) Nrr(rrg?g?g @g?r)Z freq_weights)rrr9rrinfr)rree1e2rrrtest_ecdf_discrete0s z$TestDistributions.test_ecdf_discreteN) __name__ __module__ __qualname__r r$r'r5r;r@rrrrrs r) ZnumpyrZ numpy.testingZtestingrrZstatsmodels.distributionsrrrrrrrrs