U
    KvfK                     @   s   d dl Zdd ZG dd dZG dd dZG dd	 d	ZG d
d dZG dd dZG dd dZG dd dZ	G dd dZ
dS )    Nc                 C   s@   t | } tt t| }t jt j| ||ddddddS )zq
    A convenience function to make the SAS covariance matrix
    compatible with statsmodels.rlm covariance
       )Zaxisr   )npZasarrayintsqrtlenZrollZreshape)ZarrZside r   P/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/robust/tests/results/results_rlm.py_shift_intercept   s    
r
   c                   @   s   e Zd ZdZdd Zdddddd	d
ddd
ddddddgZeeZddddddddddddddddgZeeZ	ddd d!d"d#d$d%d d$d&d'd(d%d'd)gZ
ee
Zd*S )+Huber
    c                 C   s(  t ddddg| _t ddddg| _d	| _t d
d
ddd
d
d
d
d
d
d
d
d
d
d
d
d
d
d
d
dg| _t ddddddddddddddddddd d!d"g| _d#| _d$| _t d%d&d'd(gd&d)d*d+gd'd*d,d-gd(d+d-d.gg| _	t d/d0d1d2d3d4d5d5d6d7d8d9d:d;d<d=d>d?d@dAdBg| _
t dCdDdEdFg| _d S )GNg;?g 5?gw|\]g
^eDg45Pv?g۵	k?g=~?g\#@g=@      ?g%%?gy(?g@?gf@gޞ gov@gX4@g\JeVg
g̩Kg</Sfgg8		g@K?g	E?g;v^0?ggwFg!@g$j@?g7.퓨gެ?gnO?gza?g_Ky!      禺kS\?Ynlh,_@FV?牖.n^? 3\ܔ>q}^c?HܾeĿPE*@g浯yC@g8,C@gH(Qi@@gr5@gʥT6j3@g_Ay4@gzɜ44@g:A8A1@gx]	+@g_#\(@g9Q"#'@gy*@g'Nn)@g(#@g\iRw@g>!@gntB@g};y!@g0*@gͥ<7@g3Q@gm%P7k@g'Ԩ4gpr   arrayparamsbsescaleweightsresiddf_modeldf_residZbcov_unscaledZfittedvaluesZtvaluesselfr   r   r	   __init__   s                                        	                zHuber.__init__g}8gW@gD?g3VCܿg\r),gm{?gAh:;?gKqUg3fsglxz,Cܿg =~?h㈵>g<,gm{?g=UT@gM~T?gPkwοg^2gjMS?g}9?gCfڞg3.lguοgD4ؙ?gm2[gd;Og"?gׁsQ@g,g~5gvq-gb(goŏ1g?gPyg7T7`g)Ǻ?g#Gkg(geN?N)__name__
__module____qualname____doc__r%   Zhuber_h1r
   h1Zhuber_h2h2Zhuber_h3h3r   r   r   r	   r      sp   (                                       r   c                   @   s   e Zd ZdZdd Zdddddd	d
ddd
ddddddgZeeZddddddddddddddddgZeeZ	ddd d!d"d#d$d%d&d$d'd(d)d%d(d*gZ
ee
Zd+S ),Hampelr   c                 C   s(  t ddddg| _t ddddg| _d	| _t d
d
d
d
d
d
d
d
d
d
d
d
d
d
d
d
d
d
d
d
dg| _t ddddddddddddddddddddd g| _d!| _d"| _t d#d$d%d&gd$d'd(d)gd%d(d*d+gd&d)d+d,gg| _	t d-d.d/d0d1d2d3d3d4d5d6d7d8d9d:d;d<d=d>d?d@g| _
t dAdBdCdDg| _d S )ENgr?g]D?gif¿gF<Dgm A?gGZD?g#48 ?gVK'@gMĴ@r   g/?g_xF\@g3 gMst@g*{;%@gfq{87gb9[gRg.zrg۞1g9Q?g56C#@gv0@g)gK(xпg*@gu]>L?gE1g&K0׿gk,ܿg};"}?gGlr   r   r   r   r   r   r   r   r   r   r   r   gz2wC@gQ:ÚC@goQ@@gR!66@gs3@g,gkT4@gY*W5@gԸy32@gX)@gr2/7'@gHs%@g8@)@gEc(@gj@gQT-xv@g)"&U"@g1_	 @g^c"@g0[0+@gt[6@g@'@g.RC
@g.v`gᇵ9r   r#   r   r   r	   r%   P   s                                        	               zHampel.__init__gSa@gG`?gO@age6g|?5^?ga?g	m9⪢g X4}gEg6XO?r&   gbX9ge1?g-`@gN?g9]׿g}gK?ggs?gwn\g	m9wgV-׿g:M?gggoʡgKqUٗ?gx`@g:?g(~k	grmogS㥫?g'IL?g=
YgXjpg~jtg67'?gQ,zg;Ong؞Y?N)r'   r(   r)   r*   r%   Z	hampel_h1r
   r+   Z	hampel_h2r,   Z	hampel_h3r-   r   r   r   r	   r.   M   sp   &                                       r.   c                   @   s   e Zd Zdd Zddddddd	d
dd	dddd
ddgZeeZddddddddddddddddgZeeZdddddddd ddd!d"dd d"d#gZ	ee	Z
d$S )%BiSquarec                 C   s(  t ddddg| _t ddddg| _d	| _t d
ddddddddddddddddddddg| _t dd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3g| _d4| _d5| _t d6d7d8d9gd7d:d;d<gd8d;d=d>gd9d<d>d?gg| _	t d@dAdBdCdDdEdFdFdGdHdIdJdKdLdMdNdOdPdQdRdSg| _
t dTdUdVdWg| _d S )XNgAw?gOk`?g -gb1$EgyMl?g߁*A?gW?g@w)#@g8|ObMA@gIrLX?gfQ?gߗK?g}?g@G?gz"?g(z?gM?g3c?gj*?g9˪?g$W51?gהT;?g>{E?gy]?glN?g叒?g:?g6w?g" ?g_àDb?g@gxAg1T|@g[@g2@oAgjUgQKg;FW.ٿggMg!b{KοgPx?gѯ)g?gBg'Rtg@gQ筲]?gQ>׿gS$?gsEu&?g
8?g:D+$r   r   r   r   r   r   r   r   r   r   r   r   g!0AӾC@g7*4C@gU:@@g`C5@gX3@gWj4@g^d4@g|o޴0@gQ,uz,@gzt*@gr,a)@gQ9Dn,@ge\*@g %4;@g
CI@g @g.S@g= @g^*@g"o9@gQY;+!@g,@g]Yl?g5bϾr   r#   r   r   r	   r%      s                                     	               zBiSquare.__init__gŏ1wV@g?g0ڿg;fg9m4?gsƇ?gխޗvꭁrgeaڿgQi>"?r&   g1w-!g9(a?gB=P@g]j~^?gfҿgwJg P\?gL֝gK8lg˞6?gMbPgL?g~rH@g&z<?gV`Ŀg: g3k) ?g:fg+~7egWۼ?gt^cZg۠[;?N)r'   r(   r)   r%   Zbisquare_h1r
   r+   Zbisquare_h2r,   Zbisquare_h3r-   r   r   r   r	   r/      sn   )                                       r/   c                   @   s   e Zd Zdd Zdd Zdddddd	d
ddd
ddddddgZeeZddddddddddddddddgZeeZ	ddddddddddd d!ddd!d"gZ
ee
Zd#S )$Andrewsc                 C   s   ddddg| _ ddddg| _d	| _d
| _d| _ddddddddddddddddddddd g| _d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5g| _d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJg| _d S )KNgݵ|г?g??g>W[g/$%Eg^)?gݓ?g_v?g6<R"@gv=@g      @g      1@gn?P@g3g/^p@gA
zŻ@g|vJgnMd-Vg|IgO&ٿg4ƱFg+YCMϿgX?g.W\b?gu(g7g͔@gpW?gcӷlw׿gNyC?gjq;?g^/ ?gsP$gϡ4f?gCNg#?g6''S@g!6gQr*g&lg5Kԛƿg2+Ajg|ugf[_®?gA ?gG
F1gd)!gr6?giDk?gLĿg>eWG>?g.E?g&Sk?gfdSgJ}g?gtd>?gNd>?gV?)?g񺒥=?g "j?gxi?gWCe?g,\?gyb%>?gMnW?g?g?g:VA?g[hnR?gp_}?giR?gF?gۅ Fr?g.?r   )r   r   r   r!   r"   r    sresidsr   r#   r   r   r	   r%      s                   	               	                zAndrews.__init__c                 C   s   ddddgS )N)g镲?gV-?)gڊe?gX9v?)g<,Ԛֿgb=y?)gbX9PNgS㥛7r   r#   r   r   r	   conf_int   s    zAndrews.conf_intgHU@gY!?g|ٿgeS.g
ףp=
?gs,&6gQ,rgrar?g.>g?ghUMP@gl/ں?g?Կg*Pgދ/?g]o%gH}mgNP?g_LEgCup?g]OH@goI?gϽϿgV	g Z+ڌ?g;Dgvӂgg¹?gC8
Qg.u׃I?N)r'   r(   r)   r%   r3   Z
andrews_h1r
   r+   Z
andrews_h2r,   Z
andrews_h3r-   r   r   r   r	   r1      sp                                          r1   c                   @   s   e Zd Zdd Zdd ZdS )
HuberHuberc                 C   sH  dddddddddddd	ddd	d
g| _ t| j | _ ddddddddddddddddg| _t| j| _ddddddddddddddddg| _t| j| _dd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3g| _d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHg| _dIdIdIdJdIdIdIdIdIdIdIdIdIdIdIdIdIdIdIdIdKg| _dL| _dM| _dN| _	dO| _
dP| _d S )QNgSt$\@gc]K?g"gnض(gX_"?g׻?gxĬwgKTo?gE/ӑ>gu=?g	hY@g[v?g͒ 5lտg`g<HO?g+Hgqg6U?gnagOyt#,*?gHV@g,d?ghW!gQg *?gdg"fgX zR&5?g3krgѓ2?gՕ%E@g x	g%HZ@gB+@g+-zg^чPgJgXs6 gΰ:lgIga[?gd'Р?g2(?):?g牷3? g9gr@gd<k3%?gXϙgHgiUrgV<?gg* g/A?g$^ɞg\?g~ q?gPGgܷggk{tӿg0gU$?g׭ۗ?gH]"?g{cԧg'-	ǿg1,f?gbAO4?g-'ԿgR]g ED"yVg)>?g~ڢr   gh!1?gD㢽n?)g rh?g(\?gDJgfjD)g:M?gv/?gH.?gQIf%@gz6>W[
@r   r   r+   r
   r,   r-   r    r2   r   r   r   r   r!   r"   r#   r   r   r	   r%      s                                                                                             zHuberHuber.__init__c                 C   s   ddddgS )N)g|гY?g6<R?)g:M?gFx$?)g48EGڿgCl?)gOgY 4r   r#   r   r   r	   r3   %  s     zHuberHuber.conf_intNr'   r(   r)   r%   r3   r   r   r   r	   r4      s   'r4   c                   @   s   e Zd Zdd Zdd ZdS )HampelHuberc                 C   sH  dddddddddddd	ddd	d
