M/Ph} dddlZddlZddlmZmZddlmZddlmcm Z GddZ dZ dS)N)assert_almost_equal assert_equal)Contrastc<eZdZedZdZdZdZdZdS) TestContrastctjdtjd|_dS)Ni1)( )numpyrandomseedstandard_normalX)clss e/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/stats/tests/test_contrast.py setup_classzTestContrast.setup_class s3 %    ,,W55ctj|jdddf|jdddff}t||j}dgdgdzzdgdzdgzdgdzzg}t ||jdS)Nr )np column_stackrrrcontrast_matrix)selftermc test_contrasts rtest_contrast1zTestContrast.test_contrast1sqqqs TVAAAaC[9:: T46 " "s1uqc!eqckQCE&9: M1+<=====rctjd}tj||jdddff}t ||j}dgdzdgzdgdzz}t ||jdS)N)r rrrr)rzerosrrrrr)rzerorrrs rtest_contrast2zTestContrast.test_contrast2swxdfQQQqSk233 T46 " "A qc!e+ M1+<=====rctj|jtj|j}tjd|z }tj|t jd}tj ||jdddff}t||j}t|j j ddS)Nr )r r)r )rdotrlinalgpinvidentityr r rrrrrshape)rPresidnoiserrs rtest_contrast3zTestContrast.test_contrast3s F4629>>$&11 2 2 B!#uel::7CCDDtvaaac{344 T46 " "Q&,e44444rctj|j|jdddff}t|jdddf|}dS)Nr&)rrrr)rX2rs rtest_estimablezTestContrast.test_estimable$sG _dfdfQQQqSk2 3 3 TVAAAaC[ $ $rN) __name__ __module__ __qualname__ classmethodrr r$r/r2rrrrsi66[6>>> >>>555%%%%%rrc tjddd}tjgdgdgdgdgd gd g}tj|}t ||tjddd }t ||tjgd gd gdgdgdgdg}tj|dd}t ||tjddd}t ||dS)N)k)?r>)r>r=r>)r>r>r=)r=r>)r?r>r=)r>r?r=r)r>r=r>r>r>r>)r>r>r=r>r>r>)r>r>r>r=r>r>)r>r?r=r>r>r>)r>r?r>r=r>r>)r>r>r?r=r>r>r)k_params idx_start)reyearraysmc_constraints_factorr_contrast_pairs)cm_cpairsc0c1cpairs2s rtest_constraintsrM*s> &A   C X&&&&& ())F  % %BV Q1 % %BVh666666666666666666 899G qA > > >BW Q1 % %BWr) r r numpy.random numpy.testingrrstatsmodels.stats.contrastrstatscontrastrErrMr7rrrSs;;;;;;;;//////(((((((((%%%%%%%%Dr