B }|U)@sddlmZddlmZGdddeZGdddeZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZdS))print_function) no_picklingc@s<eZdZdZddZddZddZdd Zd d Ze Z d S) _ExpandedFactorzA factor, with an additional annotation for whether it is coded full-rank (includes_intercept=True) or not. These objects are treated as immutable.cCs||_||_dS)N)includes_interceptfactor)selfrrr/lib/python3.7/site-packages/patsy/redundancy.py__init__6sz_ExpandedFactor.__init__cCstt|j|jfS)N)hashrrr)rrrr __hash__:sz_ExpandedFactor.__hash__cCs"t|to |j|jko |j|jkS)N) isinstancerrr)rotherrrr __eq__=s  z_ExpandedFactor.__eq__cCs ||k S)Nr)rrrrr __ne__Bsz_ExpandedFactor.__ne__cCs|jr d}nd}d|j|fS)N+-z%r%s)rr)rsuffixrrr __repr__Esz_ExpandedFactor.__repr__N) __name__ __module__ __qualname____doc__r r rrrr __getstate__rrrr r1src@sLeZdZdZddZddZddZdd Zd d Zd d Z ddZ e Z dS)_SubtermzAlso immutable.cCst||_dS)N) frozensetefactors)rrrrr r Psz_Subterm.__init__cCs&t|jt|jdko$|j|jS)N)lenr issuperset)rrrrr can_absorbSsz_Subterm.can_absorbcCsX|j|j}t|dkstt|d}|jr4tt|j}|td|j t |S)NrrT) r differencerAssertionErrorlistrsetaddrrr)rrZdiffZefactorZ new_factorsrrr absorbXs   z_Subterm.absorbcCstt|jfS)N)r rr)rrrr r asz_Subterm.__hash__cCst|to|j|jkS)N)r rr)rrrrr rds z_Subterm.__eq__cCs ||k S)Nr)rrrrr rhsz_Subterm.__ne__cCsd|jjt|jfS)Nz%s(%r)) __class__rr#r)rrrr rksz_Subterm.__repr__N) rrrrr r r&r rrrrrrrrr rNs rc Csfg}x\|D]T}g}x<|D]4}|ddks,t|t|ddk|ddqW|t|q W|S)N)rrr)r"appendrr)Zshort_subtermssubtermssubtermfactorsZ factor_namerrr _expand_test_abbrevsss  r-cCstddggd}tdddggd}tggd}tddggd}tdggd}tdggd}tddggd}x$|||||fD]}||rtqW||st|||kstdS)Nza-zb-rzc-zd-za+zb+)r-r r"r&)Zs_abZs_abcZs_nullZs_cdZs_aZs_apZs_abpZbadrrr test__Subterm~sr.ccsZdd}tt|}t||}||jtdx |D]}tdd|DVq:WdS)NcssD|s dVn4|d}x*t|ddD]}|V|f|Vq&WdS)Nrrr)_subsets_sorted)seqobjsubsetrrr helpers z_subsets_sorted..helper)keycSsg|] \}}|qSrr).0idxr1rrr sz#_subsets_sorted..)r# enumeratesortrtuple)Ztuplr3expandedZexpanded_subsetsr2rrr r/s    r/c Cs`ttdddddgkstttdddddddddgks@tttttd d ks\tdS) N)rr)r)r<)rr<)r=)rr=)r<r= )r#r/r"rrangerrrr test__subsets_sorteds rAcCsnxht|D]\\}}xRt||ddD]:\}}||r(||}|||d|<||dSq(Wq WdS)NrTF)r8r r&pop)r*Zshort_iZ short_subtermZlong_iZ long_subtermZ new_subtermrrr _simplify_one_subterms    rCcCsx t|r qWdS)N)rC)r*rrr _simplify_subtermss rDcCsfdd}|dgdg|ddgdg|ddddgdg|dddgddg|dddgdd gdS) NcSs6t|}t|}td|d|t|||ks2tdS)Nz testing if:z->)r-printrDr")ZgivenZexpectedrrr ts z"test__simplify_subterms..t)za-r)za+)zb-)za-zb-)za+zb+)za-zb+r)rFrrr test__simplify_subtermss rGc sfdd|jD}g}x4t|D](}tdd|D}||kr"||q"W||t|g}x6|D].}i}x|jD]} | j|| j<qzW||qjW|S)Ncsg|]}|kr|qSrr)r5f)numeric_factorsrr r7sz+pick_contrasts_for_term..cSsg|]}td|qS)F)r)r5rHrrr r7s) r,r/rr)updaterDrrr) ZtermrIZ used_subtermsZcategorical_factorsr*r2r+Zfactor_codingsZ factor_codingr;r)rIr pick_contrasts_for_terms   rKcCsddlm}t}t|gt|}|igks2tt|ddgtdg|}|ddigks^tt|ddgt|}|dddgkstt|}t|d d gt|}|d diddd gkstt|d d gt|}|d diddd gkstdS) Nr)TermaxFbT)rMrOcd)rPrQ)Z patsy.descrLr$rKr")rLZusedZcodingsZ used_snapshotrrr test_pick_contrasts_for_terms rRN)Z __future__rZ patsy.utilrobjectrrr-r.r/rArCrDrGrKrRrrrr +s  %  !