/usr/local/lib64/python3.6/site-packages/pandas/tests/indexes/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/indexes/__pycache__/common.cpython-36.pyc (24309B)
3
:%Eg} @ s d dl Z d dlmZ d dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z
d dlZd dlmZmZmZmZmZmZmZmZmZmZmZmZ d dl j Z d dl m Z G d d
d
Z!dS ) N)Type)iNaT)InvalidIndexError)is_datetime64tz_dtype)CategoricalDtype) CategoricalIndex
DatetimeIndexIndex
Int64Index
IntervalIndex
MultiIndex PeriodIndex
RangeIndexSeriesTimedeltaIndex UInt64Indexisna)DatetimeIndexOpsMixinc @ s^ e Zd ZU dZee ddddgZedddZd d
Z e
j j d d d
gdd Z
dd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Z d.d/ Z d0d1 Z d2d3 Z d4d5 Z e
j j d6e!e"e#j$e%gd7d8 Z&e
j j d9d:d;ge
j j d
d?d@gdAdB Z'dCdD Z(dEdF Z)dGdH Z*dIdJ Z+dKdL Z,dMdN Z-dOdP Z.dQdR Z/dSdT Z0dUdV Z1dWdX Z2dYdZ Z3d[d\ Z4d]d^ Z5d_d` Z6dadb Z7e
j j dcddde dfde gdgdh Z8didj Z9dkdl Z:e
j j dmdndoge
j j d d dpge
j j dqdndogdrds Z;dtdu Zdzd{ Z?d|d} Z@d~d ZAd S )Basez& base class for index sub-class tests shapendimsizenbytes)returnc C s t dd S )NzMethod not implemented)NotImplementedError)self r G/usr/local/lib64/python3.6/site-packages/pandas/tests/indexes/common.py create_index& s zBase.create_indexc C s* d}t jt|d | j W d Q R X d S )NzIndex\(\.\.\.\) must be called with a collection of some kind, None was passed|__new__\(\) missing 1 required positional argument: 'data'|__new__\(\) takes at least 2 arguments \(1 given\))match)pytestraises TypeError_holder)r msgr r r test_pickle_compat_construction) s z$Base.test_pickle_compat_constructionnameNnew_namec C s | j }|r|}n
|jpd}|j|d}|j|ks6tt|jdksHt|jd |ksZt|| j|jk snt|jd|d}|j|k std S )Nr )r& F)indexr& )r r& Zto_framer) AssertionErrorlencolumnsvalues)r r& idxZidx_namedfr r r
test_to_frame4 s
zBase.test_to_framec C sf | j }dt|j }tjt|d |jd W d Q R X tjt|d |jdd W d Q R X d S )NzNot supported for type )r r( )r type__name__r r! r shift)r r. r$ r r r
test_shiftH s zBase.test_shiftc C s6 | j }tjtdd t||g d W d Q R X d S )Nz"Index.name must be a hashable type)r )r& )r r r! r" r2 )r r. r r r test_constructor_name_unhashableR s z%Base.test_constructor_name_unhashablec C s | j }t|tsPd|_tj|}tj|| tj|dd}d|_tj|| n~ddg|_tj|}tj|ttd
dddddgdd ddgd
tj|d d gd
}tj|ttddddddgdd d d gd
d S )Nfoobar)r& onetwobazquxobject)dtype)namesAB)r7 r9 )r7 r: )r8 r9 )r; r: )r< r9 )r< r: )r7 r9 )r7 r: )r8 r9 )r; r: )r< r9 )r< r: ) r
isinstancer r& pdr tmassert_index_equalr? )r expectedresultr r r test_create_index_existing_nameY sF
z$Base.test_create_index_existing_namec , C s | j }t|t sttjtdd |d W d Q R X tjtdd d| W d Q R X d}tjt|d |d W d Q R X |jdd}tjt|d d| W d Q R X tjtdd |d W d Q R X tjtd d d| W d Q R X d S )
Nzcannot perform __mul__)r r( zcannot perform __rmul__zcannot perform __truediv__z __z __rzcannot perform __floordiv__z cannot perform __rfloordiv__)r rB r r* r r! r" replace)r r. Zdiv_errr r r test_numeric_compat s zBase.test_numeric_compatc C sP | j }tjtdd |j W d Q R X tjtdd |j W d Q R X d S )Nzcannot perform all)r zcannot perform any)r r r! r" allany)r r. r r r test_logical_compat s
zBase.test_logical_compatc C s\ | j }tj|jtjd}|j|}tj|| tj t
dd |j|dd W d Q R X d S )N)r> zInvalid fill method)r invalid)method) r nparanger intp get_indexerrD assert_numpy_array_equalr r!
ValueError)r r. rF actualr r r test_reindex_base s
zBase.test_reindex_basec C s t |trd S |js t |trR|j|dd }t |tjs@t|jtj kstn0d}t
j t |d |j|dd W d Q R X |j
|dd \}}t |tjst|jtj kstd S )Nr r1 z8Reindexing only valid with uniquely valued Index objects)r )rB r is_uniquer rS rP ndarrayr* r> rR r r! r Zget_indexer_non_unique)r r) indexere_r r r test_get_indexer_consistency s
z!Base.test_get_indexer_consistencyc C sj | j }|jj|st|j j|s*t|j}x&| jD ] }t||t||ks8tq8W |j |jj d S )N) r Tequalsr* Z transposer-
_compat_propsgetattrr )r r. r- propr r r test_ndarray_compat_properties s z#Base.test_ndarray_compat_propertiesc C s | j }tjtt|| d S )N)r rD rE evalrepr)r r. r r r test_repr_roundtrip s zBase.test_repr_roundtripc
C sF | j }|jd}tjdd t| dt|ks8tW d Q R X d S )N2 zdisplay.max_seq_itemsz...)r repeatrC Zoption_contextre strr* )r r. r r r test_repr_max_seq_item_setting s
z#Base.test_repr_max_seq_item_settingc C s t |trd S t||ddd}t||dd}||k s Z tz_localizeZ
tz_converttzr r
_with_freqrD rE asi8rT r )r r) Z init_kwargsZ
index_typerG r r r test_ensure_copied_data s&
z Base.test_ensure_copied_datac C s~ |j j |j }|jr(|dks$td S |j|d |j }|jdd}t|ttfsd||ksdt|j dkrz||ksztd S )Nr T)deepr= )
_engineZ
clear_mappingZ memory_usageemptyr* get_locrB r r Z
inferred_type)r r) rG Zresult2Zresult3r r r test_memory_usage s
zBase.test_memory_usagec C s8 t |trd S |j }tj|j }tj||dd d S )NF)Z check_dtype)rB r argsortrP arrayrD rT )r requestr) rG rF r r r test_argsort2 s
zBase.test_argsortc C s t j|}|j }tj|| tt|ttfrd}tj t
|d t j|dd W d Q R X d}tj t
|d t j|dd W d Q R X d}tj t
|d t j|d d W d Q R X d S )
Nz%the 'axis' parameter is not supported)r r( )axisz%the 'kind' parameter is not supportedZ mergesort)kindz&the 'order' parameter is not supportedab)order)r r ) rP r~ rD rT rB r2 r r r r! rU )r r) rG rF r$ r r r test_numpy_argsort; s
zBase.test_numpy_argsortc C st ddddg}t |dk r d S |j|}|| }|j|s )r rB rC r r rv r+ whererD rE r Z _na_valuetolistr> )r r r ZcondrG rF r r r
test_where s
$zBase.test_wherecaseg ?ZxxxrO intersectionunion
differencesymmetric_differencec
C s0 d}t jt|d t||| W d Q R X d S )Nz!Input must be Index or array-like)r )r r! r" ra )r r rO r) r$ r r r test_set_ops_error_cases s z Base.test_set_ops_error_casesc s t |trd S |d d }|d d |j }tj| s@tt|jrNd S fddtj t
t gD }x&|D ] }|j|}tj| sntqnW t |t rd}t
jt|d |jdddg W d Q R X d S ) Nr rl c s g | ]}| j qS r )r- ).0r )ro r r
s z/Base.test_intersection_base..z.other must be a MultiIndex or a list of tuples)r r( r1 )rB r r rD
equalContentsr* r r> rP r r listr r r! r" )r r) rn Z intersectcasesr rG r$ r )ro r test_intersection_base s
zBase.test_intersection_basec s |dd }|d d |}|j }tj||s6tt|jrDd S fddtjtt gD }x0|D ](}t
|t sd|j |}tj||sdtqdW t
|t rd}t
jt|d |j dddg W d Q R X d S ) Nrl r c s g | ]}| j qS r )r- )r r )ro r r r s z(Base.test_union_base..z.other must be a MultiIndex or a list of tuples)r r( r1 )r rD r r* r r> rP r r r rB r r r r! r" ) r r) rn Z
everythingr r r rG r$ r )ro r test_union_base s
zBase.test_union_basec s |dd }|d d t |ts*|j r0g }n |dd }|j |}tj||sXt fddtjt t
gD }xd|D ]\}t |t t frt
|t
|ksttj|j j|j j qx|j||}tj||sxtqxW t |trd}tjt|d |jdddg| W d Q R X d S ) Nr1 r c s g | ]}| j qS r )r- )r r )ro r r r s z-Base.test_difference_base..z.other must be a MultiIndex or a list of tuples)r r( rl )rB r Z
is_booleanr rD r r* rP r r r r r r2 rT Z sort_valuesrw r r r! r" ) r sortr) rn answerrG r r r$ r )ro r test_difference_base s&
zBase.test_difference_basec s t |trd S |dd }|d d |dd
g }|j }tj||sLt fddtjtt gD }x&|D ] }|j|}tj||sltqlW t |t
rd}t j t
|d |jdddg W d Q R X d S ) Nr( r c s g | ]}| j qS r )r- )r r )ro r r r s z2Base.test_symmetric_difference..z.other must be a MultiIndex or a list of tuples)r r1 rl rm rm )rB r r rD r r* rP r r r r r r! r" )r r) rn r rG r r r$ r )ro r test_symmetric_difference s
z Base.test_symmetric_differencec C s> |dd }t |sd S |dd j|jd|d s:td S )Nr( r r )r+ r_ insertr* )r r) rG r r r test_insert_base s zBase.test_insert_basec C s t |s d S t|trd S |dd }|jd}|j|s>t|j|jksNt|d d }|jd}|j|srt|j|jkstt |}d| d| }tjt |d |j| W d Q R X d S )Nr( r zindex z' is out of bounds for axis 0 with size )r rm rm )
r+ rB r deleter_ r* r& r r!
IndexError)r r) rF rG lengthr$ r r r test_delete_base s
zBase.test_delete_basec C s t |trd S |j|s t|j|j s.t|j|jtsBt|jt| sVt|jtj | sltt |t
st |td}|j|st|j|st|j dkr|jt
| std S )N)r> r( )rB r r_ r* rk astyper= r rP r r r Znlevelsr )r r) Z same_valuesr r r test_equals% s
zBase.test_equalsc < C s | j }t|tr tjd t|}|dd }|dd j|d
d }|dd }d}tjt|d ||k W d Q R X t j
dg| }t j
dg|d dg }t j ||k| t j ||k| t j
|} t j
|dd }
t j
|dd j|dd } t j
|dd } tjt|d ||
k W d Q R X t j || k| t j || k| t
| }
t
|
}t
| }t
| }tjt|d ||k W d Q R X t j ||
k| t j ||k| tjtd d ||k W d Q R X tjtd d ||k W d Q R X tjtd d || k W d Q R X d
}tjt|d |
|k W d Q R X tjtd d |
| k W d Q R X t|tst j
dgt|d ddg }|d }t j ||k| t j|
|kt
| d S )NzSkip check for PeriodIndexr r( r1 z)Lengths must match|could not be broadcast)r TFzLengths must matchz3Can only compare identically-labeled Series objectsrm rm rm rm rm r rm r )r rB r
r skipr+ appendr! rU rP r rD rT r r Zassert_series_equal)r Zindex_anZindex_bZindex_cZindex_dr$ Z expected1Z expected2Zarray_aZarray_bZarray_cZarray_dZseries_aZseries_bZseries_cZseries_dZ expected3itemr r r test_equals_op; s\
zBase.test_equals_opc C s* | j }dd |D }|j |ks&td S )Nc S s g | ] }t |qS r )ri )r xr r r r ~ s z$Base.test_format..)r formatr* )r r. rF r r r test_format{ s zBase.test_formatc C s4 | j g }|j g kst|jdddgks0td S )NT)r& )r# r r* )r Z empty_idxr r r test_format_empty s
zBase.test_format_emptyc C s t |trd S |jdd}tjdgt| td}tj|j | |j
dksNt |jdd}tj |j
}t|dkrvd S t |trt|d< n t |ttfrd S tj|d< t |trt|||jd}nJt||}tjdgt| td}d|d< tj|j | |j
dkst d S )NT)ry F)r> r r( )rs )rB r rk rP r r+ boolrD rT _isnanhasnansr* asarrayr- r r r
r nanr
r2 rs )r r) r. rF r- r r r test_hasnans_isnans s,
zBase.test_hasnans_isnansc C sl t |dkrnXt|trV|jdd}d}tjt|d |j|d W d Q R X n|jdd}|j|d }tj || ||k st
d}tjt |d |j|d g W d Q R X |jdd}t j
|j}t|trt|d< n t|ttfrd S t j|d< t|tr t|||jd}n t||}t jd gt | td
}d|d< tj|j| |jdksht
d S ) Nr T)ry z"isna is not defined for MultiIndex)r z"'value' must be a scalar, passed: r( )rs F)r> ) r+ rB r rk r r! r ZfillnarD rE r* r" rP r r- r r r
r r r
r2 rs r r rT r r )r r) r. r$ rG r- rF r r r test_fillna s8
zBase.test_fillnac C s t |dkr(tj|j tjg td nt|trb|j }d}t
j t |d |j W d Q R X nn|j
stj|j tjt |td tj|j tjt |td n*t|}tj|j | tj|j | d S )Nr )r> z"isna is not defined for MultiIndex)r )r+ rD rT r rP r r rB r rk r r! r r ZzerosZnotnaones)r r) r. r$ rG r r r
test_nulls s
zBase.test_nullsc C s* | j }|j st|d d js&td S )Nr )r r{ r* )r r) r r r
test_empty s zBase.test_emptyc C s0 | j }|jr,|j||d}||kj s,td S )N)how)r rX joinrK r* )r Z join_typer) Zjoinedr r r test_join_self_unique s zBase.test_join_self_uniquec C sB | j }t|tjr |jd}n|}|jdd }tj|| d S )Nint64c S s | S )Nr )r r r r s z Base.test_map..)r rB rC r r maprD rE )r r) rF rG r r r test_map s z
Base.test_mapmapperc C s dd t | |D S )Nc S s i | ]\}}||qS r r )r r[ r r r r
s z!Base...)zip)r- r) r r r r s z
Base.c C s t j| |S )N)rC r )r- r) r r r r s c C s | j }t|tjtjfr.tjdt| ||j|}t|tj r\t|t
r\|j d}n|}|j |}t
j|| tjtjgt| }|j |||}t
j|| d S )Nzskipping tests for r )r rB rC r r r r r2 r- r dictr r rD rE r rP r r+ )r r r) identityrF rG r r r test_map_dictlike s
zBase.test_map_dictlikec C s8 | j }|jt}tdd |D td}tj|| d S )Nc S s g | ] }t |qS r )ri )r r r r r r s z%Base.test_map_str..)r> )r r ri r r= rD rE )r r) rG rF r r r test_map_str s
zBase.test_map_strc C s | j }d}tjt|d$ |jtjt|d tjd W d Q R X tjt|d$ |jtjt|d tjd W d Q R X tjt|d |jdd W d Q R X d S )Nz,putmask: mask and data must be the same size)r r( r7 ) r r r! rU ZputmaskrP r r+ Zbool_)r r) r$ r r r test_putmask_with_wrong_mask s ((z!Base.test_putmask_with_wrong_maskrk TFr7 orderedc C s | j }|r|j|}t|d}|j||d}t|j||d}tj|| t|j j d d |}|j||d}t|j||d}tj|| |dkr|jd|d}t|j|d}tj|| d S )
N)r )rk )r& r r( )r& r> Fcategory)r& rm )
r renamer r r r- rD rE uniquer )r rk r& r r) r> rG rF r r r test_astype_category, s
zBase.test_astype_categoryc C s | j j }|jdkst|d d }|jdks4t|jd|d }|jdksRt|jdtj}|jdksnt|jdtj}|jdkstd S )NTr F)r Zdrop_duplicatesrX r* r rP r )r r) Z index_emptyZ index_dupZindex_naZ index_na_dupr r r test_is_uniqueG s zBase.test_is_uniquec C s6 | j }ttj|}|j ttj||ks2td S )N)r r+ gcZ
get_referrersrz r* )r r) Z nrefs_prer r r test_engine_reference_cycle\ s z Base.test_engine_reference_cyclec
C sN | j }tjtdd |d d d f }W d Q R X t|tjsJtt|d S )NF)Zcheck_stacklevel) r rD Zassert_produces_warning
FutureWarningrB rP rY r* r2 )r r. resr r r test_getitem_2d_deprecatedc s z Base.test_getitem_2d_deprecatedc C sj | j }d}tjt|d g |k W d Q R X djdddddg}tjt|d i |jk W d Q R X d S ) Nzunhashable type: 'list')r |zunhashable type: 'dict'z must be real number, not dictzan integer is requiredz\{\}z6pandas\._libs\.interval\.IntervalTree' is not iterable)r r r! r" r rz )r r. r$ r r r &test_contains_requires_hashable_raisesk s z+Base.test_contains_requires_hashable_raisesc C sp | j }|j|d |j }|j|jks.t|j|jk s>tx,|jj D ] \}}|j| |ksJt|qJW d S )Nr )r r| rk _cacher* items)r r. rk keyvalr r r test_copy_copies_cache~ s zBase.test_copy_copies_cachec C sp | j }|j|d |j }|j|jks.t|j|jk s>tx,|jj D ] \}}|j| |ksJt|qJW d S )Nr )r r| Z
_shallow_copyr r* r )r r. Z shallow_copyr r r r r test_shallow_copy_copies_cache s z#Base.test_shallow_copy_copies_cache)Br3
__module__ __qualname____doc__r r r# r` r r% r markZ parametrizer0 r5 r6 rH rJ rM rW r] rc rf rj rr rx r} r r r r r r r tuplerP r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r sx
5
@!'
r )"r typingr ZnumpyrP r Z pandas._libsr Z
pandas.errorsr Zpandas.core.dtypes.commonr Zpandas.core.dtypes.dtypesr ZpandasrC r r r r
r r r
r r r r r Zpandas._testingZ_testingrD Z pandas.core.indexes.datetimeliker r r r r r s 8