o #e'o@sxgdZddlZddlZddlZddlmZddlm Z zddl Z Wn e y-da Ynwda t o6e e dZt sZ4d?d@Z5dAdBZ6dCdDZ7dEdFZ8dGdHZ9dIdJZ:dKdLZ;dMdNZdSdTZ?dUdVZ@dWdXZAdYdZZBd[d\ZCd]d^ZDdS)a)atleast_2d_column_defaultuniqueify_list widest_floatwidest_complexwide_dtype_forwidenrepr_pretty_delegaterepr_pretty_implSortAnythingKeysafe_scalar_isnan safe_isnaniterable have_pandashave_pandas_categoricalhave_pandas_categorical_dtypepandas_Categorical_from_codespandas_Categorical_categoriespandas_Categorical_codes safe_is_pandas_categorical_dtypesafe_is_pandas_categoricalsafe_issubdtype no_picklingassert_no_picklingsafe_string_eqN) cStringIO)optional_dep_okFT CategoricalCategoricalDtypecCstt|d|tjS)Ndtype) isinstancegetattrpandasrxr%*lib/python3.10/site-packages/patsy/util.py/sr'apiZis_categorical_dtypecCsXtr#t|tjtjfr#i}t|dr|j|d<|j|f||d|Stj ||||dS)Nname)copyr)r*rsubok) r r r"Series DataFramehasattrr) __class__nparray)ar*rr+ extra_argsr%r%r&asarray_or_pandas9s  r4cCsddl}ttgdtjusJ|9}|ddttttgdgtjus.JtttgdgddtjusAJ|dusGJWdn1sQwYt gd}t||useJt|dd}t ||ssJd|d<t ||rJt tgdt d gd sJtgdt d j t t ksJt||j d }d |d<|dd ksJtr@tjgdd gd d}t|}|jd ksJt |jgd sJd|d<|ddksJt|dd}|jd ksJt |jgd sJt ||sJd|d<t ||rJt|t d j t t ks Jt||j d }|jd ks/Jt |jgd s;Jd |d<|dd ksHJtjgdggddgd} t| } d| jd<t | jgdskJt | jdgsvJ| jddksJt| dd} t | | sJt | jgdsJt | jdgsJd| jd<t | | rJt| t d } | d j t t ksJt | | sJt | jgdsJt | jdgsJt| | d j d } t | jgdsJt | jdgsJt | | sJt}z*datttgdtjus&Jtttgdgtjus8JW|adS|awdS)Nrrignorethe matrix subclassT)r+)r*drg?g@g@cA) r)indexer?)r>BC)columnsrC)r?r>F)warningstyper4r0ndarraycatch_warningsfilterwarningsPendingDeprecationWarningmatrixr1 array_equalallclosefloatrr r"r,r)rCr-locrG)rHwr2Za_copyZa_viewsZs_view1Zs_copyZs_view2dfZdf_view1Zdf_copyZ df_convertedZdf_view2 had_pandasr%r%r&test_asarray_or_pandasFs "          rWcCsh|rtrt|tjrt|St|tjr|St|}t|}|jdkr+| d}|jdks2J|S)Nr)rr6) r r r"r,r-r0asarrayZ atleast_1dndimreshape)r2preserve_pandasr%r%r&rs       rcCs ddl}ttgddgdgdggksJtdjdks Jtdgjdks*Jtdggjdks5JtdgggjdksAJtgdjdksLJtdgdgdggjdks[J|#}|d d tttt dtj ksuJ|dus{JWdn1swYt rttt ddgtj ksJttt dgdggtj ksJttt ddgd d t jksJttt dgdggd d t jksJt j gd dgdd}t|d d }t|t jsJt|jdgksJt|jgdksJ|'}|d d tttt dd d tj ks$J|dus+JWdn 1s6wYttgdd d tj ksKJt rt }z/da ttt ddgd d tj ksfJttt dgdggd d tj ks{JW|a dS|a wdS)Nrr5rr6r7)rr)rrrr7rr8r9T)r\)r? hir2bcrBF)rHr0allrshaperKrLrMrIrNrJr r"r,r-r rGrC)rHrSrTrUrVr%r%r&test_atleast_2d_column_defaults&        rfcCsts||St|tjs||S||jkr|St|dkr:|jddkr:|d|jdkr3td||jdStd|j|f)Nrrzarrays have incompatible sizesz4cannot reshape a DataFrame with shape %s to shape %s) r r[r r"r-relen ValueErrorrG)r2Z new_shaper%r%r&pandas_friendly_reshapes    ricCs8ddl}tttddddtdddsJtrtjdgdigdd }t|d }t |tjs9Jt |j gdsDJt |j dgsNJt|d }t |tj s[Jt |j gdsfJ|jdksmJ|tt|d |tt|d |tt|dt}zda|tt|d W|adS|awdS)Nrr?r6)r6rjr$r5ra)rCr])r7))rr7)r7r7F)pytestr0rPriaranger[r r"r-r rOrCrGr,r)raisesrhAttributeError)rlrUZnoopZsquozenrVr%r%r&test_pandas_friendly_reshapes2  rpcCs4g}t}|D]}||vr||||q|SN)setappendadd)seqZseq_newseenobjr%r%r&rs  rcCsLtgdgdks JtgdgdksJtgdgdks$JdS)Nr5)rr7r7r6r7r)rr7r6)r7r6rrkrr6r7)r7r6rrk)rr%r%r%r&test_to_uniqueify_list"srx)Zfloat128Zfloat96Zfloat64)Z complex256Z complex196Z complex128cCsLt|}t|jtjst|jtjrtSt|jtjrtSt d|jf)Nz"cannot widen a non-numeric type %r) r0rYrrintegerfloatingrZcomplexfloatingrrhZarrr%r%r&r4s  rcCstj|t|dS)Nr;)r0rYrr{r%r%r&r=srcCsttgdgdsJtgdjtksJttgdgds'Jtgdjtks2Jttgdgds@JtgdjtksKJddl}|ttdgdS)Nr5r<)y?r6r7rr`) r0rPrrrrrlrnrh)rlr%r%r&test_wide_dtype_for_and_widen@sr|c@s@eZdZddZddZddZddZeZd d Zd d Z d S)PushbackAdaptercCs||_g|_dSrq)_it_pushed)selfitr%r%r&__init__Ks zPushbackAdapter.__init__cCs|Srqr%rr%r%r&__iter__OszPushbackAdapter.__iter__cC|j|dSrq)rrsrrwr%r%r& push_backRzPushbackAdapter.push_backcCs|jr|jSt|jSrq)rpopsixadvance_iteratorr~rr%r%r&nextUs  zPushbackAdapter.nextcCs4zt|}Wn tytdw|||S)Nz no more data)rr StopIterationrhrrr%r%r&peek]s  zPushbackAdapter.peekcCs$z|WdStyYdSwNFT)rrhrr%r%r&has_morees   zPushbackAdapter.has_moreN) __name__ __module__ __qualname__rrrr__next__rrr%r%r%r&r}Js r}cCsttgd}|sJt|dksJ|dt|dks%Jt|dks.J|dks6J|d|dksCJ|d|dksPJ|sVJt|gdks`J|rfJdS) N)rr6r7rkrrr6r7r?r@)r@r?r7rk)r}iterrrrrrlist)rr%r%r&test_PushbackAdapterms     rc@sFeZdZddZddZdddZdd Zd d Zd d ZddZ dS) _MiniPPrintercCst|_d|_dSNr)StringIO_outZ indentationrr%r%r&rs z_MiniPPrinter.__init__cCrrqrwrite)rtextr%r%r&rrz_MiniPPrinter.text cCrrqr)rsepr%r%r& breakablerz_MiniPPrinter.breakablecC||dSrqrr_rr%r%r& begin_groupz_MiniPPrinter.begin_groupcCrrqrrr%r%r& end_grouprz_MiniPPrinter.end_groupcCs,t|dr ||ddS|t|dS)N _repr_pretty_F)r.rrreprrr%r%r&prettys z_MiniPPrinter.prettycCs |jSrq)rgetvaluerr%r%r&r z_MiniPPrinter.getvalueN)r) rrrrrrrrrrr%r%r%r&rs  rcCst}|||Srq)rrr)rwprinterr%r%r& _mini_prettys rcCs*trdtjvrddlm}||St|S)NZIPythonr)r)rsysmodulesZIPython.lib.prettyrr)rwrr%r%r&rs rc s|jj}t|dd|fdgfdd}|D] }||q|D]#\}}|t|dd|f|t|ddq*t|dddS) Nrz%s(Fcs&dr ddd<dS)Nr,T)rrr%pZstartedr%r&new_items  z"repr_pretty_impl..new_itemz%s=))r/rrrgrr) rrwargskwargsr)rargZlabelvaluer%rr&rs   rcCsRtddksJt}Gdddt}t||ddgddg|d ks'JdS) Nasdfz'asdf'c@ eZdZdS)z!test_repr_pretty..MyClassNrrrr%r%r%r&MyClassrr2r)foobar)rrz'MyClass('a', 1, foo='bar', asdf='asdf'))rrobjectrr)rrr%r%r&test_repr_prettys rc@s$eZdZddZddZddZdS)r cCs ||_dSrqrwrr%r%r&rrzSortAnythingKey.__init__cCs$z|j|kWStytYSwrq)rw TypeErrorNotImplemented)rZ other_objr%r%r& _python_lts   zSortAnythingKey._python_ltcCst|tsJ||j}|tur|S|dturdS|dtur%dS|j|jkr-dS|jjjt|jf|jjjt|jfkS)NrTF)r r rrwrr/rid)rotherresultr%r%r&__lt__s  zSortAnythingKey.__lt__N)rrrrrrr%r%r%r&r s r cCstgdtdgdksJtddgtdddgksJtgdtdgdks*JGdd d t}Gd d d t}Gd d d t}|}|}|}t}t||d||gtdd||||gksbJdS)N)r@r?r)key)rr?rr@r?g)r?r24@rb)r?rr2rbc@r)ztest_SortAnythingKey..aNrr%r%r%r&r2rr2c@r)ztest_SortAnythingKey..bNrr%r%r%r&rbrrbc@r)ztest_SortAnythingKey..zNrr%r%r%r&zrrr)sortedr r)r2rbrZa_objZb_objZz_objZo_objr%r%r&test_SortAnythingKeys  rc Cs,ztt|WStttfyYdSwNF)r0isnanrQrrhNotImplementedErrorr#r%r%r&r )s r )ZotypescCstdrJtdr JtdrJtdrJttgdr#Jttjgr+Jttjs2Jtttjss         V@       #$ #