U
|
ý°dm ã @ s^ d dl Z d dlmZ d dlZd dlZd dlmZ d dl Z
|
d dl m Z G dd dZdd ZdS )é N)Úcatch_warnings)Úgenericc @ s, e Zd ZdddgdddggZejjeddZe ddg¡Z ej
|
e d ¡d
|
d Z ejdd d dZejdddgdddgdZejddddgiedZejddddgiedZej e j d¡¡Zejj e ¡Z ejj e ¡Z defde fde d¡fde fd ej d dd fd ejj!d d!d"gd#dfd$ej "e #d ddg¡¡fd%efd&ej$fd'e %dddg¡fd(efd)efd*e fd+e fgZ&e'j( )d,e&¡e'j( )d-e&¡d.d/ Z*e'j( )d,e&¡e'j( )d-e&¡d0d1 Z+d2d3 e&D d'd(gd)d*d d+gd4Z,e'j( )d5e, -¡ ¡e'j( )d6e&¡d7d8 Z.e'j( )d9d:d3 e/j0D ¡d;d< Z1d=S )>ÚTestABCClassesé é ÚredÚblue)ÚnumberÚcolor)Únamesz2000/1/1z2010/1/1é Ús)Úunitz 2010/1/1/ÚM)Úfreqé )Ú
|
categoriesÚvalues©Úindexr ÚaÚbÚcé
|
Z ABCMultiIndexZABCDatetimeIndexZ ABCRangeIndexZABCTimedeltaIndexZABCIntervalIndexr )ÚstartÚendZABCPeriodArrayiÐ iÑ iÒ ÚDZABCPandasArrayZABCPeriodIndexZABCCategoricalIndexZ ABCSeriesZ ABCDataFrameZABCCategoricalZABCDatetimeArrayZABCTimedeltaArrayzabctype1, instz abctype2, _c C sN ||kr6t |tt|s tt t|tt|rJtnt |tt|rJtd S ©N)Ú
|
isinstanceÚgetattrÚgtÚAssertionErrorÚtype©ÚselfZabctype1Zabctype2ÚinstÚ_© r' úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/dtypes/test_generic.pyÚ test_abc_pairs_instance_check+ s z,TestABCClasses.test_abc_pairs_instance_checkc C sn ||krRt t|tt|s ttjtt d¡d t |tt| W 5 Q R X nt t|tt|rjtd S )Nz"issubclass() arg 1 must be a class)Úmatch)
|
Ú
|
issubclassr" r r r! ÚpytestZraisesÚ TypeErrorÚreÚescaper# r' r' r( Ú test_abc_pairs_subclass_check5 s ÿ z,TestABCClasses.test_abc_pairs_subclass_checkc C s$ g | ] \}}d |kr|dkr|qS )ZIndexÚABCIndexr' )Ú.0Úabctyper& r' r' r( Ú
|
<listcomp>D s þzTestABCClasses.<listcomp>)r1 Z
|
ABCNDFrameZABCExtensionArrayz parent, subsz abctype, instc C s6 ||kr t |tt|s2tnt |tt|r2td S r )r r r r! )r$ ÚparentÚsubsr3 r% r' r' r( Útest_abc_hierarchyR s z!TestABCClasses.test_abc_hierarchyr3 c C s g | ]}| d ¡r|qS )ÚABC)Ú
|
startswith)r2 Úer' r' r( r4 [ s
|
c C s& |dd | j D ks"|| jks"td S )Nc s s | ]\}}|V qd S r r' )r2 r: r& r' r' r( Ú <genexpr>_ s z3TestABCClasses.test_abc_coverage.<locals>.<genexpr>)Ú abc_pairsÚabc_subclassesr! )r$ r3 r' r' r( Útest_abc_coverage[ s ÿÿz TestABCClasses.test_abc_coverageN)2Ú__name__Ú
|
__module__Ú __qualname__ZtuplesÚpdZ
|
MultiIndexZ from_arraysZ multi_indexÚ to_datetimeZdatetime_indexZ to_timedeltaÚnpZarangeZtimedelta_indexZ period_rangeZ period_indexZ CategoricalZ categoricalÚ DataFrameZcategorical_dfÚdfZarraysZ SparseArrayÚrandomZrandnZ sparse_arrayÚcoreZ DatetimeArrayZdatetime_arrayZTimedeltaArrayZtimedelta_arrayZ
|
RangeIndexZinterval_rangeZ PeriodArrayZ PandasArrayÚarrayr ÚSeriesr< r, ÚmarkZ parametrizer) r0 r= Úitemsr7 r Ú__dict__r> r' r' r' r( r s^ ò þüù r c C sL t jdddgdddgdt jddddgdddd gdd
|
} t | ¡}td d <}|jd |d< t|dksnt|j ¡ |j ¡ kstW 5 Q R X td d 8}| j d 7 _ t|dksºt|j j
|
d dksÎtW 5 Q R X td d $}d dg|j_ t|dkstW 5 Q R X t t¡, |jd |_|j ¡ |j ¡ ks>tW 5 Q R X d S )Ng ð?g @g @r r r r g @Úd)ÚoneÚtwoT)Úrecordr Úthreer r )rB rJ rE r rP Úlenr! rR ÚsumrO ZilocZ not_an_indexÚtmZassert_produces_warningÚ UserWarningZfour)rN rF Úwr' r' r( Útest_setattr_warningsc s$ þ
|
" rX )r. Úwarningsr ÚnumpyrD r, Zpandas.core.dtypesr r ZpandasrB Zpandas._testingZ_testingrU r rX r' r' r' r( Ú<module> s V
|
