U md@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_equalselfxyfvalsrb/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/distributions/tests/test_ecdf.pytest_StepFunction s     z#TestDistributions.test_StepFunctioncCsHtd}td}ttt||td}td}ttt||dS)Nr)rr)rrr ValueErrorrZzeros)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.r/r0r1r2cSsd|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)rrr8rrinfr)rree1e2rrrtest_ecdf_discrete0s z$TestDistributions.test_ecdf_discreteN) __name__ __module__ __qualname__r r#r&r4r:r?rrrrrs r) numpyrZ numpy.testingtestingrrZstatsmodels.distributionsrrrrrrrrs