U md0@s:dZddlZddlmZddlmZddlmZm Z m Z ddl m m mmZddlmZeZeeje_eeje_eejdddfejdddf<eejdd Zejd ZeeeddZGd d d ZGd ddeZGdddeZGdddeZGdddeZGdddeZdS)z= Created on Thu Aug 3 21:08:49 2017 Author: Josef Perktold N)assert_allclose)load)NegativeBinomialNegativeBinomialPPoisson) add_constantF)prependc@seZdZdZddZdS)CheckMarginMixinr cCs|j}|j}|j}t|jj|jj|d|dt|jj|jj|d|dt|jj|jj|d|dt|j |j |ddfd|ddS)Ngh㈵>)Zrtolgư>gh㈵>) res1 res1_slicertol_facrmargeffparamsZ margeff_seZbseZpvaluesZconf_intZ margins_table)selfrslrfr`/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/discrete/tests/test_margins.pytest_margins_table sz#CheckMarginMixin.test_margins_tableN)__name__ __module__ __qualname__rrrrrrr sr c@seZdZeddZdS)TestPoissonMargincCs\dddddddg}ttt}|j|d}|}||_||_d |_tddd|_ t j |_ dS) Nm4W,@y&1?3ı.n sh|?ۊe @37gj+6 start_paramsr ) rendogexogfit get_margeffresrrslicer res_stataZresults_poisson_margins_contrclsr%Zmod_poiZres_poiZ marge_poirrr setup_class-s   zTestPoissonMargin.setup_classNrrr classmethodr/rrrrr+src@seZdZeddZdS)TestPoissonMarginDummycCs^dddddddg}ttt}|j|d}|jd d }||_||_d d d dddg|_tj |_ dS)Nrrrr r!r"r#r$Tdummyrr rr ) rr&r'r(r)r*rrr,Zresults_poisson_margins_dummyrr-rrrr/@s    z"TestPoissonMarginDummy.setup_classNr0rrrrr2>sr2c@seZdZeddZdS)TestNegBinMargincCsbddddddddg}ttt}|j|d d d }|}||_||_tddd|_t j |_ d |_ dS N.1f*@矫_v?灕Clg镲 ׁsF@xz,C! ףp= = ףp=?nmr%methodmaxiterI@) rr&r'r(r)r*rr+rr,results_negbin_margins_contrrr.r%modr*Zmargerrrr/Qs  zTestNegBinMargin.setup_classNr0rrrrr7Osr7c@seZdZeddZdS)TestNegBinMarginDummycCspddddddddg}ttt}|j|d d d }|jd d }||_||_ddddddg|_|_tj |_ d|_ dS)Nr9r:r;r<r=r>r?r@rArBrCTr3rr r5rr6r rF) rr&r'r(r)r*rrr,Zresults_negbin_margins_dummyrrrHrrrr/ds   z!TestNegBinMarginDummy.setup_classNr0rrrrrJbsrJc@seZdZeddZdS)TestNegBinPMargincCsbddddddddg}ttt}|j|d d d }|}||_||_tddd|_t j |_ d |_ dSr8) rr&r'r(r)r*rr+rr,rGrrrHrrrr/ws  zTestNegBinPMargin.setup_classNr0rrrrrKtsrK) __doc__numpynpZ numpy.testingrZstatsmodels.datasets.cpunishrZ#statsmodels.discrete.discrete_modelrrrZ8statsmodels.discrete.tests.results.results_count_marginsZdiscretetestsresultsZresults_count_marginsr,Zstatsmodels.tools.toolsrZ cpunish_dataZasarrayr'r&logroundmaxr rr2r7rJrKrrrrs&   &