U md@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?locscaledistkwargsg࿩decimal)nprandomseedrrvsrnormdictnptassert_almost_equalarraystdmeanvarselfmixresr#e/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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) rrmvdMVNormalrrrrrrrrr Zcov3muZmu2Zmvn3Zmvn32r"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 rrg\=g|^r|>g)y) ?gճY?gc.Ѣ?g3@?gR?gZ?gfƂ.{?gg^d*=>gNڲş?g}u"5?gkE?gU*?gy@?g%=B?g?) rrr6Zcdfrrrrrrr7r#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).Nr;r&r rr'r(r)rr*r+g?g?r3r,gC߼οg w?g?gof A*@gU?g>Bs