B < [o@sdddddddddd d d d d ddddddddddgZddlZddlZddlZddlmZddlm Z y ddl Z Wne k rda YnXda t oe e dZt sdZdZn2e e d rdZe jjjZnee jjd!dZedk Zdad"d#Zd$d%Zdbd&dZd'd(Zd)d*Zd+d,Zd-dZd.d/Zx0d0D]Ze eer&eeeZ Pq&WdsRt!x0d1D]Z"e ee"rXeee"Z#PqXWdst!d2dZ$d3dZ%d4d5Z&Gd6d7d7e'Z(d8d9Z)Gd:d;d;e'Z*ddZ,gfd?dZ-d@dAZ.GdBdde'Z/dCdDZ0dEd Z1ej2e1e3gdFZ4dGdHZ5dIdJZ6dKd Z7dLdMZ8dNdZ9dOdPZ:dQdZ;dRdZdVdZ?dWdXZ@dYdZAdZd[ZBd\dZCd]dZDd^dZEd_d`ZFdS)catleast_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 Categoricalapiis_categorical_dtypecCsXtrFt|tjtjfrFi}t|dr.|j|d<|j|f||d|Stj ||||dS)Nname)copydtype)r!r"subok) r isinstancepandasSeries DataFramehasattrr __class__nparray)ar!r"r#Z extra_argsr-)lib/python3.7/site-packages/patsy/util.pyasarray_or_pandas<s  r/c Csddl}ttdddgtjks$t|l}|ddtttt dddggtjks`tttt dddggddtj kst|dkstWdQRXt dddg}t||kstt|dd }t ||std |d<t ||rtt tdddgt d d d dgsttdddgt d jtt kswr,Za_copyZa_viewsZs_view1Zs_copyZs_view2dfZdf_view1Zdf_copyZ df_convertedZdf_view2 had_pandasr-r-r.test_asarray_or_pandasIs $$ $      rMcCsh|r.tr.t|tjrt|St|tjr.|St|}t|}|jdkrV| d}|jdksdt |S)Nr)rr0) r r$r%r&r'r*asarrayZ atleast_1dndimreshaperA)r,preserve_pandasr-r-r.rs       c Csddl}ttdddgdgdgdggks0ttdjdksBttdgjdksVttdggjdkslttdgggjdksttdddgjdksttdgdgdggjdkst|<}|dd tt tt dtj kst|dkstWdQRXt rt tt ddgtj ks,tt tt dgdggtj ksRtt tt ddgd d t jksxtt tt dgdggd d t jkstt jd d dgddddgd}t|d d }t|t jstt|jdgkstt|jdddgkst|D}|dd tt tt dd d tj ksJt|dksXtWdQRXt tdddgd d tj kstt rt }zXda t tt ddgd d tj kstt tt dgdggd d tj kstWd|a XdS)Nrrr0r1)rr)rrr)r1rr2zthe matrix subclassT)rRr6 hir,bc)r r9F)r>r*allrrAshaperBrCrDr?rEr@r r%r&r'r$r=r9)r>rIrJrKrLr-r-r.test_atleast_2d_column_defaultsb(    rZcCsts||St|tjs$||S||jkr2|St|dkrt|jddkrt|d|jdkrftd||jdStd|j|fdS)Nrrzarrays have incompatible sizesz4cannot reshape a DataFrame with shape %s to shape %s) r rQr$r%r'rYlen ValueErrorr=)r,Z new_shaper-r-r.pandas_friendly_reshapes    r]cCs8ddlm}tttddddtddds>ttr4t j dddd gid d d gd }t|d}t |t j s|tt |j d d d gstt |jdgstt|d}t |t jstt |j d d d gst|jdkst|tt|d|tt|d|tt|dt}zda|tt|dWd|aXdS)Nr) assert_raisesr6r0)r0r_xrr1r,rVrW)r9)r1r)r1))rr1)r1r1F) nose.toolsr^r*rGr]arangerQrAr r%r'r$rFr9r=r&r r\AttributeError)r^rKZnoopZsquozenrLr-r-r.test_pandas_friendly_reshapes,   recCs8g}t}x(|D] }||kr||||qW|S)N)setappendadd)seqZseq_newseenobjr-r-r.rs  cCshtdddgdddgksttddddddgdddgks>ttdddddddgddddgksdtdS)Nrr0r1ra)rrAr-r-r-r.test_to_uniqueify_list%s"rl)Zfloat128Zfloat96Zfloat64)Z complex256Z complex196Z complex128cCsPt|}t|jtjs&t|jtjr*tSt|jtjr.new_itemz%s=))r)rrr[rr) rrkargskwargsr rargZlabelvaluer-)rrr.rs  cCsRtddkstt}Gdddt}t||ddgddg|d ksNtdS) Nasdfz'asdf'c@s eZdZdS)z!test_repr_pretty..MyClassN)rrrr-r-r-r.MyClasssrr,r)foobar)rrz'MyClass('a', 1, foo='bar', asdf='asdf'))rrArobjectrr)rrr-r-r.test_repr_prettys rc@s$eZdZddZddZddZdS)r cCs ||_dS)N)rk)rtrkr-r-r.rvszSortAnythingKey.__init__cCs$y |j|kStk rtSXdS)N)rk TypeErrorNotImplemented)rtZ other_objr-r-r. _python_lts zSortAnythingKey._python_ltcCst|tst||j}|tk r&|S|dtk r8dS|dtk rJdS|j|jkrZdS|jjjt|jf|jjjt|jfkS)NrTF) r$r rArrkrr)rid)rtotherresultr-r-r.__lt__s  zSortAnythingKey.__lt__N)rrrrvrrr-r-r-r.r s cCstddddgtdddddgks$ttddgtdddgks@ttdddd gtddddd gksdtGd ddt}Gd d d t}Gd d d t}|}|}|}t}t||d||gtdd||||gkstdS)Nr7r6r)keygr,g4@rVc@s eZdZdS)ztest_SortAnythingKey..aN)rrrr-r-r-r.r,sc@s eZdZdS)ztest_SortAnythingKey..bN)rrrr-r-r-r.rVsc@s eZdZdS)ztest_SortAnythingKey..zN)rrrr-r-r-r.z!srr)sortedr rAr)r,rVrZa_objZb_objZz_objZo_objr-r-r.test_SortAnythingKeys$$rc Cs.ytt|Stttfk r(dSXdS)NF)r*isnanrHrr\NotImplementedError)r`r-r-r.r ,s)ZotypescCstdr ttdrttdr$ttdr0tttdddgrHtttjgrXtttjsfttttjszttttjstdS)NTZsadf)rr0r1rr0r1)r rAr*rOnanZfloat32rHr-r-r-r.test_safe_scalar_isnan3s    rcCshttdddtjdgdddddgs*tttjjdks>tttjsLttdrXttdrdtdS)NrTrFr)r*rFr rrArPr-r-r-r.test_safe_isnan>s  rcCs&y t|Wntk r dSXdS)NFT)r Exception)rkr-r-r.r Gs  cCsDtds ttgsttddis(ttdr4tttr@tdS)Nrr,r)r rAr-r-r-r. test_iterableNs    rcCs<tstt|}ttjdr,tj||St||SdS)N from_codes)rrAr*rOr(r%rr)codes categoriesr-r-r.rfs   cCs`tsdStddddgddg}tt|dddddgksDttt|ds\tdS)NrrrNr,rV)rrr*rXrOrAr)rWr-r-r."test_pandas_Categorical_from_codesqs &rcCs*t|dr|j}t|dr |jS|jSdS)Ncatr)r(rrZlevels)rr-r-r.rys   cCs*t|dr|j}t|dr |jS|jSdS)Nrr)r(rrlabels)rr-r-r.rs   cCstsdStddddgddg}tt|ddgks8ttt|ddddgksVttrt |}tt|ddgks~ttt|ddddgkstdS)NrrrNr,rV) rrr*rXrrArrr%r&)rWrJr-r-r.!test_pandas_Categorical_accessorss rcCstsdSt|S)NF)r_pandas_is_categorical_dtype)Zdtr-r-r.rscCs0tsdSt|tjrdSt|dr,t|jSdS)NFTr")rr$r%rr(rr")datar-r-r.rs   cCsVttdrttr0tddg}t|s0ttrRtjddgdd}t|sRtdS)Nr6r,rVcategory)r") rr*rcrArr%rrr&)Zc_objZs_objr-r-r.test_safe_is_pandas_categoricals rcCst|r dSt||S)NF)rr*Z issubdtype)Zdt1Zdt2r-r-r.rscCsvtttjstttttjs&ttttjr6tttttjrLttrrt j ddgdd}t|tjrrtdS)Nr,rVr)r") rintr*rmrAr"rHrnrr%r&)Z bad_dtyper-r-r.test_safe_issubdtypesrcOs tddS)NzeSorry, pickling not yet supported. See https://github.com/pydata/patsy/issues/26 if you want to help.)r)rrr-r-r.rscCs&ddl}ddlm}|t|j|dS)Nr)r^)picklerbr^rdumps)rkrr^r-r-r.rs cCst|tjr||kSdSdS)NF)r$rzZ string_types)rkrr-r-r.rs cCsLtddsttddrttjs4ttdds4tttddrHtdS)Nrr)r0r0)rrArzZPY3Zunicoder*emptyr-r-r-r.test_safe_string_eqs r)FNF)F)G__all__rZnumpyr*rzZ six.movesrrcompatrr% ImportErrorr r(rrrrtypesrgetattrZcorecommonr/rMrrZr]rerrlZ float_typerrAZ complex_typerrrrprrqrrrrrrr rr Z vectorizeboolr rrr rrrrrrrrrrrrrrrr-r-r-r.s       V @           #$ #