U Kv f@sZddlZddlmZddlmZmZddlm m m Z ddl mZGdddZdS)N)mv_mixture_rvsMixtureDistribution)statsc@s<eZdZddZddZddZddZd d Zd d Zd S)TestMixtureDistributionsc Cs|tjdt}|jddgdtjtjgtdddtdddfd }tj t | | | gt dd dgdd dS) Nr??i?locZscaledistkwargsg࿩decimal)nprandomseedrrvsrnormdictnptassert_almost_equalarraystdmeanvarselfmixresr"P/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/distributions/tests/test_mixture.pytest_mixture_rvs_random s z0TestMixtureDistributions.test_mixture_rvs_randomcCstdddgdddgdddgg}tdddg}td ddg}t||}t||d}tjd td dgd ||gd }tjt| | | gtdddgdddS)N?r r?333333?@rrg?igCl?g|?5^?gjt @r r) rrmvdMVNormalrrrrrrrrrZcov3muZmu2Zmvn3Zmvn32r!r"r"r#test_mv_mixture_rvs_randoms   z3TestMixtureDistributions.test_mv_mixture_rvs_randomcCspt}tddd}|j|ddgtjtjgtdddtd d dfd }t|t d d ddddddddg dS)Nr* UUUUUU?UUUUUU?rrr r rr g\=g|^r|>g)y) ?gճY?gc.Ѣ?g3@?gR?gZ?gfƂ.{?gg^d*=>gNڲş?g}u"5?gkE?gU*?gy@?g%=B?g?) rrr5Zcdfrrrrrrr6r"r"r#test_mixture_cdf3s" z)TestMixtureDistributions.test_mixture_cdfc7Cst}tjd|jddgdtjtjgtdddtdddfd }t |t d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;g2dS)geg"޿gt݄g'(,V?gzz?gW$gOXE, gχvgh6BHg>>?gm>gg0{gO gnahgtEp?gˆ\^g~U8g)J׆ig'^gb[ّg6?g Ag8񴀹տgBӎgѕE^g3pg[۸gt%?gK=g`T| ?gQ gP[tӿg46lBg޸?gZD5 3g 9g3) rrrrrrrrrrrrr"r"r#test_mixture_rvs_fixed?sp  z/TestMixtureDistributions.test_mixture_rvs_fixedcCstjdtdddgdddgdddgg}tdd dg}td ddg}t||}t||d }td d gd||gd}t|tdddgdddgdddgdddgdddgdd d!gd"d#d$gd%d&d'gd(d)d*gd+d,d-gg dS).Nr9r%r rr&r'r(rr)r*g?g?r2r+gC߼οg w?g?gof A*@gU?g>Bs