U Kv f¢)ã@s„ddlmZddlmZGdd„deƒZGdd„deƒZdd„Zd d „Zd d „Z d d„Z dd„Z dd„Z dd„Z dd„Zdd„ZdS)é)Úprint_function)Ú no_picklingc@s<eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Ze Z d S) Ú_ExpandedFactorzœA 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_interceptÚfactor)Úselfrr©r ú4/tmp/pip-unpacked-wheel-68fdvdus/patsy/redundancy.pyÚ__init__6sz_ExpandedFactor.__init__cCstt|j|jfƒSr)Úhashrrr©r r r r Ú__hash__:sz_ExpandedFactor.__hash__cCs"t|tƒo |j|jko |j|jkSr)Ú isinstancerrr©r Úotherr r r Ú__eq__=s   ÿ þz_ExpandedFactor.__eq__cCs ||k Srr rr r r Ú__ne__Bsz_ExpandedFactor.__ne__cCs|jr d}nd}d|j|fS)Nú+ú-z%r%sr)r Úsuffixr r r Ú__repr__Esz_ExpandedFactor.__repr__N) Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrrrrÚ __getstate__r r r r r1src@sLeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z e Z dS)Ú_SubtermzAlso immutable.cCst|ƒ|_dSr)Ú frozensetÚefactors)r r r r r r Psz_Subterm.__init__cCs&t|jƒt|jƒdko$|j |j¡S)Né)Úlenr Ú issupersetrr r r Ú can_absorbSs ÿz_Subterm.can_absorbcCsX|j |j¡}t|ƒdkst‚t|ƒd}|jr4t‚t|jƒ}| td|j ƒ¡t |ƒS)Nr!rT) r Ú differencer"ÚAssertionErrorÚlistrÚsetÚaddrrr)r rZdiffZefactorZ new_factorsr r r ÚabsorbXs   z_Subterm.absorbcCstt|jfƒSr)r rr rr r r rasz_Subterm.__hash__cCst|tƒo|j|jkSr)rrr rr r r rds  ÿz_Subterm.__eq__cCs ||k Srr rr r r rhsz_Subterm.__ne__cCsd|jjt|jƒfS)Nz%s(%r))Ú __class__rr'r rr r r rksz_Subterm.__repr__N) rrrrr r$r*rrrrrrr r r r rNs rc Cs^g}|D]P}g}|D]4}|ddks(t‚| t|ddk|dd…ƒ¡q| t|ƒ¡q|S)Néÿÿÿÿ)rrr)r&Úappendrr)Zshort_subtermsÚsubtermsÚsubtermÚfactorsZ factor_namer r r Ú_expand_test_abbrevsss ÿr1cCs¾tddggƒd}tdddggƒd}tggƒd}tddggƒd}tdggƒd}tdggƒd}tddggƒd}|||||fD]}| |¡r†t‚q†| |¡s¨t‚| |¡|ksºt‚dS)Núa-úb-rzc-zd-úa+úb+)r1r$r&r*)Zs_abZs_abcZs_nullZs_cdZs_aZs_apZs_abpÚbadr r r Ú test__Subterm~sr7ccsVdd„}tt|ƒƒ}t||ƒƒ}| ¡|jtd|D]}tdd„|DƒƒVq8dS)Ncss@|s dVn0|d}t|dd…ƒD]}|V|f|Vq$dS)Nr rr!)Ú_subsets_sorted)ÚseqÚobjÚsubsetr r r Úhelpers z_subsets_sorted..helper)ÚkeycSsg|] \}}|‘qSr r )Ú.0Úidxr:r r r Ú £sz#_subsets_sorted..)r'Ú enumerateÚsortr"Útuple)Ztuplr<ÚexpandedZexpanded_subsetsr;r r r r8s   r8c Cs`ttdƒƒddddgkst‚ttdƒƒddddddddgks@t‚ttttd ƒƒƒƒd ks\t‚dS) N)r!ér )r!)rE)r!rEé)rF)r!rF)rErFéé )r'r8r&r"Úranger r r r Útest__subsets_sorted¥s  ÿrJcCsjt|ƒD]\\}}t||dd…ƒD]>\}}| |¡r$| |¡}|||d|<| |¡dSq$qdS)Nr!TF)rAr$r*Úpop)r.Zshort_iZ short_subtermZlong_iZ long_subtermZ new_subtermr r r Ú_simplify_one_subterm«s    rLcCst|ƒr qdSr)rL)r.r r r Ú_simplify_subterms·srMcCsfdd„}|dgdgƒ|ddgdgƒ|ddddgdgƒ|dddgddgƒ|dddgdd gƒdS) NcSs6t|ƒ}t|ƒ}td|d|ƒt|ƒ||ks2t‚dS)Nz testing if:z->)r1ÚprintrMr&)ÚgivenÚexpectedr r r Út¼s z"test__simplify_subterms..t)r2r )r4)r3)r2r3)r4r5)r2r5r )rQr r r Útest__simplify_subterms»s rRc s”‡fdd„|jDƒ}g}t|ƒD](}tdd„|Dƒƒ}||kr | |¡q | |¡t|ƒg}|D]*}i}|jD]} | j|| j<qr| |¡qd|S)Ncsg|]}|ˆkr|‘qSr r ©r>Úf©Únumeric_factorsr r r@Ýsz+pick_contrasts_for_term..cSsg|]}td|ƒ‘qS)F)rrSr r r r@ãs) r0r8rr-ÚupdaterMr rr) ZtermrVZ used_subtermsZcategorical_factorsr.r;r/Zfactor_codingsZ factor_codingrDr rUr Úpick_contrasts_for_termÜs     rXcCsôddlm}tƒ}t|gƒtƒ|ƒ}|igks2t‚t|ddgƒtdgƒ|ƒ}|ddigks^t‚t|ddgƒtƒ|ƒ}|dddœgksˆt‚t|ƒ}t|d d gƒtƒ|ƒ}|d diddd œgksÀt‚t|d d gƒtƒ|ƒ}|d diddd œgksðt‚dS) Nr)ÚTermÚaÚxFÚbT)rZr\ÚcÚd)r]r^)Z patsy.descrYr(rXr&)rYÚusedZcodingsZ used_snapshotr r r Útest_pick_contrasts_for_termðs r`N)Ú __future__rZ patsy.utilrÚobjectrrr1r7r8rJrLrMrRrXr`r r r r Ú+s  %  !