U ·#˜e¢)ã@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 úa/mounts/lovelace/software/anaconda3/envs/rescript/lib/python3.8/site-packages/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 rÚdiffZefactorZ 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 ÿr2cCs¾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+)r2r$r&r+)Zs_abZs_abcZs_nullZs_cdZs_aZs_apZs_abpÚbadr r r Ú test__Subterm~sr8ccsVdd„}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)Útuplr=ÚexpandedZexpanded_subsetsr<r r r r9s   r9c Cs`ttdƒƒddddgkst‚ttdƒƒddddddddgks@t‚ttttd ƒƒƒƒd ks\t‚dS) N)r!ér )r!)rG)r!rGé)rH)r!rH)rGrHéé )r'r9r&r"Úranger r r r Útest__subsets_sorted¥s  ÿrLcCsjt|ƒD]\\}}t||dd…ƒD]>\}}| |¡r$| |¡}|||d|<| |¡dSq$qdS)Nr!TF)rBr$r+Úpop)r/Zshort_iZ short_subtermZlong_iZ long_subtermZ new_subtermr r r Ú_simplify_one_subterm«s    rNcCst|ƒr qdSr)rN)r/r r r Ú_simplify_subterms·srOcCsfdd„}|dgdgƒ|ddgdgƒ|ddddgdgƒ|dddgddgƒ|dddgdd gƒdS) NcSs6t|ƒ}t|ƒ}td|d|ƒt|ƒ||ks2t‚dS)Nz testing if:z->)r2ÚprintrOr&)ÚgivenÚexpectedr r r Út¼s z"test__simplify_subterms..t)r3r )r5)r4)r3r4)r5r6)r3r6r )rSr r r Útest__simplify_subterms»s rTc 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 rAÝsz+pick_contrasts_for_term..cSsg|]}td|ƒ‘qS)F)rrUr r r rAãs) r1r9rr.ÚupdaterOr rr) ÚtermrXÚ used_subtermsZcategorical_factorsr/r<r0Úfactor_codingsÚ factor_codingrFr rWr Úpick_contrasts_for_termÜs     r^cCsô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)r`rbÚcÚd)rcrd)Ú patsy.descr_r(r^r&)r_ÚusedZcodingsZ used_snapshotr r r Útest_pick_contrasts_for_termðs rgN)Ú __future__rÚ patsy.utilrÚobjectrrr2r8r9rLrNrOrTr^rgr r r r Ú+s  %  !