ó šxŠ\c@sddlZddlZddlmZddljZddlmZddlm Z ddl Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZddlmZddlmZddljjZ de!fd „ƒYZ"dS( iÿÿÿÿN(tiNaT(tPY3(tCategoricalDtype( tCategoricalIndext DatetimeIndext Float64IndextIndext Int64Indext IntervalIndext MultiIndext PeriodIndext RangeIndextSeriestTimedeltaIndext UInt64Indextisna(tInvalidIndexError(tDatetimeIndexOpsMixintBasecBsÌeZdZdEZddddgZd„Zd„Zd„Zd„Z e j j d dEd gƒd „ƒZ d „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z!e j j d e"e#e$j%e&gƒd!„ƒZ'e j j d"d#d$gƒe j j d%d&d'd(d)gƒd*„ƒƒZ(d+„Z)d,„Z*e j j d-dEe+gƒd.„ƒZ,d/„Z-d0„Z.d1„Z/d2„Z0d3„Z1d4„Z2d5„Z3d6„Z4d7„Z5d8„Z6d9„Z7d:„Z8e j j d;d<„d=„gƒd>„ƒZ9d?„Z:e j j d@e;e+gƒe j j d dEdAgƒe j j dBe;e+gƒdC„ƒƒƒZ<dD„Z=RS(Fs& base class for index sub-class tests tshapetndimtsizetnbytescCs4x-|jjƒD]\}}t|||ƒqWdS(N(tindicestitemstsetattr(tselftnametidx((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyt setup_indicesscCstjt|jƒdS(N(tpytesttraisest TypeErrort_holder(R((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_pickle_compat_constructionscCsa|jƒ}|jƒ}|j|jk s0t‚|j|k sEt‚|j|jks]t‚dS(N(t create_indext to_seriestvaluestAssertionErrortindexR(RRts((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_to_series#s   cCsÊ|jƒ}|jd|ƒ}|j|jk s6t‚|j|ksKt‚|j|jksct‚|jƒ}|jddƒ}|j|jk s™t‚|j|k s®t‚|j|jksÆt‚dS(NR'Rt__test(R#R$R%R&R'R(RRR(((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_to_series_with_arguments,s  Rtnew_namecCsÒ|jƒ}|r|}n|jp'd}|jd|ƒ}|j|ksQt‚t|jƒdkslt‚|jd|ks…t‚||j|jk s¡t‚|jdtd|ƒ}|j|k sÎt‚dS(NiRiR'( R#Rtto_frameR'R&tlentcolumnsR%tFalse(RRRtidx_nametdf((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyt test_to_frame?s  cCs?|jƒ}tjt|jdƒtjt|jddƒdS(Nii(R#RRtNotImplementedErrortshift(RR((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyt test_shiftSs c Cs1|jƒ}t|tƒstd|_tj|ƒ}tj||ƒtj|ddƒ}d|_tj||ƒn¹ddg|_tj|ƒ}tj|ttd dddddgdd ƒd ddgƒƒtj|d d d gƒ}tj|ttddddddgdd ƒd d d gƒƒdS(NtfooRtbartonettwotbaztquxtdtypetobjecttnamestAtB(R7R9(R7R:(R8R9(R;R:(R<R9(R<R:(R7R9(R7R:(R8R9(R;R:(R<R9(R<R:( R#t isinstanceR RtpdRttmtassert_index_equalR?(Rtexpectedtresult((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_create_index_existing_nameZs,     c Cs |jƒ}tjtddƒ |dWdQXtjtddƒ d|WdQXtr`dnd}tjtd|ƒ |dWdQX|jddƒ}tjtd|ƒ d|WdQXtjtdd ƒ |dWdQXtjtdd ƒ d|WdQXdS( Ntmatchscannot perform __mul__iscannot perform __rmul__scannot perform __truediv__scannot perform __div__s __s __rscannot perform __floordiv__scannot perform __rfloordiv__(R#RRR Rtreplace(RRtdiv_err((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_numeric_compatxs   cCs\|jƒ}tjtddƒ|jƒWdQXtjtddƒ|jƒWdQXdS(NRIscannot perform allscannot perform any(R#RRR talltany(RR((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_logical_compats  cCs5|jƒ}tjtddƒ|r+nWdQXdS(NRIsThe truth value of a(R#RRt ValueError(RR((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_boolean_context_compat”s cCsy|jƒ}tj|jdtjƒ}|j|ƒ}tj||ƒtj t ddƒ|j|ddƒWdQXdS(NR=RIsInvalid fill methodtmethodtinvalid( R#tnptarangeRtintpt get_indexerRDtassert_numpy_array_equalRRRP(RRRFtactual((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_reindex_bases  c Csx|jjƒD]\}}t|tƒr1qn|jsIt|tƒr’|j|dd!ƒ}t|tjƒswt ‚|j tj ksÈt ‚n6d}t j td|ƒ|j|dd!ƒWdQX|j|dd!ƒ\}}t|tjƒsüt ‚|j tj kst ‚qWdS(Niis8Reindexing only valid with uniquely valued Index objectsRI(RRRBRt is_uniqueRRWRTtndarrayR&R=RVRRRtget_indexer_non_unique(RRR'tindexertet_((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_get_indexer_consistency§scCs•|jƒ}|jj|ƒs$t‚|jƒj|ƒs?t‚|j}x5|jD]*}t||ƒt||ƒksRt‚qRW|j|jjdS(N( R#tTtequalsR&t transposeR%t _compat_propstgetattrR(RRR%tprop((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_ndarray_compat_propertiesºs  (cCs,|jƒ}tjtt|ƒƒ|ƒdS(N(R#RDREtevaltrepr(RR((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_repr_roundtripÇs cCsO|jƒ}d|_dt|ƒks-t‚|jjt|ƒksKt‚dS(NR7s'foo'(R#RtstrR&t __class__t__name__(RR((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_strÌs  cCsZ|jƒ}|jdƒ}tjddƒ't|ƒdt|ƒksPt‚WdQXdS(Ni2sdisplay.max_seq_itemss...(R#trepeatRCtoption_contexttNoneRjRlR&(RR((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_repr_max_seq_item_settingÔs   cCsxtj|jƒD]þ\}}t|tƒr4qn|j|dtddƒ}|j|dtƒ}||k svt‚|j |ƒs‹t‚|j dks t‚|j dksµt‚t dd|ƒ}t dd|d ƒ}t|t ƒs||}|j j dkst‚qqWdS(NtcopyRtmarioiR'iiÿÿÿÿ(tcompatt iteritemsRRBR RmtTrueR0R&RcRR RR'(RRR'tfirsttsecondts1ts2ts3((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_copy_nameÜs cCsbx[tj|jƒD]G\}}i}t|tƒrD|j|d( RvRwRt memory_usageR.tget_locRxRBR RR&t inferred_type(RRR'RGtresult2tresult3((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_memory_usages   cCsmxf|jjƒD]U\}}|dkr.qn|jƒ}tj|ƒjƒ}tj||dtƒqWdS(NtcatIndext check_dtype(R(RRtargsortRTtarrayRDRXR0(RtktindRGRF((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyt test_argsort3s   c Cs x|jjƒD]ô\}}tj|ƒ}|jƒ}tj||ƒtt|ƒtt fƒrd}t j t d|ƒtj|ddƒWdQXd}t j t d|ƒtj|ddƒWdQXd}t j t d|ƒtj|d d ƒWdQXqqWdS( Ns%the 'axis' parameter is not supportedRItaxisis%the 'kind' parameter is not supportedtkindt mergesorts&the 'order' parameter is not supportedtordertatb(RšR›( RRRTR‘RDRXRBttypeRR RRRP(RR“R”RGRFtmsg((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_numpy_argsort>s  cCs®ddddg}x•|jjƒD]„\}}|dkr@q"n|j|ƒ}||}|j|ƒsnt‚t|tttfƒs"t j t ƒ |j WdQXq"q"WdS( Niiiit boolIndexttuplestempty(RŸR R¡( RRttakeRcR&RBRR R RRtAttributeErrorR(RR^R“R”RGRF((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyt test_takeXs  c Cs»|jƒ}ddg}d}tjtd|ƒ|j|ddƒWdQXd}tjtd|ƒ|j|d|ƒWdQXd}tjtd|ƒ|j|d d ƒWdQXdS( Niis1take\(\) got an unexpected keyword argument 'foo'RIR7s$the 'out' parameter is not supportedtouts%the 'mode' parameter is not supportedtmodetclip(R#RRR R¢RP(RRRR((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_take_invalid_kwargsjs  cCs±d}|jƒ}tj|jj|ƒd|jƒ}tj|j|ƒ|ƒ|jƒ}tj t |ƒƒ}tj|jj|ƒd|jƒ}tj|j|ƒ|ƒdS(NiR( R#RCRR%RpRRDRERTRUR.(RtreptiRF((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyt test_repeatzs $ $c Csyd}|jƒ}|j|ƒ}tjtj||ƒ|ƒd}tjtd|ƒtj||ddƒWdQXdS(Nis%the 'axis' parameter is not supportedRIR–i(R#RpRDRERTRRRP(RR©RªRFR((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_numpy_repeat…s tklasscCs½|jƒ}tgt|ƒ}|j||ƒƒ}|}tj||ƒtgtgt|dƒ}tj|j g|dj ƒd|j ƒ}|j||ƒƒ}tj||ƒdS(NiR=( R#RxR.twhereRDRER0RCRt _na_valuettolistR=(RR­RªtcondRGRF((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyt test_wheres ,tcasegà?txxxRRt intersectiontuniont differencetsymmetric_differencecCs\xUtj|jƒD]A\}}d}tjtd|ƒt||ƒ|ƒWdQXqWdS(Ns!Input must be Index or array-likeRI(RvRwRRRR Rf(RR³RRRRR((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_set_ops_error_casessc Cswxptj|jƒD]\\}}|d }|d }|j|ƒ}t|tƒrTntj||ƒslt‚gt j t t gD]}||j ƒ^q}xˆ|D]€}t|tƒrèd} tjtd| ƒ|j|ƒWdQXq¡t|tƒrúq¡|j|ƒ} tj| |ƒs¡t‚q¡Wt|tƒrd} tjtd| ƒ|jdddgƒWdQXqqWdS(Niis/can only call with other PeriodIndex-ed objectsRIs.other must be a MultiIndex or a list of tuplesii(RvRwRRµRBRRDt equalContentsR&RTR’R tlistR%R RRRPR R ( RRRRyRzt intersectR­tcasesR³RRG((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_intersection_base¨s,  + c Cskxdtj|jƒD]P\}}|d}|d }|}|j|ƒ}tj||ƒs`t‚gtjt t gD]}||j ƒ^qs}xˆ|D]€} t |t ƒrÜd} tjtd| ƒ|j| ƒWdQXq•t |tƒrîq•|j| ƒ} tj| |ƒs•t‚q•Wt |tƒrd} tjtd| ƒ|jdddgƒWdQXqqWdS(Niis/can only call with other PeriodIndex-ed objectsRIs.other must be a MultiIndex or a list of tuplesii(RvRwRR¶RDRºR&RTR’R R»R%RBR RRRPRR R ( RRRRyRzt everythingR¶R­R½R³RRG((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_union_baseÆs*  + tsortc CsßxØtj|jƒD]Ä\}}|d}|d }|d}|j||ƒ}t|tƒrantj||ƒsyt‚gt j t t gD]}||j ƒ^qŒ} xà| D]Ø} t|tƒrød} tjtd| ƒ|j| |ƒWdQXq®t|tƒr q®t|ttfƒr\|j|jks7t‚tj|jƒj|jƒjƒq®|j| |ƒ}tj||ƒs®t‚q®Wt|tƒrd} tjtd| ƒ|jdddg|ƒWdQXqqWdS(Niis/can only call with other PeriodIndex-ed objectsRIs.other must be a MultiIndex or a list of tuplesii(RvRwRR·RBRRDRºR&RTR’R R»R%R RRRPRR RmRXt sort_valuesR„R R ( RRÁRRRyRztanswerRGR­R½R³R((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_difference_baseás6   + c Cs‡x€tj|jƒD]l\}}|d}|d }t|tƒrEn7|ddg}|j|ƒ}tj||ƒs|t‚gt j t t gD]}||j ƒ^q}xˆ|D]€} t|tƒrød} tjtd| ƒ|j| ƒWdQXq±t|tƒr q±|j| ƒ}tj||ƒs±t‚q±Wt|tƒrd} tjtd| ƒ|jdddgƒWdQXqqWdS( Niiÿÿÿÿis/can only call with other PeriodIndex-ed objectsRIs.other must be a MultiIndex or a list of tuplesii(RvRwRRBRR¸RDRºR&RTR’R R»R%R RRRPR R ( RRRRyRzRÃRGR­R½R³R((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_symmetric_differences.  + cCsrxktj|jƒD]W\}}|dd!}t|ƒs>qn|dd!j|jd|dƒƒst‚qWdS(Niii(RvRwRR.RctinsertR&(RRRRG((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_insert_base#s   cCs xtj|jƒD]î\}}t|ƒs1qnt|tƒrFqn|d}|jdƒ}|j|ƒstt‚|j |j ksŒt‚|d }|jdƒ}|j|ƒsºt‚|j |j ksÒt‚t j t t fƒ|jt|ƒƒWdQXqWdS(Niiiÿÿÿÿ(RvRwRR.RBR tdeleteRcR&RRRt IndexErrorRP(RRRRFRG((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_delete_base.s   cCs/x(tj|jƒD]\}}|j|ƒs4t‚|j|jƒƒsOt‚|j|jtƒƒsmt‚|jt|ƒƒ s‰t‚|jt j |ƒƒ s¨t‚t |t ƒsöt |dtƒ}|j|ƒsÞt‚|j|ƒsöt‚n|jdkr|jtj|ƒƒ s't‚qqWdS(NR=i(RvRwRRcR&RttastypeR>R»RTR’RBR RtnlevelsRCR (RRRt same_values((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyt test_equalsGsc Csr|jƒ}t|tƒr+tjdƒnt|ƒ}|dd!}|dd!j|dd!ƒ}|dd!}d}tjtd|ƒ||kWdQXt j t g|ƒ}t j t g|dt gƒ}t j||k|ƒt j||k|ƒt j |ƒ} t j |dd!ƒ} t j |dd!j|dd!ƒƒ} t j |dd!ƒ} tjtd|ƒ|| kWdQXt j|| k|ƒt j|| k|ƒt| ƒ} t| ƒ}t| ƒ}t| ƒ}tjtd|ƒ||kWdQXt j|| k|ƒt j||k|ƒtjtddƒ||kWdQXtjtddƒ||kWdQXtjtddƒ|| kWdQXd }tjtd|ƒ| |kWdQXtjtddƒ| | kWdQXt|tƒsnt j t gt|ƒd t t gƒ}|d}t j||k|ƒt j| |kt|ƒƒndS( NsSkip check for PeriodIndexiiÿÿÿÿiþÿÿÿis)Lengths must match|could not be broadcastRIsLengths must matchs3Can only compare identically-labeled Series objectsi(R#RBR RtskipR.tappendRRPRTR’RxR0RDRXR R tassert_series_equal(Rtindex_atntindex_btindex_ctindex_dRt expected1t expected2tarray_atarray_btarray_ctarray_dtseries_atseries_btseries_ctseries_dt expected3titem((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_equals_op[s\    !&    * cCsÆx¿tj|jƒD]«\}}x´tjtjtjtjtjtj tj tj tj tj tjtjtjtjtjtjtjtjtjtjtjtjgD](}t|tƒrûtjtƒ(tjddƒ||ƒWdQXWdQXqªt|t t!t"fƒrtjddƒ/||ƒ}t#||j$ƒd|j%ƒ}WdQXt&j'||ƒt|t(j ƒsÒt)‚qªt*|ƒdkr™qªqªtjtƒ(tjddƒ||ƒWdQXWdQXqªWxåtj+tj,tj-tj.gD]Å}t|tƒr-tjtƒ||ƒWdQXqõt|t t!t"fƒr‚||ƒ}t|tj/ƒsit)‚t|t#ƒ sºt)‚qõt*|ƒdkršqõqõtjtƒ||ƒWdQXqõWqWdS(NRMtignoreRi(0RvRwRRTtexptexp2texpm1tlogtlog2tlog10tlog1ptsqrttsintcosttantarcsintarccostarctantsinhtcoshttanhtarcsinhtarccoshtarctanhtdeg2radtrad2degRBRRRt ExceptionterrstateRRRRR%RRDRERCR&R.tisfinitetisinftisnantsignbitR\(RRRtfuncRGRå((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_numpy_ufuncs›sB'$ $% cCs“xŒ|jjƒD]{\}}t|tƒr.q|jƒ}tjtgt|ƒdt ƒ}t j |j |ƒ|j tks„t‚|jƒ}tj|jƒ}t|ƒdkrºqnDt|tƒrÖt|dGs(R#RBRCRRËtmapRDRE(RR'RFRG((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_map=s  tmappercCsd„t||ƒDƒS(NcSsi|]\}}||“qS(((t.0R_Rª((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pys Ms (tzip(R%R'((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyRMscCstj||ƒS(N(RCR (R%R'((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyRNscCsû|jƒ}t|tjtjfƒrFtjdjt|ƒƒƒn||j |ƒ}t|tj ƒr‹t|t ƒr‹|j dƒ}n|}|j |ƒ}tj||ƒtjtjgt|ƒƒ}|j |||ƒƒ}tj||ƒdS(Nsskipping tests for {}R(R#RBRCRRRRÏtformatRœR%RtdictRËRRDRERRTRR.(RRR'tidentityRFRG((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_map_dictlikeJs !c Cs´|jƒ}tjtƒ.|jtjt|ƒdtjƒdƒWdQXtjtƒ.|jtjt|ƒdtjƒdƒWdQXtjtƒ|jddƒWdQXdS(NiR7( R#RRRPtputmaskRTR R.R(RR'((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_putmask_with_wrong_maskes //RtR7torderedcCs|jƒ}|r$|j|ƒ}ntd|ƒ}|j|d|ƒ}t|jd|d|ƒ}tj||ƒt|jƒj ƒd |ƒ}|j|d|ƒ}t|jd|d|ƒ}tj||ƒ|t kr|jdd|ƒ}t|jd|ƒ}tj||ƒndS(NR#RtRiÿÿÿÿR=tcategory( R#trenameRRËRR%RDREtuniqueR°R0(RRtRR#R'R=RGRF((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_astype_categoryrs  cCsÉ|jƒjƒ}|jtks't‚|d }|jtksFt‚|jd|dƒ}|jtksqt‚|jdtjƒ}|jtks›t‚|jdtjƒ}|jtksÅt‚dS(Ni( R#tdrop_duplicatesR[RxR&RÆR0RTR(RR't index_emptyt index_duptindex_nat index_na_dup((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyttest_is_uniques N(>Rnt __module__t__doc__RrR!ReRR"R)R+Rtmarkt parametrizeR3R6RHRLRORQRZRaRhRkRoRsR~R‡RŽR•RžR¤R¨R«R¬R»ttupleRTR’R R²R¹R¾RÀR0RÄRÅRÇRÊRÎRãRRR RRRRR R"RxR'R-(((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyRsp   $          &     -  $$    @ 2 # (     *(#tnumpyRTRtpandas._libs.tslibRt pandas.compatRvRtpandas.core.dtypes.dtypesRtpandasRCRRRRRRR R R R R RRtpandas.core.indexes.baseRt pandas.core.indexes.datetimelikeRtpandas.util.testingtutilttestingRDR>R(((s:lib/python2.7/site-packages/pandas/tests/indexes/common.pyts   X