/Dj.data_iter_maker]sTF|r-r__patsy_get_model_desc__)r5r6strrrtuplehasattrrrrr,pytestraisesrrrrappendr:r<AssertionError)r8r$r9r!r%r&r'r(rBbuilderr"r#builders one_mat_fs two_mat_fsfs ` r+trPPs%   <#y*!=> |U + <?J0W / F;:F;> G  G ct|tr|dz }ttg}tr|t t gz }|D]} ||||ty#|$rYwxYw)Nr2)r5r6rrrr:r<rJ)r8r$r9excfsrOs r+ t_invalidrTsn%   9 B ~/00 ! ! lD% (! !   s AAAc ~ tgdgdgiddgdgdggdtdgdgdgfiddgdgdggdttjgdgdgiddgdgdggdtdtjgdgdgfiddgdgdggdtgdgdgd}t|iddgdgdggdtd|fiddgdgdggdtd d ggdgdgfiddgdgdggdd gd ggd gtd gd gggdgdgfiddgdgdggdd gd ggd gttjd d gtjgdgdgfiddgdgdggdd gd ggd gttjd gd ggtjgdgdgfiddgdgdggdd gd ggd gtgdgdgd}td d gd }t||fiddgdgdggdd gd ggd gt gdgdgdgfidt gdgfidt gdggdggdgfidt rtt jdgdiiddd gd gdggdgtt jgddiddd gd gdggdgtt jdgdit jdgdifiddd gd gdggdgdgdgdggdgtt jgddt jgddfiddd gd gdggdgdgdgdggdgtt jgdgt jgdggdfiddgdggdgdggdtt jgdiddd gd gdggdgt t jd ggd gt jd ggd gfidGddt}|}t|d d gd d gddggdddd d gddggd d!gd gd ggd"gGd#d$t}t |idtd%d d gddgd&ddd dgd dggd'dgd gd ggdgtd(d d gddgd&ddd dgd dggd'dgtd)d d gddgd&ddgd*gd+ggd,tgttdgg}t|dgd-iddd.gd/gd0ggdgtgtgttdgg}t|dgd-iddd d.gd d/gd d0ggd'dgtttdggtgttdgg}t|gd-gd1d2ddd d.gd d/gd d0ggd'dgd3gd4gd5ggdggttdggtgttdggf}t|d6d7}t|d|d fdgd1iddd d3gd d4gd d5ggd'dgt|d dgd1iddd d3gd d4gd d5ggd'dgt|d |d fdgd1iddd d3gd d4gd d5ggd'dgd3gd4gd5ggdggd} td8iddd d gd d gd dggd'd9gtd8d9gd1iddd d3gd d4gd d5ggd'd9gt d8id ttf:d;} | t!j"d  t d8i ttf:t!j"d fd<} | y)=Nr2rF)x0x1x2foo)default_column_prefix)foo0foo1foo2r2rWy0barbar0xrXasdf)nameyrZr[r\) )columns)x7x8x9)rey1y2r]indexceZdZdZy).test_formula_likes..ForeignModelSourcecjtttdggttdggS)NYX)rr rselfr$s r+rCzGtest_formula_likes..ForeignModelSource.__patsy_get_model_desc__Os/dL$5#6784cAR@S;T:UV Vr-N__name__ __module__ __qualname__rCr-r+ForeignModelSourcerxNs Wr-r)rzr{TX[0]X[1]rzceZdZdZy)1test_formula_likes..BadForeignModelSourcec|Srrr|s r+rCzJtest_formula_likes..BadForeignModelSource.__patsy_get_model_desc___sKr-Nr~rr-r+BadForeignModelSourcer^s r-ry ~ x)rkrh Interceptz~ xx + y)r2rXr2)r2rZrW)rrhrk)?@ @rrr rhrkrrrc"tdgdigS)NrhrVr?rr-r+z$test_formula_likes..s4#y!1 23r-)eval_env ~ x_in_envx_in_env)rRc >d}tdiddddgddgddggddgy) Nrirr2TrWrXrrrP)rs r+check_nested_callz-test_formula_likes..check_nested_calls9   VaVaV $ * %  r-c @d}tdidddgddgddggddgy) NrirTr2rWrXrrr)res r+check_nested_call_2z/test_formula_likes..check_nested_call_2s:   VaVaV $ * %  r-)rPrasarrayrrTrpandas DataFrameSeriesobjectrr rr NameErrorrrcapture) dmx_dmy_dmr foreign_modelrdesc termlistsrLrrrrs @r+test_formula_likesrsy)b!UY ,BDVW 9%&   I  Iy)*   I  rzz9i012   I  y),E JBb"aI.0HItRj"aI 68PQ Q)Y'(   I qc   #si+,   I qc   QF RZZI(>?@   I qc   aS1#J Y ,B!CD   I qc   I.e LD Ae ?D o  SAs8aX& c  dL-./048T?  "A lB :'>?"A  r-c tsytjgddgd}tjgddgd}tdd }t j |d d gd d gd d ggsJt j |jd dgsJ|jjd dgk(sJt j |jgdsJtdd \}}t j |dgdgdggsJt j |jdgsJ|jjdgk(sJt j |jgdsJt j |d d gd d gd d ggsJt j |jd dgsJ|jjd dgk(sJt j |jgdsJt|d }t j |d gd gd ggsJt j |jdgsJ|jjdgk(sJt j |jgdsJt||fd \}}t j |dgdgdggsJt j |jdgsJ|jjdgk(sJt j |jgdsJt j |d gd gd ggsJt j |jdgsJ|jjdgk(sJt j |jgdsJt|gdfd \}} t j |j|jsJt j | j|jsJtgd|fd \} } t j | j|jsJt j | j|jsJddl } | jj} d| j_tjt tddd gidd tjt tdd gd gddd | | j_y#| | j_wxYw)NrVAAr)rjrvrYBBs1r1r3r2rWrXrzs2 ~ s1rZr[r\s2)r rFrhrr)rrrrrr array_equalrorrrvrpatsy.highlevel highlevelrGrHr)rrdf1df2df3df4df5df6df7df8df9df10patsy had_pandass r+test_return_pandasrs  yt< @B yt< @B $K 0C ;;saVaVaV4 55 >>#++ T': ;; ?? ' 'K+> >> >>#))\ 22 >#++v .. ?? ' 'D6 11 >>#))\ 22 ;;saVaVaV4 55 >>#++ T': ;; ?? ' 'K+> >> >>#))\ 22 "+ .C ;;saS1#sO ,, >>#++v .. ?? ' 'D6 11 >>#))\ 22"b{;HC ;;saS1#sO ,, >>#++v .. ?? ' 'D6 11 >>#))\ 22 ;;saS1#sO ,, >>#++v .. ?? ' 'D6 11 >>#))\ 22"l+EHC >>#))RXX .. >>#))RXX ..<,+FIC >>#))RXX .. >>$**bhh //,,J 1&+# j'3qc A;W   #QC #  '1#j#s ?AS00Tc6tdd}td|}|jjgdk(sJ|jjddgk(sJt |jj dk(sJ|jj dtk(sJy)NrWaba:b)rb[T.b2]z a[T.a2]:b[b1]z a[T.a2]:b[b2]rr)r rrr term_nameslenr r )r$r#s r+test_term_infor#s a1 D % C ?? ' ',  ?? % %+u)= == s$$ % ** ??  #y 00 0r-c ngdgdtjgdtjgdgdtgdtgdtjgdt d}t d|d d d gd gd ggdgt d|d d d gd gd ggdgt d|d d d gd gd ggdgt d|d d d d gd d gd d ggddgt d|d d gdgdgdggdt d|d d gdgdgdggdt d|d d gdgdgdggdt d|d d gdgdgdggd y)!NrV)?@g@)dtype)TFT)r`rfbaz)r`r2)r2hi)rrcdrrOghz~ 0 + arTr2rWrXrz~ 0 + brz~ 0 + crz~ 0 + dzd[False]zd[True]z~ 0 + e)rrr2)r2rr)rr2r)ze[bar]ze[baz]ze[foo]z~ 0 + f)zf[1]zf[2]zf[3]z~ 0 + g)zg[bar]zg[baz]zg[foo]z~ 0 + h)zh[1]zh[foo]z h[(1, 'hi')])rrfloat32r arrayrrPrAs r+test_data_typesr1s\   ZZ  4 " y\ $ % XX+6 : Diq$!qcA3#7iq$!qcA3#7iq$!qcA3#7iq$!Q!Q!Q 8:y:QR   Iy)&    Iy)    Iy)&    Iy)* r-c ftdd}td|ddddgddgddgddggddgtd |ddddgddgddgddggdd gtd |ddddgddgddgddggdd gtd |ddgdgdgdgdggdtd|ddgdgdgdgdggdtd|ddgdgdgdgdggdt|dt|d<td|ddddgddgddgddggddgtd|ddddgddgddgddggddgtd|ddddgddgddgddggddgy) NrWrz~ C(a)rTr2rz C(a)[T.a2]z~ C(a, levels=['a2', 'a1'])zC(a, levels=['a2', 'a1'])[T.a1]z~ C(a, Treatment(reference=-1))z#C(a, Treatment(reference=-1))[T.a1]za*b)r2rrr)r2rr2r)r2r2rr)r2r2r2r2)ra[T.a2]rza[T.a2]:b[T.b2]z0 + a:b)rrr2r)rr2rr)rrrr2)z a[a1]:b[b1]z a[a2]:b[b1]z a[a1]:b[b2]z a[a2]:b[b2]z 1 + a + a:b)r2r2rr2)rrz a[a1]:b[T.b2]z a[a2]:b[T.b2]rza[H.a2]zC(a, Treatment)zC(a, Treatment)[T.a2])r rPr r rAs r+test_categoricalrcs a1 D   Q!Q!Q!Q( l# %   Q!Q!Q!Q( 78 )   Q!Q!Q!Q( ;<     |\<@>    |\<@D    |\<@B $s)W%DIc4DAr7QGaVaV<{I>VW   Q!Q!Q!Q( -. c4DAr7QGaVaV<{I>VWr-c gdgdgdd}td|dddgd gd gd gd gd ggdgtd|ddddgddgddggddgtd|ddddgddgddggddgy)NrVrYr)rhrkza b cz 0 + I(x + y)rTr2rWrXrZr[r\zI(x + y)z Q('a b c')rrrrz center(x)rrrAs r+ test_builtinsrs\ BDndAtqcA3aS1#s%Cj\R   R1b'Ar7# l# k4DAr7QFQF";k;=WXr-c gdgddgdgddgtjgd}tj|tj|z }|tj|z }fd}t d|}t |d \}}|j jd gk(sJ|j jgd k(sJtj|d gd gd ggsJtj|tjgd gd |ddfsJtd|}t |gd \}|j jgd k(sJtj|d gd gd ggsJtj|tjgd gd |ddfsJtjttd|tjttd|y)N)a2rrrV)rrh)rra1rY)r2rWrXrZr[r\ctSrr?)datassr+rBz)test_incremental..data_iter_makers E{r-z!1 ~ a + center(np.sin(center(x)))r2r)rrzcenter(np.sin(center(x))))r2r2r2)r2r2rrXz~ a + center(np.sin(center(x)))zx ~ xrh)rrsinmeanrrrrr column_stackrrGrHr) rh sin_center_xx_colrBrLr"r#rKrs @r+test_incrementalrs!y1 y1 E %&A66!bggaj.)L 277<0 0EA?SH$XuQx8HC ?? ' 'K= 88 ?? ' ',  ;;saS1#sO ,, ;;sBOOY 59,MN OO=OG "G9eAh 7FS ?? ' ',  ;;saS1#sO ,, ;;sBOOY 59,MN OO MM*mWoF MM*nc?Cr-c tddgdidddtjdgdtjdgdtjdggd d gy) Nz ~ np.sin(x)rhrVrTr2rWrXrz np.sin(x))rPrrrr-r+test_env_transformrsQ i  RVVAY!RVVAY!RVVAY8 k" r-ctddtjdddd<ddzd<fd}|d gd |d gd |d gd |dgd|dgd|dgdy)NrWrrr2rZr^r_cTt|}|jj|k(sJyr)rrr)formulaordermr$s r+t_termsz test_term_order..t_termss' GT "}}''5000r-za + b + x1 + x2)rrrr^r_zb + a + x2 + x1)rrrr_r^z0 + x1 + a + x2 + b + 1z0 + a:b + a + b + 1)rrrrza + a:x1 + x2 + x1 + b)rrrr^a:x1r_z@0 + a:x1:x2 + a + x2:x1:b + x2 + x1 + a:x1 + x1:x2 + x1:a:x2:a:b)rzx1:x2za:x1:x2zx2:x1:bz x1:a:x2:br_r^r)r rlinspace)rr$s @r+test_term_orderrs a1 DQ1%DJdqDJ1 BC BC %'JK !#AB $&QR JMr-cddd}td|d}|rtj|dggsJytj|dggsJy)Nr[rWrz 0 + I(x / y)r2r)rrr)expect_true_divisionr$rs r+_check_divisionrsO Da(A{{1ug&&&{{1se$$$r-c ddgddgddggddgddggd}td|d d gd gd ggd td|d d gdgdggdy)Nrrr2rWrXrZ)rr{rzzX*YrT) r2r2rWr2rXr2rWrXr\) r2rXrZrWrZr\rmr) rrrY[0]Y[1]z X[0]:Y[0]z X[1]:Y[0]z X[0]:Y[1]z X[1]:Y[1]za:X + Y)r2r2rrWrr2rX)r2rrXrrZrWrZ)rz a[a1]:X[0]z a[a2]:X[0]z a[a1]:X[1]z a[a2]:X[1]rrrrAs r+test_multicolumnrszD\!fq!f !fq!f  D     7 7  *    56 r-cgd}gd}tjtdddgddgddggsJtd\}}tj|dgd gd ggsJtj|ddgddgddggsJy) NrVrYrhr2rWrXrrZr[r\)rrrr)rhrkr"r#s r+test_dmatrix_dmatrices_no_datar@sAA ;;ws|q!fq!fq!f%= >>!HC ;;saS1#sO ,, ;;saVaVaV4 55 5r-ctdgdgdgdd\}}|jjdk(sJ|jjdk(sJy)Nz y ~ x + arVrY)rra3)rkrhrrkz 1 + a + x)rrdescribe)r"r#s r+test_designinfo_describerIsR99;MNHC ?? # # % ,, ?? # # % 44 4r-c dd}d} t|gdgddJ#t$r5}|jt|d|j dk(sJYd}~nd}~wwxYw tdd iiJ#t $r8}t |tsJ|jtdd d k(sJYd}~yd}~wwxYw) Nctd)NWHEEEEEEr)rhs r+raise_patsy_errorz2test_evalfactor_reraise..raise_patsy_errorUs $$r-zraise_patsy_error(X) + YrVrY)r{rzr z1 + x[1]rhrZrm)rroriginrrv Exceptionr5)rrrs r+test_evalfactor_reraiser Qs%)Gyy9:  Bxx6'1gmmC.@AAAAB S"I&  4!Z((xx6*a33334s, A+AAA.. B/7.B**B/c dddtjgtjdddgd}dg}tr|jd |D]9}t d || }tj |gd gd gsJ|d k(r2|j jtjddgsJtjttd ||dtd|| \}}tj |dgdggsJtj |ddgddggsJ|d k(rd|j jtjddgsJ|j jtjddgsJtjttd||dtd|| \}}tj |dgdgdggsJtj |dgdgdggsJ|d k(rd|j jtjgdsJ|j jtjgdsJtjttd||d<y)Nr2rWrXrr(rr.r1r)r$r4)r2rWr)r2rXrraise)r$r4 NA_actionrzy ~ 1rV)rnanrrIrrrvequalsrIndexrGrHrr)r$ return_typesr4matlmatrmats r+test_dmatrix_NA_actionrks Q266""&&"b")= >D:LK(#, gDkB~~cJ #;<< + %99##FLL!Q$89 9   #  wT{K d~~dbTB4L11~~daVaV$455 + %::$$V\\1a&%9: :::$$V\\1a&%9: :   #  wT{K d~~dbTB4"$677~~daS1#sO44 + %::$$V\\)%<= =::$$V\\)%<= =   #  K, r-cdddd}dgdgfdgdgffD]\}}t||}tj||sJtjt|jg|d |sJt sat j|}tjt|||sJtjt|jg|d |rJy) N皙?333333?r)r^r_rzx1 + x2)r2rrzC(a, levels=('a1', 'a2')) + x1)r2rrr)rrrrrrrr)data_0drexpectedr data_seriess r+ test_0d_datars$/G ]O$ )K=9gw'{{3)){{ !3??"3W =a @(    --0K;;ww HJ+MH]66;N ;;}n 55 5r-ctsytgdddgg}tr'|jt j |d|D]}d|i}t jtd|ddgddgddggsJt jtd|ddgddgddggsJt jtd|ddgddgddggsJt jtd |ddgddgddggrJy) N)r2rr2rrrobjr2zC(obj)zC(obj, levels=['b', 'a'])zC(obj, levels=['a', 'b'])) rrrrIrrrrr)objsr%rs r+test_C_and_pandas_categoricalr's  " ))c3Z @ AD$ FMM$q'*+   CL{{75!,1v1v1v.FGG{{78Q/1a&1a&1a&1IJJ{{ / 3q!fq!fq!f5M  {{ / 3q!fq!fq!f5M    r-ctsyddlm}d}|d}tjdt j dggddgd  }t||d|d }|j}tjd dgddgd}t||dd }t j|gdgdgsJt|jgdk(sJy)Nr)NAActionz 1 + x + zdroprrrV)zrh)rrrrur1)rr4@@r\rlr3)rg@r,)rg@r-)rrhr+) r patsy.missingr)rrrrrrrlistro)r)ractionr$rdidata2dm2s r+"test_NAActioon_pandas_string_indexr4s &G f F   BFFC y1 D $V MB B   C:QF; ??   7 77 7r-)NN)=numpyrrGrrpatsy.design_inforr patsy.evalr patsy.descrr r patsy.categoricalr patsy.contrastsr patsy.user_utilr r patsy.buildrrrrrrr patsy.utilrrrr patsy.originrrr,r:r<rPrTrrrrrrrrrrrrrr rrr#r'r4rr-r+r?s 6&11#2EMM