ó áp7]c@sdZddlZddlmZmZddljjZ ddl m Z de fd„ƒYZ de fd„ƒYZd e fd „ƒYZdS( s´Test for a helper function for PanelHAC robust covariance the functions should be rewritten to make it more efficient Created on Thu May 17 21:09:41 2012 Author: Josef Perktold iÿÿÿÿN(t assert_equalt assert_raises(t GroupSortedtCheckPanelLagMixincBs)eZed„ƒZd„Zd„ZRS(cCsWt|jƒ|_}gtdƒD]'}|tj|j||jƒf^q#|_dS(Ni( Rtgindtgtrangetswt lagged_groupstxtgroupidxtalla(tclsRtlag((sElib/python2.7/site-packages/statsmodels/stats/tests/test_groups_sw.pyt calculatescCsLxE|jD]:\}\}}t||j|jƒt|||ƒq WdS(N(R RtalletT(tselfR ty0tylag((sElib/python2.7/site-packages/statsmodels/stats/tests/test_groups_sw.pyt test_valuesscCs/|j}tttj|j||jjƒdS(N(tmlagRt ValueErrorRRR RR (RR((sElib/python2.7/site-packages/statsmodels/stats/tests/test_groups_sw.pyt test_raisess (t__name__t __module__t classmethodRRR(((sElib/python2.7/site-packages/statsmodels/stats/tests/test_groups_sw.pyRs t TestBalancedcBseZed„ƒZRS(cCs]tjdddgdƒ|_d|_tjdƒ}|d|j7}|dd…df|_itjddddddd d d d d ddddggƒd6tjddddd d d d ddddg gƒd6tjdddd d d dddg gƒd6tjddd d ddggƒd6tjdd dggƒd6|_|j ƒdS(Niiiiii iiiiiiinioipiqir( tnptrepeatRRtarangetNoneR tarrayRR(R R ((sElib/python2.7/site-packages/statsmodels/stats/tests/test_groups_sw.pyt setup_class%s $$.%"(RRRR!(((sElib/python2.7/site-packages/statsmodels/stats/tests/test_groups_sw.pyR#stTestUnBalancedcBseZed„ƒZRS(cCs¹tjdddgdddgƒ|_}d|_tjdƒ}|d|7}|dd…df|_itjddddd d d d d dddddddddggƒd6tjddd d d d dddddddddggƒd6tjdd d d ddddddddg gƒd6tjd d dddddddg gƒd6tjd ddddddggƒd6tjdddddggƒd6|_|j ƒdS(Niiiiii ii iiiiiliminioipiqirisitiui( RRRRRRR R RR(R RR ((sElib/python2.7/site-packages/statsmodels/stats/tests/test_groups_sw.pyR!:s + $%$$.(((RRRR!(((sElib/python2.7/site-packages/statsmodels/stats/tests/test_groups_sw.pyR"8s(t__doc__tnumpyRt numpy.testingRRt%statsmodels.stats.sandwich_covariancetstatstsandwich_covarianceRtstatsmodels.tools.grouputilsRtobjectRRR"(((sElib/python2.7/site-packages/statsmodels/stats/tests/test_groups_sw.pyt s