o #e)@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||_||_dSNincludes_interceptfactor)selfrrr 0lib/python3.10/site-packages/patsy/redundancy.py__init__6s z_ExpandedFactor.__init__cCstt|j|jfSr)hashrrrr r r r __hash__:sz_ExpandedFactor.__hash__cCs"t|to|j|jko|j|jkSr) isinstancerrrr otherr r r __eq__=s   z_ExpandedFactor.__eq__cC ||k Srr rr r r __ne__B z_ExpandedFactor.__ne__cCs|jrd}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@sLeZdZdZddZddZddZdd Zd d Zd d Z ddZ e Z dS)_SubtermzAlso immutable.cCst||_dSr) frozensetefactors)r r"r r r r Pz_Subterm.__init__cCs&t|jt|jdko|j|jS)N)lenr" issupersetrr r r can_absorbSs z_Subterm.can_absorbcCsX|j|j}t|dksJt|d}|jrJt|j}|td|jt |S)Nr$rT) r" differencer%listrsetaddrrr )r rdiffZefactorZ new_factorsr r r absorbXs   z_Subterm.absorbcCstt|jfSr)r r r"rr r r rar#z_Subterm.__hash__cCst|to |j|jkSr)rr r"rr r r rds  z_Subterm.__eq__cCrrr rr r r rhrz_Subterm.__ne__cCsd|jjt|jfS)Nz%s(%r)) __class__rr)r"rr r r rksz_Subterm.__repr__N) rrrrr r'r-rrrrrrr r r r r Ns r c Cs^g}|D](}g}|D]}|ddvsJ|t|ddk|ddq |t|q|S)N)rrr)appendrr )Zshort_subtermssubtermssubtermfactorsZ factor_namer r r _expand_test_abbrevsss r4cCstddggd}tgdgd}tggd}tddggd}tdggd}tdggd}tddggd}|||||fD] }||rKJqB||sSJ|||ks\JdS) Na-b-r)r5r6c-r7zd-a+b+)r4r'r-)Zs_abZs_abcZs_nullZs_cdZs_aZs_apZs_abpZbadr r r test__Subterm~sr:ccsXdd}tt|}t||}||jtd|D] }tdd|DVqdS)NcssD|sdVdS|d}t|ddD] }|V|f|VqdS)Nr rr$)_subsets_sorted)seqobjsubsetr r r helpers z_subsets_sorted..helper)keycSsg|]\}}|qSr r ).0idxr=r r r sz#_subsets_sorted..)r) enumeratesortr%tuple)Ztuplr?expandedZexpanded_subsetsr>r r r r;s  r;cCsPttdgdks JttdgdksJttttddks&JdS)Nr$)r r$rIrHr$rI)r rJrK)rMrH)r$rM)rIrMrL )r)r;r%ranger r r r test__subsets_sorteds   rQcCsjt|D].\}}t||ddD]\}}||r1||}|||d|<||dSqqdS)Nr$TF)rDr'r-pop)r1Zshort_iZ short_subtermZlong_iZ long_subtermZ new_subtermr r r _simplify_one_subterms   rScCst|r t|sdSdSr)rS)r1r r r _simplify_subtermssrTcCs^dd}|dgdg|ddgdg|gddg|gddd g|gd d d gdS) NcSs6t|}t|}td|d|t|||ksJdS)Nz testing if:z->)r4printrT)ZgivenZexpectedr r r ts z"test__simplify_subterms..tr5r )r8)r rWr6r5r6)r8r9)r rWrYrY)rWrXrYrX)r5r9r )rVr r r test__simplify_subtermss rZc sfdd|jD}g}t|D]}tdd|D}||vr$||q||t|g}|D]}i}|jD]} | j|| j<q9||q2|S)Ncsg|]}|vr|qSr r rAfnumeric_factorsr r rCsz+pick_contrasts_for_term..cSsg|]}td|qS)F)rr[r r r rCs) r3r;r r0updaterTr"rr) Ztermr^Z used_subtermsZcategorical_factorsr1r>r2Zfactor_codingsZ factor_codingrGr r]r pick_contrasts_for_terms      r`cCsddlm}t}t|gt|}|igksJt|ddgtdg|}|ddigks/Jt|ddgt|}|dddgksDJt|}t|d d gt|}|d diddd gks`Jt|d d gt|}|d diddd gksxJdS) Nr)TermaxFbT)rbrdcd)rerf)Z patsy.descrar*r`)rausedZcodingsZ used_snapshotr r r test_pick_contrasts_for_terms rhN)Z __future__rZ patsy.utilrobjectrr r4r:r;rQrSrTrZr`rhr r r r s * %  !