g| _ t| j | _ ddddddddddddddddg| _t| j| _ddddddddddddddddg| _t| j| _dd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3g| _d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHg| _dIdIdIdIdIdIdIdIdIdIdIdIdIdIdIdIdIdIdIdIdJg| _dK| _dL| _dM| _	dN| _
dO| _d S )PNg7[ ob@g3.?g?RDUgOjMgiq0'h?g9#J{g~tgy~gjh?g&|>g֩=#?g~ja@gj=&R?g4Opؿg82gE
)??gbEagVF?xg.s/?g5)^hgV
\?g(~k`@g鷯?g_5!gp:ǀgGoܚ?gKgTpgnʆ5?g_Q{gJ_?g?l|	@gր- gA!?@g_)@gjbgK_g>g|;<gP2׋g[Eg?g6@gqnF:@g*˚gJƿgL@ggF?g:0g:?ٿg`TE߿gT57b
?g68gBOx?g~|gl?gv0y;v?gΝ=agpmz2g8dg]h|ڿg	J4ggl?gݲR'1?gWQ;1?g[VRK޿g66go?g88	c?gWۿg,ľg+ήeÿgE:s?g#GJr   g{A?)gBfj?gc]F?g	c¿gmV}"D)g`TR'?g]Fx?gY8m?gaTR'I(@g47B
@r   r   r5   r#   r   r   r	   r%   +  s                                                                                            zHampelHuber.__init__c                 C   s   ddddgS )N)g oŏ?gǺ?)gX9v?gz):?)gQ|ݿg0*?)gԚPgcZB>x0r   r#   r   r   r	   r3   S  s     zHampelHuber.conf_intNr6   r   r   r   r	   r7   *  s   (r7   c                   @   s   e Zd Zdd Zdd ZdS )BisquareHuberc                 C   sH  dddddddddddd	ddd	d
g| _ t| j | _ ddddddddddddddddg| _t| j| _ddddddddddddddddg| _t| j| _dd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3g| _d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHg| _dIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]g| _d^| _d_| _d`| _	da| _
db| _d S )cNg؁sF>`@g0~?ge`TR'geag=#?g*t^czg&OYM׿?g+<->gg%?gw/q[@g>Ȳ`?g+MJAпg(~kg'?gGŧ gY4rg"J?g{jgƟlX?g˜.V@gCVzNzg6?g-	PSgfHū?g~@gj>"Dbgc!:?gZӼ}g?g'CwG@g{- ghc@g54>@gYtg&IKg gK)I)g=ڹgz?gө?gLaHC?gc( g0'8)go@gP?ggް{`gu.͢gΤ?gH: gb?gQgNy?g(m~?grugjrxgVdŇvg| :ӿgGgo\l?gf׉?g8?gNg9l0ǿgߠ7?g'dk?g3b'yԿgrg=+J	g(N=|?ghg?g'V?g.w$fg?g?geZG8?gxp?g[$>?gh{n?g;?gޢM?gDd?g=3?gڍ̐?goX?gYl?gɆ?gEʳ?gE!:?gFi?gzl?g[t?)gea?gN@a?gJ+g_LrD)aTR'?g&S?"lxz,?g镲&@g镲q
@r   r   r5   r#   r   r   r	   r%   Y  s                                                                                           zBisquareHuber.__init__c                 C   s   ddddgS )N)gfc]F?gX9v?)g7A`?g:pΈ?)K=Uۿg{Gz?)gzOg46<2r   r#   r   r   r	   r3     s     zBisquareHuber.conf_intNr6   r   r   r   r	   r8   X  s   (r8   c                   @   s   e Zd Zdd Zdd ZdS )AndrewsHuberc                 C   sH  dddddddddddd	ddd	d
g| _ t| j | _ ddddddddddddddddg| _t| j| _ddddddddddddddddg| _t| j| _dd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3g| _d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHg| _dIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]g| _d^| _d_| _d`| _	da| _
db| _d S )cNg|a2=`@g`6?g%g~@gsd?ga)r9   gȮԿ?gOڟ+>g؁sF?gS㥛p[@gJ\Ǹ?g$~пgdzg?,Օ?g^2᧿r0   g_5?g&KjgaO;5Y?gTƿV@gXni5$g
)??gn/igOGɫs?g)sgHPsbg:>Z1?gǵb|g![?gnd5R@g:ӧ g/E@g7B@gmgPHg?t g~4f'gE᲻g4?g /?g,U?go" g2+1gCF@gԠe?g|1>gƿgrgH~?g˝dnh gRom?gվ3Bg5;B~?g\?g#v0ge~0Qg_Ov^sg3ӿgG񏖁g|g?gAMv?g %ž?gK3gz0ǿg?gd0MS?g|vӁԿg =8gx27g.ԦY|?ghߐbg?gйa/?go@mV?g-?giL84?gf}??g?gM,?gGD4?giU?g^i7?g~f?gY-֎?g M?g儦x?gLa?grx	?gK;?gU\Ӿ?gCi?g
s_?)g^?g^)?go_gpΈpD)r:   gTt$?r;   g$(~&@gHPs
@r   r   r5   r#   r   r   r	   r%     s                                                                                         zAndrewsHuber.__init__c                 C   s   ddddgS )N)g$C?g#~j?)gzG?gͪ?)r<   gPkw?)gjMSOg<,Ԋ2r   r#   r   r   r	   r3     s     zAndrewsHuber.conf_intNr6   r   r   r   r	   r=     s   *r=   )Znumpyr   r
   r   r.   r/   r1   r4   r7   r8   r=   r   r   r   r	   <module>   s   
=;<9-..