U
|
ý°d¢Ï ã @ sÐ d dl Z d dlmZmZmZmZ d dlmZ d dlm Z mZm
|
Z
|
d dl Z d dl Z d dl Z d dlmZ d dlZd dlZd dlmZ d dlmZ d dlZd dlZd dlmZ d dlm m Z d d l m Z d d
|
l m!Z!m"Z"m#Z#m$Z$ d dl%Z&d d l%m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5 d dl6m7Z8 d d l9m:Z:m;Z;m<Z<m=Z=m>Z> d ddgZ?ddddddddgZ@G dd dZAG dd dZBG d d d ZCG d d d ZDd d! ZEG d"d# d#ZFG d$d% d%ZGdS )&é N)Ú OrderedDictÚabcÚ defaultdictÚ
|
namedtuple)Úmake_dataclass)ÚdateÚdatetimeÚ timedelta)ÚIterator)Úma)Úmrecords)ÚIntCastingNaNError)Úis_integer_dtype)ÚDatetimeTZDtypeÚ IntervalDtypeÚ PandasDtypeÚ PeriodDtype)Ú CategoricalÚCategoricalIndexÚ DataFrameÚ DatetimeIndexÚIndexÚIntervalÚ
|
MultiIndexÚPeriodÚ
|
RangeIndexÚSeriesÚ TimedeltaÚ TimestampÚcutÚ
|
date_rangeÚisna)Ú DatetimeArrayÚ IntervalArrayÚ PeriodArrayÚ SparseArrayÚTimedeltaArrayÚfloat16Úfloat32Úfloat64Úuint8Úuint16Úuint32Úuint64Zint8Úint16Úint32Úint64c @ sÔ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j dddg¡dd Z dd Ze
|
j ddd dd dd dd d d d d d d d d d d d!d d"d g ¡d#d$ Ze
|
j dd%d d&d g¡d'd( Ze
|
j d)g ged*ed+fg g ged,ed+fd-d. d/D ged*ed+fg¡d0d1 Zd2d3 Zd4d5 Zd6d7 Zd8d9 Zd:d; Zejd<d= Zejd>d? Zd@dA Z dBdC Z e
|
j dDdEi gdEd*dFdGdHgdIi gg¡dJdK Z dLdM Z dNdO Z dPdQ Z!dRdS Z"dTdU Z#e
|
j dVe$j%dWge&dXe$ %d,dY g¡dZge$j%d[ge&dXe$ %d\g¡d,dY d, gg¡d]d^ Z'e
|
j dVe$j%d*ge$j(dXe$j%d*ge$j)dXe$j%d*ge$j*dXe$ (d*¡ge$ )d*¡ge$ *d*¡gg¡d_d` Z+dadb Z,dcdd Z-dedf Z.dgdh Z/didj Z0dkdl Z1dmdn Z2dodp Z3e
|
j dqd,e$j4drdsg¡dtdu Z5e
|
j dvd,e$j4dre6dwg¡dxdy Z7e
|
j dve$j4dre6dwg¡dzd{ Z8d|d} Z9d~d Z:dd Z;dd Z<dd
Z=dd Z>dd Z?dd Z@dd ZAdd ZBdd ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJe
|
j d d¡d d¢fd£d d¤fd¥d d¦fd§d d¨fg¡d©dª ZKd«d¬ ZLdd® ZMe
|
j d¯eNd°eOdfePd+d±d²eQd³d´feRdµd¶d·eSd¶d·fg¡d¸d¹ ZTdºd» ZUd¼d½ ZVd¾d¿ ZWdÀdÁ ZXdÂdà ZYdÄdÅ ZZdÆdÇ Z[dÈdÉ Z\e
|
j dÊdre]e^dËdÌdÍge&e$j_fdrdrdÌdÍgd³e$ `d³¡fdre]e^dËdÌdÍgeae$ `dΡfi drdFdGgdre$j_fdÍd*ie]e^dËe]dÏeae$ `dΡfg¡dÐdÑ Zbe
|
j dÒdÓdÔdrgdÕecjdfdÖd×drgdØecjefd*d,drgdÙecjffdÌdÍdÚgdÛecjgff¡dÜdÝ ZhdÞdß Zidàdá Zjdâdã Zkdädå Zldædç Zmdèdé Zndêdë Zodìdí Zpdîdï Zqdðdñ Zrdòdó Zse
|
j dôeRdõggdöeRdõigdöeRdõgidöeRdõig¡d÷dø Ztdùdú Zudûdü Zvdýdþ Zwdÿd Zxdd Zydd Zzdd Z{dd Z|d d
|
Z}d d Z~e
|
j d eeg¡dd Ze
|
j d eeg¡dd Zdd Zdd Zdd Z
dd Zdd Zd d Zd d Ze
|
j d d/g fd/g fd!d/d/gfd!g g gfg g fg g gfd*gd,gfd*gd,ggfd*d,d"gd#d±d$gfd*d,d"gd#d±d$ggfg¡d%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Ze
|
j d9d:d;d<d=d>d?d@g¡dAdB ZdCdD ZdEdF ZdGdH ZdIdJ ZdKdL ZdMdN ZdOdP ZdQdR ZdSdT ZdUdV ZdWdX ZdYdZ Z d[d\ Z¡d]d^ Z¢e
|
j d_e$ %drdrdrdre£ ¤¡ drg¡e$ %drdre£ ¤¡ drg¡e$ ¥d`¡gdrgge$ ¥d`¡gecj¦ggdrge$ ¥d`¡ggdrgecj¦ggecj¦ge$ ¥d`¡ggecj¦gdrggg¡dadb Z§e
|
j dcdddedfdgg¡e
|
j dhddsdiddjdkdldmg¡dndo Z¨e
|
j dcdddedfdgg¡e
|
j dhdsdiddjdkdldmg¡dpdq Z©drds Zªdtdu Z«dvdw Z¬dxdy Zdzd{ Z®e
|
j d|e¯d*d,d"gd#d±d$ggd*e$j4gd}e¯d*d,d"gd#d±d$ggd~de$j4gde¯d+d*d,d"gd#d±d$dgge$j4d~de$j4gde¯dd*d,dgd#d±d$dgge$j4d~de$j4gde¯dd*d,dgd#d±d$dgge$j4d*d,d,gdg¡dd
Z°dd Z±dd Z²dd Z³dd Z´dd Zµdd Z¶dd Z·dd Z¸e
|
j de¹jºe¹j» e¹j¼ e¹j½ ¡dd Z¾e
|
j de¹j¿e¹jÀ e¹jÁ ¡dd ZÂdd ZÃe
|
j jÄe
|
j ddrddg¡dd ZÅd¡d¢ ZÆe
|
j d£eÇe]d¤eÈd*e$j4e$j4e$j4geÉePd+d*ePd*d±geÊecjËd¥d¦dd§g¡d¨d© ZÌdªd« ZÍd¬d ZÎd®d¯ ZÏd°d± ZÐd²d³ ZÑd´dµ ZÒd¶d· ZÓd¸d¹ ZÔe
|
j dºd*d,gd»d¼ge]eÕd½d,d¾d¿e]eÕd½d,d¾d¶dÀePd+d±d²gg¡dÁd ZÖdÃdÄ Z×e
|
j dÅdÔdÓg¡dÆdÇ ZØdÈdÉ ZÙdÊdË ZÚdÌdÍ ZÛe
|
j dÎd*gd,gge$ %d*gd,gg¡fe$ %d*gd,gg¡d*gd,ggfe$ %d*gd,gg¡e$ %d*gd,gg¡fg¡dÏdÐ ZÜdÑdÒ ZÝdÓdÔ ZÞdrS (Õ ÚTestDataFrameConstructorsc C s< t d¡ dd¡}t|td}t| t¡}t ||¡ d S )Né é é ©Údtype)ÚnpÚarangeÚreshaper ÚstrÚastypeÚtmÚassert_frame_equal)ÚselfÚarrÚdfÚexpected© rB ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_constructors.pyÚ,test_constructor_from_ndarray_with_str_dtypeN s zFTestDataFrameConstructors.test_constructor_from_ndarray_with_str_dtypec C sr t dddd}|j dd¡}t|}t|d d
df |d d
df d }t ||¡ |snt|jjdksnt d S )
|
Nú
|
2016-01-01é ú
|
US/Pacific©ÚperiodsÚtzr4 é r é ©r rL )
|
r Ú_datar9 r r< r= ÚlenÚ_mgrÚblocksÚAssertionError)r> Úusing_array_managerÚdtiZdtar@ rA rB rB rC Ú&test_constructor_from_2d_datetimearrayV s & z@TestDataFrameConstructors.test_constructor_from_2d_datetimearrayc C sl t d d¡}td|idgd}td|gi}t ||¡ t|dgd}t|gdgd}t ||¡ d S )Nz2019-11-03 01:00:00-0700zAmerica/Los_AngelesÚdtr ©ÚindexrL )rV Úvalue)r Z
|
tz_convertr r< r= )r> rV r@ rA rB rB rC Ú)test_constructor_dict_with_tzaware_scalara s zCTestDataFrameConstructors.test_constructor_dict_with_tzaware_scalarc C sr t dt jgddgg¡}d}tjt|d t|dd W 5 Q R X tjt|d t|d ddd W 5 Q R X d S )
|
NrL rK r4 ú9Cannot convert non-finite values \(NA or inf\) to integer©ÚmatchÚi8r5 r ©r6 Úname)r7 ÚarrayÚnanÚpytestÚraisesr r r )r> r? ÚmsgrB rB rC Ú-test_construct_ndarray_with_nas_and_int_dtypen s zGTestDataFrameConstructors.test_construct_ndarray_with_nas_and_int_dtypec C s0 t t ¡ t ¡ g}|d jt d¡ks,td S )Nr úM8[ns])r r Únowr6 r7 rR ©r> r@ rB rB rC Ú%test_construct_from_list_of_datetimesy s z?TestDataFrameConstructors.test_construct_from_list_of_datetimesc C s\ t ddgdd}| d¡}ttj| ¡ dddd}|j|jksDtt|}t ||¡ d S )Nz2013-1-1 13:00z2013-1-2 14:00ÚB©r` rG Úobjectr5 )
|
r Ú tz_localizer r7 ra Útolistr6 rR r< Úassert_series_equal)r> ZnaiveÚidxrA ÚresultrB rB rC Ú+test_constructor_from_tzaware_datetimeindex} s
|
zETestDataFrameConstructors.test_constructor_from_tzaware_datetimeindexc C s^ t dd¡}t j|gtd}|tkr0| dd¡}d}tjt|d ||dd W 5 Q R X d S )NÚNaTÚnsr5 rL z=Invalid type for timedelta scalar: <class 'numpy.datetime64'>r\ úm8[ns]) r7 Ú
|
datetime64ra rm r r9 rc rd Ú TypeError)r> Úframe_or_seriesZnatr? re rB rB rC Ú,test_array_of_dt64_nat_with_td64dtype_raises s zFTestDataFrameConstructors.test_array_of_dt64_nat_with_td64dtype_raisesÚkindÚmÚMc C s| |dkrd}t }nd}t}tjddd |¡ dd¡}|tkrN|d d
d f }||td}|jj d j
|
tkspt t |jj d ¡ d |st |||td}|jj d j
|
tks´t t |jj d ¡ d |sÒt |||ttd}|jj d j
|
tksüt t |jj d ¡ d |s t |tkrxd
|
d |D }||td}|jj d j
|
tksXt t |jj d ¡ d |sxt d S ) Nr} rg rv rF r^ r5 r4 rK r c S s g | ] }t |qS rB ©r ©Ú.0ÚxrB rB rC Ú
|
<listcomp>² s zXTestDataFrameConstructors.test_datetimelike_values_with_object_dtype.<locals>.<listcomp>)r r r7 r8 Úviewr9 r rm rP Úarraysr6 rR Ú
|
isinstanceÚravelr r )r> r{ ry r6 Z scalar_typer? ÚobjZsersrB rB rC Ú*test_datetimelike_values_with_object_dtype s,
|
zDTestDataFrameConstructors.test_datetimelike_values_with_object_dtypec C st t tddd}t tddd}t|dgd}tg dgd}t ||¡ t|dgd}tg dgd}t ||¡ d S )Né rL rl r ©Úcolumns)r Úranger r< r= )r> r Úyrr rA rB rB rC Ú)test_series_with_name_not_matching_column· s zCTestDataFrameConstructors.test_series_with_name_not_matching_columnÚ constructorc C s t S ©N©r rB rB rB rC Ú<lambda>Ç ó z"TestDataFrameConstructors.<lambda>c C s t d S r r rB rB rB rC r È r c C s t dS ©NrB r rB rB rB rC r É r c C s t g S r r rB rB rB rC r Ê r c C s t dd dD S )Nc s s | ]
|
}|V qd S r rB ©r Ú_rB rB rC Ú <genexpr>Ë s ú5TestDataFrameConstructors.<lambda>.<locals>.<genexpr>rB r rB rB rB rC r Ë r c C s t tdS ©Nr ©r r rB rB rB rC r Ì r c C s
|
t d dS ©N©Údatar rB rB rB rC r Í r c C s
|
t ddS )NrB r r rB rB rB rC r Î r c C s
|
t g dS r r rB rB rB rC r Ï r c C s t dd dD dS )Nc s s | ]
|
}|V qd S r rB r rB rB rC r Ð s r rB r r rB rB rB rC r Ð r c C s t tddS )Nr r r rB rB rB rC r Ñ r c C s@ t }| }t|jdks tt|jdks0tt ||¡ d S r )r rO rX rR r r< r= ©r> r rA rr rB rB rC Útest_empty_constructorÄ s
|
z0TestDataFrameConstructors.test_empty_constructorc C s t i S r r rB rB rB rC r Þ r c C s
|
t i dS r r rB rB rB rC r ß r c C sR t tdtdd}| }t|jdks,tt|jdks>ttj||dd d S )Nr ©rX r T)Zcheck_index_type)r r rO rX rR r r< r= r rB rB rC Ú#test_empty_constructor_object_indexÛ s
|
z=TestDataFrameConstructors.test_empty_constructor_object_indexz)emptylike,expected_index,expected_columnsrL r rK c c s | ]
|
}|V qd S r rB r rB rB rC r î s z#TestDataFrameConstructors.<genexpr>rB c C s$ t ||d}t |}t ||¡ d S )Nr ©r r< r= )r> Z emptylikeZexpected_indexZexpected_columnsrA rr rB rB rC Útest_emptylike_constructoré s z4TestDataFrameConstructors.test_emptylike_constructorc C s8 t ¡ \}}t||d}t|}|d jtjks4td S )NrW Úfoo)r< ZgetMixedTypeDictr r6 r7 Úobject_rR )r> Zfloat_string_framerX r Z indexed_frameZunindexed_framerB rB rC Útest_constructor_mixedö s z0TestDataFrameConstructors.test_constructor_mixedc C s d}t jt|d tddddgitjd W 5 Q R X tt d¡}t d¡ ¡ |d< d }t jt|d t d¡|d
|
< W 5 Q R X t d¡ ¡ |d < d S ) Nz!could not convert string to floatr\ ÚaÚbÚcr5 )r3 rK r¤ z5Expected a 1D array, got an array with shape \(4, 2\)ÚtestZfoo2)rc rd Ú
|
ValueErrorr r7 r) Úonesro )r> re r@ rB rB rC Ú test_constructor_cast_failureÿ s "z7TestDataFrameConstructors.test_constructor_cast_failurec C sD t dgdgdgd}t |tdd}d|d< |d d dks@td S )
|
Nç ð?ç @ç @)Úcol1Úcol2Úcol3T)r6 Úcopyg i@r± r )r ÚfloatrR )r> Zorig_dfZnew_dfrB rB rC Útest_constructor_dtype_copy s z5TestDataFrameConstructors.test_constructor_dtype_copyc C sd t ddgg}t ||d jd}|rBd|jd< |jd dks`tn d|d d< |jd dks`td S )NrL rK r r5 éc ©r r )r r6 ÚilocÚvaluesrR )r> Úusing_copy_on_writer@ Úshould_be_viewrB rB rC Ú,test_constructor_dtype_nocast_view_dataframe s
|
zFTestDataFrameConstructors.test_constructor_dtype_nocast_view_dataframec C s t ddgddggdd}|sT|sTt |j|d jd}d|d d< |jd dks|tn(t |j|d jd}|jjd jjs|td S )
|
NrL rK r4 r3 r0 r5 r éa r¸ )r rº r6 rR rP r ÚflagsÚ c_contiguous)r> rS r» r@ r¼ Údf2rB rB rC Ú+test_constructor_dtype_nocast_view_2d_array& s zETestDataFrameConstructors.test_constructor_dtype_nocast_view_2d_arrayc C s4 t jddgdd}t|dd}t |j|¡s0td S )Nr§ r¨ rm r5 F©r´ ©r7 ra r Ú shares_memoryrº rR ©r> r? r@ rB rB rC Ú"test_1d_object_array_does_not_copy4 s z<TestDataFrameConstructors.test_1d_object_array_does_not_copyc C s< t jddgddggdd}t|dd}t |j|¡s8td S ) Nr§ r¨ r© Údrm r5 Frà rÄ rÆ rB rB rC Ú"test_2d_object_array_does_not_copy; s z<TestDataFrameConstructors.test_2d_object_array_does_not_copyc C s@ t ddgd dggtd}|jd d ks*t|jd dks<td S )NrL Ú2r§ r5 ©rL r rM )r rm ÚlocrR ri rB rB rC Ú test_constructor_dtype_list_dataB s z:TestDataFrameConstructors.test_constructor_dtype_list_datac C s t }t d¡}tjtdd t |g W 5 Q R X tjtdd t |g W 5 Q R X t dddgi}tjtdd t ||g W 5 Q R X d S )Nr¸ zshape=\(1, 0, 0\)r\ ÚArL rK zshape=\(2, 2, 1\))r r7 Úemptyrc rd r« )r> r§ r¨ rB rB rC Ú"test_constructor_list_of_2d_raisesG s
|
z<TestDataFrameConstructors.test_constructor_list_of_2d_raisesztyp, adrµ r¤ Úbar©rÎ rk ÚCÚintc C s¤ |dkr t }dd |D }n|dkr6t}dd |D }t||D ]\}}|j|ks@tq@| tt||¡ t|}tt }|D ] }||kr|j| |kstqd S )NrÔ c S s" g | ]}t jt j d ¡|dqS )é
|
r5 )r7 ra ÚrandomÚrand©r rÈ rB rB rC r c s zKTestDataFrameConstructors.test_constructor_mixed_dtypes.<locals>.<listcomp>rµ c S s& g | ] }t jt jjd d d|dqS )rÕ )Úsizer5 )r7 ra rÖ ÚrandintrØ rB rB rC r f s ) ÚMIXED_INT_DTYPESÚMIXED_FLOAT_DTYPESÚzipr6 rR ÚupdateÚdictr Údtypes)r> ÚtypÚadrà r rÈ r§ r@ rB rB rC Ú test_constructor_mixed_dtypesU s z7TestDataFrameConstructors.test_constructor_mixed_dtypesc C s^ t j d¡ t j¡}t j d¡ t j¡}t||d}|j|jjksHt |j|j
|
jksZt d S )NrÕ ©r§ r¨ ) r7 rÖ r× r; Z complex64Ú
|
complex128r r6 r§ rR r¨ ©r> r§ r¨ r@ rB rB rC Ú test_constructor_complex_dtypesr s
|
z9TestDataFrameConstructors.test_constructor_complex_dtypesc C sx t ddd gi|d}| ¡ }t dddgi}t ||¡ |jd d ksJtt ddtjgi|d}t |jd ¡sttd S )NrÎ r r5 FTrË ) r r! r< r= r¹ rR r7 rb Úisnan)r> Ú string_dtyper@ rr rA rB rB rC Ú$test_constructor_dtype_str_na_values{ s z>TestDataFrameConstructors.test_constructor_dtype_str_na_valuesc C sÌ |j dd}t|jjd d d
|j_|j}t|}t |jt |jj¡ t||d}t |jt |jj¡ t |j|¡ t
|
t |¡d d d
}t||ddgd}t||dj ddgd}t ||¡ d S )NFrW éÿÿÿÿrÓ rk r r )Z
|
to_recordsÚlistr6 ÚnamesrX r r< Úassert_index_equalr r r7 r8 rO Úreindexr= ) r> Ú float_frameZrecrX r@ rÁ ÚrngÚdf3rA rB rB rC Útest_constructor_rec s z.TestDataFrameConstructors.test_constructor_recc C s8 t tjdtdtjdtdd}|jjtjks4td S )NrÕ r5 rM ) r r7 r¬ ÚboolÚzerosrº r6 Úbool_rR ri rB rB rC Útest_constructor_bool s "z/TestDataFrameConstructors.test_constructor_boolc C s t jdd tddD t jd}td|i}|d jt jks@tddd d
|
d g}d d g}t jt|f|d}||d d
< t|}|d jt jkstd S )Nc S s g | ] }d | qS )ì rB ©r ÚirB rB rC r s zMTestDataFrameConstructors.test_constructor_overflow_int64.<locals>.<listcomp>rL rÕ r5 r§ )l Yª2< é )l FJLW]L* é )l ´MÏ:M
^ é- )l 8²Ma i )l r*R3_PvN rû )ÚuidÚu8)Zscorerÿ rþ ) r7 ra r r- r r6 rR rõ rO )r> rº rr Z data_scoresr6 r Z df_crawlsrB rB rC Ú test_constructor_overflow_int64 s û z9TestDataFrameConstructors.test_constructor_overflow_int64rº rø r5 éA l lûÿÿÿ lûÿÿÿ c C s: |d }t |}|d jtks"t|d d |ks6td S r )r r6 rm rR ©r> rº rY rr rB rB rC Ú test_constructor_int_overflow² s z7TestDataFrameConstructors.test_constructor_int_overflowc C s< |d }t |}|d j|jks$t|d d |ks8td S r ©r r6 rR r rB rB rC Ú test_constructor_numpy_uintsÅ s z6TestDataFrameConstructors.test_constructor_numpy_uintsc C sX d}t t|}t |¡ dd |D }ttt|dgg| }|t |jksTtd S )Néd c S s g | ]}d |dqS )rÎ rÈ rB rù rB rB rC r Ü s zJTestDataFrameConstructors.test_constructor_ordereddict.<locals>.<listcomp>r ) rì r rÖ Úshuffler r rÝ r rR )r> ZnitemsÚnumsrA r@ rB rB rC Ú test_constructor_ordereddictØ s
|
z6TestDataFrameConstructors.test_constructor_ordereddictc C s2 t jdd}t jdddd
}t||d}t|dks>tt|dksNtt |d | d¡¡ tt tj
|
gd |j g¡|j dd}t ||d ¡ t||ddd d
|
gd }t|t|ksÄtd|ksÐtt |d ¡ sättt d ksötd }tjt|d tddddddgd W 5 Q R X d S )Né ©Znperr ©r± r² é r± r² ©rX r` r³ Zcol4r r ú<Mixing dicts with non-Series may lead to ambiguous ordering.r\ r§ r¨ rä r© ©rÎ rk )r< ÚmakeTimeSeriesr rO rR rp Úrenamer r7 Z concatenaterb rº rX r! Úallrc rd r« )r> Údatetime_seriesÚdatetime_series_shortÚframeÚexpre rB rB rC Útest_constructor_dictà s. ýþ z/TestDataFrameConstructors.test_constructor_dictc C s, t ddddi}t |jtddg¡ d S )NrÎ rL rK ©Ú1rÊ r rÊ )r r< rî rX r ©r> r rB rB rC Ú test_constructor_dict_length1 s z7TestDataFrameConstructors.test_constructor_dict_length1c C s, t dddg}ti |d}|j|ks(td S )Nr rL rK rW )r r rX rR ©r> rq r rB rB rC Ú test_constructor_dict_with_index s z:TestDataFrameConstructors.test_constructor_dict_with_indexc C sN t dddg}ti ||d}|j|ks*t|j|ks8tt|jdksJtd S )Nr rL rK r r4 )r r rX rR r rO Ú_seriesr rB rB rC Ú,test_constructor_dict_with_index_and_columns s
|
zFTestDataFrameConstructors.test_constructor_dict_with_index_and_columnsc C s0 t g g dddgd}tj|jtddd d S )Nr rÎ rk r r T©Úexact)r r< rî rX r r rB rB rC Ú$test_constructor_dict_of_empty_lists s z>TestDataFrameConstructors.test_constructor_dict_of_empty_listsc C s^ t dd idgd}t dd gidgd}| dd¡d ks:t| dd¡d ksNtt ||¡ d S )Nr§ r rW )r Ú
|
_get_valuerR r< r= )r> Z
|
frame_noneZframe_none_listrB rB rC Ú test_constructor_dict_with_none s
|
z9TestDataFrameConstructors.test_constructor_dict_with_nonec C sZ d}t jt|d tddi W 5 Q R X t jt|d tddidgd W 5 Q R X d S )Nú2If using all scalar values, you must pass an indexr\ r§ gffffffæ?r ©rc rd r« r ©r> re rB rB rC Ú test_constructor_dict_errors' s
|
z6TestDataFrameConstructors.test_constructor_dict_errorsÚscalarNÚDc C s. t d|idgd}t dgd}t ||¡ d S )Nr§ r¨ r r¢ )r> r* rr rA rB rB rC Ú%test_constructor_invalid_items_unused1 s z?TestDataFrameConstructors.test_constructor_invalid_items_unusedrY rb c sÆ d|dg d|gddgddgddgg fdd t dD }t| d¡jddd
|
}ttjd d d dd¡ d}t ||¡ t|djddd
|
}t ||¡ t| d}t ||¡ d S )NrL r4 r§ r r3 rK r c s" i | ]} | t | d qS ©rW r~ ©r r© ©Úcolsrq rº rB rC Ú
|
<dictcomp>> s zKTestDataFrameConstructors.test_constructor_dict_nan_key.<locals>.<dictcomp>©ZaxisrF r0 r5 r rW )r r Ú sort_valuesr7 r8 r9 r< r= ©r> rY r rr rA rB r/ rC Ú test_constructor_dict_nan_key8 s
|
ÿ z7TestDataFrameConstructors.test_constructor_dict_nan_keyc sæ t d|dfd|fg t d|f|dfgddgdd gdd
|
gg fd d tdD }t| d¡jd|fdd }ttjddd dd¡ d}t ||¡ t|djd|fdd }t ||¡ t| d}t ||¡ d S )N)é é é é r§ rK r r4 rL r3 r c s" i | ]} | t | d qS r- r~ r. r/ rB rC r1 Q s zQTestDataFrameConstructors.test_constructor_dict_nan_tuple_key.<locals>.<dictcomp>r2 rF r0 r5 r rW ) r r r r3 r7 r8 r9 r< r= r4 rB r/ rC Ú#test_constructor_dict_nan_tuple_keyK s ÿ z=TestDataFrameConstructors.test_constructor_dict_nan_tuple_keyc C sL t jdd}t jdd}||d}t|d}t|tdd}t ||¡ d S )Nr
|
r r ©r¨ r§ r Úba©r r )r< r r rì r= )r> r r rÈ r rA rB rB rC Ú%test_constructor_dict_order_insertion^ s
|
|
z?TestDataFrameConstructors.test_constructor_dict_order_insertionc C sR t tjddgdddgitjdgd}t ddgddggtjdgd}t ||¡ d S )NrL rK r4 r )r r7 rb r< r= ©r> rr rA rB rB rC Ú)test_constructor_dict_nan_key_and_columnsi s $ zCTestDataFrameConstructors.test_constructor_dict_nan_key_and_columnsc C sp dddg}t |¡}t||d}t|j ¡ ¡ s6tdddg}t |¡}t||d}t|j ¡ ¡ sltd S )N©rK r4 )r4 r4 r )r Ú from_tuplesr r! rº r r rR )r> ÚtuplesÚmir@ rB rB rC Ú test_constructor_multi_indexo s
|
|
|
|
z6TestDataFrameConstructors.test_constructor_multi_indexc C s¶ t dggdggddgd}t ddgtddgddtdggdggdd}t ||¡ t dggdggddggd}t ddgtddggddggdtdggdggdd}t ||¡ d S ) NrL rK ©r rX r0 r5 r )ÚlevelsÚcodesr )r r r r< r= ©r> r@ rA rB rB rC Útest_constructor_2d_index| s ý ýz3TestDataFrameConstructors.test_constructor_2d_indexc C sV d}t jt|d tt d¡tdd W 5 Q R X d}t jt|d tddd ddd
|
gd W 5 Q R X d }t jt|d. tt d ¡ d¡dddgt dddd W 5 Q R X t
|
dddgg¡}d}t jt|d tdgt dd|d W 5 Q R X t
|
dddg¡}d}t jt|d tdgt dd|d W 5 Q R X t jtd d" tt d ¡d d d gd!gd W 5 Q R X d"}t jt|d& ttj d#d¡d d d gd!gd W 5 Q R X d$}t jt|d& ttj d#d¡d d gd!d#gd W 5 Q R X d%}t jt|d tt d&t d&d'fd(d W 5 Q R X d)}t jt|d td*d+d W 5 Q R X d S ),Nú)Empty data passed with indices specified.r\ r r r r r§ r¨ rä r© r z:Shape of passed values is \(4, 3\), indices imply \(3, 3\)r2 )r3 r4 r¤ rÑ Úbazú
|
2000-01-01r4 ©rI rF r3 r rF z:Shape of passed values is \(1, 3\), indices imply \(1, 4\)©rX r r z:Shape of passed values is \(3, 1\), indices imply \(1, 4\)úMust pass 2-d input©r4 r4 r4 rÎ rk rÓ rL ú:Shape of passed values is \(2, 3\), indices imply \(1, 3\)rK ú:Shape of passed values is \(2, 3\), indices imply \(2, 2\)z,2 columns passed, passed data had 10 columnsrÕ é )r¬ Ztwosr& FT)rc rd r« r r7 rÏ rì r8 r9 r ra r rõ rÖ r× )r> re r? rB rB rC Útest_constructor_error_msgs sF $
|
ý &**$z5TestDataFrameConstructors.test_constructor_error_msgsc C s| |dd t dD |dd t dD d}t|}tdd | ¡ D }t ||¡ || ¡ }t|}t ||¡ d S )Nc s s | ]}|d | fV qdS )g $@NrB r rB rB rC r Æ s zKTestDataFrameConstructors.test_constructor_subclass_dict.<locals>.<genexpr>rÕ c s s | ]}|d | fV qdS )g 4@NrB r rB rB rC r Ç s r c S s i | ]\}}|t | ¡ qS rB )rß Úitems)r ÚcolÚvalrB rB rC r1 Ê s zLTestDataFrameConstructors.test_constructor_subclass_dict.<locals>.<dictcomp>)r r rV r< r= )r> Z dict_subclassr r@ ZrefdfrB rB rC Ú test_constructor_subclass_dictà s þ z8TestDataFrameConstructors.test_constructor_subclass_dictc C st i }t j|jd |jd
df< | ¡ D ]&\}}tt}| | ¡ ¡ |||< q&t |}|j
|
|jd}t ||¡ d S )NrÕ rk rW ) r7 rb rÌ rX rV r rß rÞ Úto_dictr rï r< r= )r> rð r ÚkÚvÚdctr rA rB rB rC Ú test_constructor_defaultdictÑ s
|
z6TestDataFrameConstructors.test_constructor_defaultdictc C sL t ddddgg¡}tdgdgdgdgddddd gd
|
}t |j|¡ d S ) Nç @r° r¯ r® )rÈ r© r¨ r§ rÈ r© r¨ r§ r )r7 ra r r< Úassert_numpy_array_equalrº ©r> rA r@ rB rB rC Útest_constructor_dict_blockÞ s
|
þz5TestDataFrameConstructors.test_constructor_dict_blockc C s dddddddd}t |td }t|d
|
ks4t|d jtjksHt|d jtjks\tt |}t|d
|
kstt|d jtjkst|d jtjkstd S ) NrL rK r r rÊ Ú3)r rÊ rc r r5 r4 rk rÎ )r rµ rO rR r6 r7 r) r¥ )r> Ú test_datar rB rB rC Útest_constructor_dict_castæ s z4TestDataFrameConstructors.test_constructor_dict_castc C s^ t ttdt d¡t ttdtj d¡d}tj t
|
dd t |t d W 5 Q R X d S )NrT é r zcould not convert stringr\ r5 ) rß rÝ r r< ZmakeStringIndexr7 rÖ Úrandnrc rd r« r rµ )r> rd rB rB rC Útest_constructor_dict_cast2ó s
|
þz5TestDataFrameConstructors.test_constructor_dict_cast2c C s2 ddt jdi}t|}t|d d ts.td S )NZCol1zA String)ZRow1ÚRow2ri )r7 rb r r
rµ rR )r> rÈ r@ rB rB rC Ú!test_constructor_dict_dont_upcastü s z;TestDataFrameConstructors.test_constructor_dict_dont_upcastc C s< t ddgddggddgddgd}t|d d ts8td S )NrL rK r§ r¨ r )r r
rÔ rR )r> ÚdmrB rB rC Ú"test_constructor_dict_dont_upcast2 s "z<TestDataFrameConstructors.test_constructor_dict_dont_upcast2c C s< ddd}t |}t dd | ¡ D }tj||dd d S )N)rL rK r4 )r3 r rF rä c S s i | ]\}}|t |qS rB )rì )r r[ r\ rB rB rC r1
|
s zMTestDataFrameConstructors.test_constructor_dict_of_tuples.<locals>.<dictcomp>F©Z check_dtype)r rV r< r= ©r> r rr rA rB rB rC Ú test_constructor_dict_of_tuples s
|
z9TestDataFrameConstructors.test_constructor_dict_of_tuplesc C sF t dt ddd}t|}tdddgdddgd}t ||¡ d S ) Nr4 rF rä r rL rK r3 r ©r r r< r= rn rB rB rC Ú test_constructor_dict_of_ranges s z9TestDataFrameConstructors.test_constructor_dict_of_rangesc C sL t tdttdd}t|}tdddgdddgd}t ||¡ d S )Nr4 rä r rL rK )Úiterr Úreversedr r< r= rn rB rB rC Ú"test_constructor_dict_of_iterators s z<TestDataFrameConstructors.test_constructor_dict_of_iteratorsc C s\ dd t dD dd tt dD d}t|}tdddgdddgd}t ||¡ d S )Nc s s | ]
|
}|V qd S r rB rù rB rB rC r s zPTestDataFrameConstructors.test_constructor_dict_of_generators.<locals>.<genexpr>r4 rä r rL rK )r rs r r< r= rn rB rB rC Ú#test_constructor_dict_of_generators s *z=TestDataFrameConstructors.test_constructor_dict_of_generatorsc C s dddddddddd d
|
dd }t | ¡ }t|}td d |D t dd |D ¡dj}t |j¡|_t ||¡ ddddd|d< | dd|d f¡ tdd |D t
|
dd |D dddj}t
|
|jdd|_t|}|j |j |jd}t ||¡ d S )Nr rL rK )©rú rú ©rú Új©rx rú rF r r3 é é é ))r§ r§ r; )r¨ r© c S s g | ] }|d qS ©rL rB r rB rB rC r - s zNTestDataFrameConstructors.test_constructor_dict_multiindex.<locals>.<listcomp>c S s g | ] }|d qS ©r rB r rB rB rC r - s rW g À^@éo )r rv rw ry Úzc S s g | ] }|d qS r} rB r rB rB rC r 8 s c S s g | ] }|d qS r~ rB r rB rB rC r 8 s F)Z tupleize_colsrF ) ÚsortedrV r r rB ÚTrX r< r= Úinsertr rï r )r> rÈ Z_dr@ rA rB rB rC Ú test_constructor_dict_multiindex$ s2
|
|
|
ý ÿþ ÿz:TestDataFrameConstructors.test_constructor_dict_multiindexc s¼ ddddg fdd}|t j}|dd }|t}td d d d d
|
d d d d d
|
d d d d d
|
d d d d d
|
gdd D d}t|}t|}t|}t ||¡ t ||¡ t ||¡ d S )Nz
|
1984-02-19z
|
1988-11-06z
|
1989-12-03z
|
1990-03-15c s fddt D S )Nc s" i | ]\}}| |d | iqS ©rK rB ©r rú Ús©r rB rC r1 D s ziTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.create_data.<locals>.<dictcomp>)Ú enumerater ©Z dates_as_strr rC Ú create_dataC s zUTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.create_datac S s t | d¡S )Nz%Y-%m-%d)r Ústrptime©r rB rB rC r G r zRTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.<lambda>r ©r rL rK r4 rK r3 rF c S s g | ] }t |qS rB )r )r rV rB rB rC r Q s zTTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.<listcomp>rW )r7 rw r r r< r= ) r> r Zdata_datetime64Z data_datetimeZdata_TimestamprA Zresult_datetime64Zresult_datetimeZresult_TimestamprB r rC Ú&test_constructor_dict_datetime64_index? s&
|
ü ù
|
z@TestDataFrameConstructors.test_constructor_dict_datetime64_indexz
|
klass,namec C s t | d¡S ©Nr+ )r7 Ú timedelta64r rB rB rC r ^ r r c C s
|
t | dS )N)Údays)r r rB rB rC r _ r Z pytimedeltac C s
|
t | dS r ©r r rB rB rC r ` r z Timedelta[ns]c C s t | d d¡S )Nr+ r )r Zas_unitr rB rB rC r a r z Timedelta[s]c s ddddg} fddt |D }tdd d d dd dd d dd d dd dd d d d dgd
|
d |D d }t|}t ||¡ d S ) NrL rK r4 r3 c s" i | ]\}}| |d | iqS r
rB r ©ÚklassrB rC r1 h s zUTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.<dictcomp>r r rF c S s g | ]}t |d qS )r+ r )r ÚtdrB rB rC r q s zUTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.<listcomp>rW )r r r< r= )r> r r` Z td_as_intr rA rr rB r rC Ú'test_constructor_dict_timedelta64_index[ s ü ù
|
zATestDataFrameConstructors.test_constructor_dict_timedelta64_indexc C s¬ t jdddgdd}t jdddgdd}t||d }|d
|
j|jksJt|d j|jks^tt| t¡ ¡ | t¡ ¡ d }|d
|
j|jkst|d j|jks¨td S ) Nz2012-01rt z2012-04r} ©Úfreqz
|
2012-02-01z
|
2012-03-01r+ rä r§ r¨ )ÚpdZ PeriodIndexr r6 rR r; rm ro ræ rB rB rC Ú test_constructor_period_dictx s "z6TestDataFrameConstructors.test_constructor_period_dictc C sN |\}}t d|idgd}|d j|ks,tt dgdg|d}t ||¡ d S )Nr§ r rW rO )r r6 rR r< r= )r> Zea_scalar_and_dtypeZ ea_scalarZea_dtyper@ rA rB rB rC Ú&test_constructor_dict_extension_scalar
s
|
z@TestDataFrameConstructors.test_constructor_dict_extension_scalarz
|
data,dtypez2020-01r )ÚleftÚrightr0 r z
|
2011-01-01ú
|
US/Eastern©rJ c C sl t ddgddg|d}|d j|ks(t|d j|ks:ttj|gd |d}t ||d}t ||¡ d S ) Nr rL r§ r¨ rO rK r5 rä )r r6 rR r ra r< r= )r> r r6 r@ r? rA rB rB rC Ú&test_constructor_extension_scalar_data s z@TestDataFrameConstructors.test_constructor_extension_scalar_datac C s t jddd}ttj dd¡|d}i }|jD ](}|jD ] }| ||¡| |i ¡|< q8q.t||d}t
|
||¡ i }|jD ](}|jD ] }| ||¡| |i ¡|< qqzt||dj }t
|
||¡ d S )Nz1/1/2000r rN rÕ r rW ) r Ú period_ranger r7 rÖ rg r rX r$ Ú
|
setdefaultr< r= r )r> rñ r@ r rW Úrowrr rB rB rC Ú"test_nested_dict_frame_constructor¥ s
|
|
|
|
z<TestDataFrameConstructors.test_nested_dict_frame_constructorc C sn |dt d}t|dddgddgd}t|jdks6tt|jd ksHtt|d
|
dgddd gd}t|jd ksttt|jdkst|tjk rÎd }tj t
|
|d " t|dddgddgtj d W 5 Q R X d S t|dddgddgtj d }|j j tj ksütd}tj t|d t|dddgdgd W 5 Q R X d}tj t|d t|ddgddgd W 5 Q R X tj tdd t|ddddgdgd W 5 Q R X t|}tj|jttddd tj|jttd dd t|ddgd}tj|jttd dd t|dddgd}tj|jttddd t|d}t|jdksJtt|d}t|jdksjtd S )NrA r5 rÎ rk rÓ rL rK rF r4 )r4 r[ r\ ©r rX r6 rR rS rP rQ Tr! rW r )r r4 r )r4 r )rµ r rO rX rR r r7 r¬ rc rd r r0 rº r6 r« r< rî r r )r> rÏ Úmatr re rB rB rC Ú_check_basic_constructor¹ sL
|
& ÿ $ z2TestDataFrameConstructors._check_basic_constructorc C s8 | tj¡ tddgddgdgd}t|dks4td S )Nr¤ rÑ r rL rÎ r rK )r¨ r7 r¬ r rO rR r rB rB rC Útest_constructor_ndarrayñ s z2TestDataFrameConstructors.test_constructor_ndarrayc C s® | tj¡ tjdtd}d|d< d|d< t|ddd gd
|
d gd }d|d d
|
ksVtd|d d ksjttjdtd}t|ddd gd
|
d gd }t t ||k¡ ¡sªtd S ) NrA r5 r® r¸ r¯ ©rL rK rÎ rk rÓ rL rK rF ) r¨ r Ú
|
masked_allrµ r rR r7 r Úasarray)r> r§ r rB rB rC Ú test_constructor_maskedarray÷ s z6TestDataFrameConstructors.test_constructor_maskedarrayc
|
C s t jdtd}t|dddgddgd}t|jdks8tt|jd ksJtt t
|
||k¡ ¡sdtt|dddgddgtj d
|
}|j j tj kstt |¡}d|d < d|d < t|dddgddgd}d|d d ksØtd|d d ksìtt jdd d}t|dddgddgd}t|jdks&tt|jd ks:tt|j ¡ sNtd}tjt|dF t ¡ 2 tjdtdd t|dddgddgtjd
|
W 5 Q R X W 5 Q R X t |¡}d|d < d|d < t|dddgddgd}d|d d¡d ksútd|d d¡d kstt jdtd}t|dddgddgd}t|jdksPtt|jd ksdtt t
|
||k¡ ¡stt|dddgddgtd
|
}|j j tks¬tt |¡}d|d < d|d < t|dddgddgd}|d d dksôt|d d dks
|
td S )NrA r5 rÎ rk rÓ rL rK rF r4 r¦ r¸ rª rg z8datetime64\[ns\] values and dtype=int64 is not supportedr\ Úignorez elementwise comparison failed)ÚcategoryÚmessager^ TF)r r« rÔ r rO rX rR r r7 r r¬ r) rº r6 r´ r! rc rd rx ÚwarningsÚcatch_warningsÚfilterwarningsÚDeprecationWarningr0 r rô rm )r> r§ r Zmat2re rB rB rC Ú%test_constructor_maskedarray_nonfloat s\
|
|
ý0
|
|
z?TestDataFrameConstructors.test_constructor_maskedarray_nonfloatc C sÀ t jdtd ¡ }t|ddgddgd}ttjtjgtjtjgdddgddgtd }t ||¡ t j dtd ¡ }t|ddgddgd}td
|
d
|
gd
|
d
|
gdddgddgtd }t ||¡ d S ) N)rK rK r5 rÎ rk rL rK rF r r¦ r® )
|
r r« rµ Z harden_maskr r7 rb r< r= r¬ )r> Zmat_hardrr rA rB rB rC Ú%test_constructor_maskedarray_hardenedN s$ ü üz?TestDataFrameConstructors.test_constructor_maskedarray_hardenedc C s\ t jjt jjdddgddgd d}| tj¡}tjt dd t
|
|t d W 5 Q R X d S ) Nr )r ú<f8)Zpricer· r5 F)ÚmaskzPass \{name: data\[name\]r\ ) r7 r ra rõ r r Z mrecarrayrc rd rx r rÔ ©r> r rB rB rC Ú%test_constructor_maskedrecarray_dtyped s ÿ z?TestDataFrameConstructors.test_constructor_maskedrecarray_dtypec C s t g d}|jjdkstd S )NrW r¸ )r rº ÚshaperR ri rB rB rC Ú test_constructor_corner_shapen s
|
z7TestDataFrameConstructors.test_constructor_corner_shapez%data, index, columns, dtype, expectedrÕ r§ r¨ r) r c C s" t ||||}|jj|ks td S r )r rº r6 rR )r> r rX r r6 rA r@ rB rB rC Útest_constructor_dtyper s z0TestDataFrameConstructors.test_constructor_dtypez data,input_dtype,expected_dtypeTFÚbooleanr® r¯ ZFloat64ZInt64r© Ústringc C s( t d|i|d}|d j| ks$td S )Nr§ r5 r )r> r Z input_dtypeZexpected_dtyper@ rB rB rC Ú0test_constructor_dtype_nullable_extension_arrays s zJTestDataFrameConstructors.test_constructor_dtype_nullable_extension_arraysc C s dddddd}t |t d¡d}|d jtjks6t|d
|
jtjksJt|d jtjks^t|d jtjksrt|d jtj kstd S )NrL Tr° y @r¤ )rÔ rô rµ Úcomplexrm rÕ rW rÔ rô rµ rÁ rm )
|
r r7 r8 r6 r0 rR rö r) rå r¥ )r> r r@ rB rB rC Ú!test_constructor_scalar_inference s z;TestDataFrameConstructors.test_constructor_scalar_inferencec C sh t tj d¡dd}t |d jdgd d}t ||¡ tjt dd t ddd W 5 Q R X d S )NrÕ Trä r§ zmust pass an indexr\ F)
|
r r7 rÖ rg rº r< r= rc rd r« )r> r@ r rB rB rC Ú#test_constructor_arrays_and_scalars s
|
z=TestDataFrameConstructors.test_constructor_arrays_and_scalarsc C s8 t |}t ||¡ t |tjd}|jjtjks4td S ©Nr5 )r r< r= r7 r0 rº r6 rR )r> rð r@ Z df_castedrB rB rC Útest_constructor_DataFrame¡ s z4TestDataFrameConstructors.test_constructor_DataFramec C s* t t dd}t g dd}t ||¡ d S )Nrm r5 r¢ )r> ÚactualrA rB rB rC Ú test_constructor_empty_dataframe¨ s z:TestDataFrameConstructors.test_constructor_empty_dataframec C s t j d¡}t|dgt d¡d}|jjdks2tt j d¡}t|}|jjdksVt|jjdksfttddgt d¡d}|jjdksttddgd}|jjd ksªttt d¡d
|
}|jjd ksÊtt j d d gt
|
d dd¡}d}t j t|d t|ddgdgtd W 5 Q R X tt|j}t ||¡ tt jdtdt jdt jddt d¡d
|
}t|jdkstt|jjt jkstd S )NrÕ rÎ rF rK r rk )rÕ rK r )r rK rW )rÕ r r¤ rÑ r5 rL z(could not convert string to float: 'foo'r\ ©rX r r6 r )r7 rÖ rg r r8 rº ÚndimrR r» ra rm r9 rc rd r« rµ r r< r= r¬ rÔ r) rO r r6 )r> rð r? rk r§ re rB rB rC Útest_constructor_more® s2 þz/TestDataFrameConstructors.test_constructor_morec C sx t g g d}t g d}t ||¡ t g ddgd}t i ddgd}t ||¡ dd }t | ddgd}t ||¡ d S )NrW rÎ rk r c s s dE d H d S r rB rB rB rB rC Ú empty_genà s zHTestDataFrameConstructors.test_constructor_empty_list.<locals>.empty_genr¢ )r> r@ rA rË rB rB rC Útest_constructor_empty_listÕ s
|
z5TestDataFrameConstructors.test_constructor_empty_listc C s| t ddgddggddgd}t|d s,t|d jtjks@tt dt d ¡i}d
|
d td D }t |}t ||¡ d S ) NrL r§ rK r¨ Únumr: r= r rÕ c S s g | ]}t |¡qS rB ©r7 ra r rB rB rC r ï s zLTestDataFrameConstructors.test_constructor_list_of_lists.<locals>.<listcomp>)
|
r r rR r6 r7 r¥ r8 r r< r= )r> r@ rA r rr rB rB rC Ú test_constructor_list_of_listsæ s z8TestDataFrameConstructors.test_constructor_list_of_listsc C sx t ddg}tjd d gtd}||d< |d |d< t|}tt |¡}t ||¡ |jdksbt t
|
|d j |¡ d S )NrL rK r5 r )rK rL ) r r7 ra rm r r r< r= r» rR r` rº )r> Úserr? r@ rA rB rB rC Ú.test_nested_pandasarray_matches_nested_ndarrayó s zHTestDataFrameConstructors.test_nested_pandasarray_matches_nested_ndarrayc C sj t dt dg}tddddgdddd gg|d
|
}t |¡}tddddgdddd gg|d
|
}t ||¡ d S ) NÚabcdZcdefrL rK r4 r3 r rF rz r )rì r r Z from_arraysr< r= )r> r rr rD rA rB rB rC Ú2test_constructor_list_like_data_nested_list_column s
|
|
zLTestDataFrameConstructors.test_constructor_list_like_data_nested_list_columnc C sR t dt dg}d}tjt|d& tddddgdd d
|
d gg|d W 5 Q R X d S ) Nr Úcdez#3 columns passed, passed data had 4r\ rL rK r4 r3 r rF rz r ©rì rc rd r« r ©r> r re rB rB rC Ú0test_constructor_wrong_length_nested_list_column s zJTestDataFrameConstructors.test_constructor_wrong_length_nested_list_columnc C sR t dt dg}d}tjt|d& tddddgdd d
|
d gg|d W 5 Q R X d S ) NrÒ rÔ z all arrays must be same lengthr\ rL rK r4 r3 r rF rz r rÕ rÖ rB rB rC Ú2test_constructor_unequal_length_nested_list_column s zLTestDataFrameConstructors.test_constructor_unequal_length_nested_list_columnr ú
|
2021-01-01r c C s< t |dddgdgd}t dtdgd i}t ||¡ d S )Nr rL rK r r rÙ r4 )r r r< r= rn rB rB rC Ú&test_constructor_one_element_data_list s z@TestDataFrameConstructors.test_constructor_one_element_data_listc C sj G dd dt j}|ddg|ddgg}ddg}t||d }tddgddgg|d }tj||d
|
d d S ) Nc @ s0 e Zd ZddddZdd ZedddZdS ) zPTestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainerN©Úreturnc S s
|
|| _ d S r )Ú_lst)r> ÚlstrB rB rC Ú__init__0 s zYTestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainer.__init__c S s | j |¡S r )rÝ Ú __getitem__)r> ÚnrB rB rC rà 3 s z\TestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainer.__getitem__c S s
|
| j ¡ S r )rÝ Ú__len__)r> rB rB rC râ 6 s zXTestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainer.__len__)Ú__name__Ú
|
__module__Ú __qualname__rß rà rÔ râ rB rB rB rC ÚDummyContainer/ s ræ rL r§ rK r¨ rÍ r: r Frm )r ÚSequencer r< r= )r> ræ Zlst_containersr rr rA rB rB rC Ú test_constructor_sequence_like+ s
|
z8TestDataFrameConstructors.test_constructor_sequence_likec C s t dt dtd¡i}t dttdi}tj||dd t ttdttdg}t t dtd¡t dtd¡g}tj||dd d S )NrÎ rú rÕ Frm )r ra r rì r< r= r? rB rB rC Ú test_constructor_stdlib_array? s $z7TestDataFrameConstructors.test_constructor_stdlib_arrayc C s, t td}t ttd}t ||¡ d S ©NrÕ ©r r rì r< r= r? rB rB rC Útest_constructor_rangeJ s z0TestDataFrameConstructors.test_constructor_rangec C s@ t tdtdg}t ttdttdg}t ||¡ d S rê rë r? rB rB rC Ú test_constructor_list_of_rangesP s z9TestDataFrameConstructors.test_constructor_list_of_rangesc C s< G dd d}t dddggd }t | }t ||¡ d S )Nc @ s e Zd ZedddZdS )zATestDataFrameConstructors.test_constructor_iterable.<locals>.IterrÛ c s s t dD ]}dddgV qd S )NrÕ rL rK r4 )r )r> rú rB rB rC Ú__iter__X s zJTestDataFrameConstructors.test_constructor_iterable.<locals>.Iter.__iter__N)rã rä rå r
|
rî rB rB rB rC ÚIterW s rï rL rK r4 rÕ r¢ )r> rï rA rr rB rB rC Útest_constructor_iterableU s
|
z3TestDataFrameConstructors.test_constructor_iterablec C s0 t ttd}t ttd}t ||¡ d S rê ©r rr r rì r< r= r? rB rB rC Útest_constructor_iterator` s z3TestDataFrameConstructors.test_constructor_iteratorc C sH t ttdttdg}t ttdttdg}t ||¡ d S rê rñ r? rB rB rC Ú"test_constructor_list_of_iteratorse s z<TestDataFrameConstructors.test_constructor_list_of_iteratorsc C s dd t dD }dd t dD }ttt dtt dg}t||g}t ||¡ dd t dD }t|}tt ddd}tj||dd d S )
|
Nc s s | ]
|
}|V qd S r rB rù rB rB rC r m s zGTestDataFrameConstructors.test_constructor_generator.<locals>.<genexpr>rÕ c s s | ]
|
}|V qd S r rB rù rB rB rC r n s c s s | ]}|d gV qdS )r§ NrB rù rB rB rC r t s r§ rM Frm )r r rì r< r= )r> Zgen1Zgen2rA rr ÚgenrB rB rC Útest_constructor_generatorj s z4TestDataFrameConstructors.test_constructor_generatorc C s* t i g}t tdg d}t ||¡ d S )NrL r )r r r< r= r? rB rB rC Ú test_constructor_list_of_dictsy s
|
z8TestDataFrameConstructors.test_constructor_list_of_dictsc C sf t ddg}t ddg}t d|fd|fg}t|}ddgdd gd
|
}t|d d gd }t ||¡ d S )N)r¨ rL )r§ rK )r¨ rK )r§ r r² r± rL rK r )r² r± r¨ r§ )r rX )r r r< r= )r> Znested1Znested2r rr rA rB rB rC Ú3test_constructor_ordered_dict_nested_preserve_order~ s zMTestDataFrameConstructors.test_constructor_ordered_dict_nested_preserve_orderÚ dict_typec C st t ddggddgd}| }dg|d< dg|d< t |}t ||¡ | }d|d< d|d< t |g}t ||¡ d S )NrK rL r¨ r§ r r¢ )r> rø rA r rr rB rB rC Ú,test_constructor_ordered_dict_preserve_order s
|
|
|
zFTestDataFrameConstructors.test_constructor_ordered_dict_preserve_orderc C sª | }d|d< d|d< | }d|d< d|d< ddd}t ddgddggddgd}t ||g}t ||¡ t ddgddgddggddgd}t |||g}t ||¡ d S )NrK r¨ rL r§ r; r r¢ )r> rø Zrow_oneZrow_twoZ row_threerA rr rB rB rC Ú0test_constructor_ordered_dict_conflicting_orders s
|
"zJTestDataFrameConstructors.test_constructor_ordered_dict_conflicting_ordersc C s^ dd t dD }t|}tdddgdddgdddgddd d
|
gd d d gd}t ||¡ d S )Nc S s$ g | ] }t |d ddgt|dqS )r¨ r§ r© r )r r: rù rB rB rC r ² s z[TestDataFrameConstructors.test_constructor_list_of_series_aligned_index.<locals>.<listcomp>r4 r rL rK )r¨ r§ r© r¨ r§ r© Ú0r rÊ rF rp )r> Úseriesrr rA rB rB rC Ú-test_constructor_list_of_series_aligned_index± s ýzGTestDataFrameConstructors.test_constructor_list_of_series_aligned_indexc C sJ G dd dt }ddd}||g}|g}t|}t|}t ||¡ d S )Nc @ s e Zd ZdS )zTTestDataFrameConstructors.test_constructor_list_of_derived_dicts.<locals>.CustomDictN©rã rä rå rB rB rB rC Ú
|
CustomDict¼ s rÿ ç ø?r4 rä )rß r r< r= )r> rÿ rÈ Z data_customr Z result_customrr rB rB rC Ú&test_constructor_list_of_derived_dicts» s
|
|
z@TestDataFrameConstructors.test_constructor_list_of_derived_dictsc C s@ t j d¡t j d¡d}tjtdd t| W 5 Q R X d S )NrÕ r{ r z%All arrays must be of the same lengthr\ )r7 rÖ rg rc rd r« r r¹ rB rB rC Útest_constructor_raggedÈ s z1TestDataFrameConstructors.test_constructor_raggedc C sF t td}tddi|d}tddddgi|d}tj||dd d S )Nr4 r§ r rW Frm )r r r r< r= )r> rq r@ rA rB rB rC Útest_constructor_scalarÍ s z1TestDataFrameConstructors.test_constructor_scalarc C s" t |d |jdgd}| ¡ d S )NrÎ r )r rX r´ )r> rð r@ rB rB rC Ú test_constructor_Series_copy_bugÓ s z:TestDataFrameConstructors.test_constructor_Series_copy_bugc C sÊ i }dddd|d< t ddddgddd d
|
gd |d < t|}|jjsHttjtd d tddgdddd W 5 Q R X tddgt ddgddgd d}tddgddgdddgd }t ||¡ d S )NrL rK r4 )r¤ rÑ rL rÎ r3 rÑ ZquxrL r¤ rW rk zambiguous orderingr\ r§ r¨ rä r )
|
r r rX Zis_monotonic_increasingrR rc rd r« r< r= rn rB rB rC Ú&test_constructor_mixed_dict_and_Series× s "" z@TestDataFrameConstructors.test_constructor_mixed_dict_and_Seriesc C s8 ddgdg}t |}t ddgddgg}t ||¡ d S )NrL rK ©r4 r3 r4 r3 r¢ rn rB rB rC Ú test_constructor_mixed_type_rowsè s z:TestDataFrameConstructors.test_constructor_mixed_type_rowsz tuples,lists)rB rB r4 r3 rF c C s t |}t |}t ||¡ d S r r¢ )r> rC Úlistsrr rA rB rB rC Útest_constructor_tupleï s z0TestDataFrameConstructors.test_constructor_tuplec C s4 t dddgi}t dtddgi}t ||¡ d S )NrÎ rª r ©r r r< r= r? rB rB rC Ú test_constructor_list_of_tuples s z9TestDataFrameConstructors.test_constructor_list_of_tuplesc C s t dtd}|dd|ddg}tddgddgd}t|}t ||¡ tddgddgd}t|d d
|
gd }t ||¡ d S ) NZPandasÚabrL r4 rK r3 rä )r r r r r )r rì r r< r= )r> Z named_tuplerC rA rr rB rB rC Ú$test_constructor_list_of_namedtuples s z>TestDataFrameConstructors.test_constructor_list_of_namedtuplesc C sX t ddtfdtfg}|dd|ddg}tddgddgd}t|}t ||¡ d S )NÚPointr r r r4 rL ©r r )r rÔ r r< r= )r> r r rA rr rB rB rC Ú$test_constructor_list_of_dataclasses s
|
z>TestDataFrameConstructors.test_constructor_list_of_dataclassesc C s t ddtfdtfg}t ddtfdtfdtfg}|dd|d ddg}tdtjgddgtjd gtjdgd
|
}t|}t ||¡ d S ) Nr r r ÚHLineÚx0Úx1r r4 rL )r r r r )r rÔ r r7 rb r< r= )r> r r r rA rr rB rB rC Ú7test_constructor_list_of_dataclasses_with_varying_types s "ÿzQTestDataFrameConstructors.test_constructor_list_of_dataclasses_with_varying_typesc C sV t ddtfdtfg}d}tjtt |¡d t|dddddg W 5 Q R X d S ) Nr r r z0asdict() should be called on dataclass instancesr\ r rL r )r rÔ rc rd rx ÚreÚescaper )r> r re rB rB rC Ú1test_constructor_list_of_dataclasses_error_thrown* s zKTestDataFrameConstructors.test_constructor_list_of_dataclasses_error_thrownc C s~ ddddddddd d
|
d d d ddddg}t ddd gddd gdd dgddd gd d dgd d dgd}t |}t ||¡ d S )NrL r3 rz rÕ )ÚFirstÚSecondÚThirdÚFourthr rK r6 r{ )r r r r rF r4 r2 r| é r9 )r r r r ÚYYYÚXXX)r r r r r r r¢ )r> r rA rr rB rB rC Ú#test_constructor_list_of_dict_order3 s ýúÿ
|
z=TestDataFrameConstructors.test_constructor_list_of_dict_orderc C sv t dddgdddgdd}t|}|jd dks4tt |j|j¡ tj d
|
¡}t |dd }t|}td|i}t
|
||¡ t |t dd d }t|}td |i}t
|
||¡ d}t j t|d t|ddgd W 5 Q R X t g dtd}t|}|jd dkstt |dd }t||gj}t||dddgd}t
|
||¡ t||gj}t||dd dgd}t
|
||¡ d S )NrL rK r4 r§ r¨ r© r r r rÕ rl r9 rW z<Shape of passed values is \(10, 1\), indices imply \(10, 2\)r\ r ©r` r6 )r ú Unnamed 0r! rË )r r r rR r< rî rX r7 rÖ rg r= r rc rd r« rm r )r> r§ r@ r? r rA re Ús1rB rB rC Ú test_constructor_Series_namedG s4 z7TestDataFrameConstructors.test_constructor_Series_namedc C s t tddd}t tddd}t t|dgd| ¡ ¡ t t|dgd| ¡ ¡ t|dgdjshtt|dgdjs|td S )Nr r rl rL r )r r r< r= r Úto_framerÏ rR )r> Ús0r" rB rB rC Ú)test_constructor_Series_named_and_columnsm s zCTestDataFrameConstructors.test_constructor_Series_named_and_columnsc C sÀ t dddgdddgdd}t dddgdddgd }tddg}t||d }t| |¡}|jd
|
dksjtt ||¡ t||d }t| |¡}|jd
|
d
|
ks¢tt |j |¡ t ||¡ d S ) NrL rK r4 r§ r¨ r© r r rW r )
|
r r r rï r rR r< r= rî rX )r> r" Ús2Z other_indexÚdf1Zexp1rÁ Zexp2rB rB rC Ú+test_constructor_Series_differently_indexed{ s zETestDataFrameConstructors.test_constructor_Series_differently_indexedz#name_in1,name_in2,name_in3,name_out)rq rq rq rq )rq rq NN)rq NNN)Úidx1Úidx2NN)r* r* r+ N)r* r+ Zidx3N)NNNNc
|
C sº t dddg|dt dddg|dt dddg|dg}dd t|d d
|
d gD }t|}t dddddg|d}td d dtjtjgtjd d dtjgtjtjd d dgd|d} t || ¡ d S )Nr§ r¨ r© rl rÈ Úec S s$ i | ] \}}|t d ddg|dqS )r rL rK rW r~ )r rú r© rB rB rC r1 ¢ s zJTestDataFrameConstructors.test_constructor_index_names.<locals>.<dictcomp>r r r r rL rK )r r r rW )r rÝ r r7 rb r< r= )
|
r> Zname_in1Zname_in2Zname_in3Zname_outÚindicesrü rr Zexp_indrA rB rB rC Ú test_constructor_index_names s" ýÿýú z6TestDataFrameConstructors.test_constructor_index_namesc C s\ t |jd d
}t |jd d
}t|j||d}t |jt|¡ t |jt|¡ d S )Nr r4 r )rì rX r r rP r< rî r )r> rð rX r rr rB rB rC Ú test_constructor_manager_resize³ s
|
z9TestDataFrameConstructors.test_constructor_manager_resizec C s t |d t|d dddgd}t ||jd d
ddgf ¡ d}tjt|d( t |d t|d d d
d W 5 Q R X d S )NrÎ rk r r zdoes not match index lengthr\ éþÿÿÿ)r rì r< r= rÌ rc rd r« )r> rð r@ re rB rB rC Ú%test_constructor_mix_series_nonseries» s ÿ z?TestDataFrameConstructors.test_constructor_mix_series_nonseriesc C s> d}t jt|d" ttjdgddggtjd W 5 Q R X d S )Nr[ r\ rL r r5 )rc rd r r r7 rb r0 r( rB rB rC Ú%test_constructor_miscast_na_int_dtypeÅ s z?TestDataFrameConstructors.test_constructor_miscast_na_int_dtypec C s^ t ddggddgd}t ddgg}ddg|_t ||¡ t jdgddgd}t ||¡ d S )Nr{ r r§ r )r{ r )r r r< r= Z from_records)r> r@ ZedfZidfrB rB rC Ú"test_constructor_column_duplicatesË s
|
z<TestDataFrameConstructors.test_constructor_column_duplicatesc C s¦ t ddgddgtd}t ddgddgtd}t ||¡ t ddgddgtjd}t ||¡ t ddgddgtjd}t ||¡ t ddgddgdd}t ||¡ d S )Nr rL rÈ ZU5)r rm r: r< r= r7 Zstr_Zunicode_ra rB rB rC Ú(test_constructor_empty_with_string_dtype× s zBTestDataFrameConstructors.test_constructor_empty_with_string_dtypec C s, t dg|d}t dg|d}t ||¡ d S )NZc1©r r6 r¢ )r> Znullable_string_dtyperA r@ rB rB rC Ú,test_constructor_empty_with_string_extensionä s zFTestDataFrameConstructors.test_constructor_empty_with_string_extensionc C sX t ddddgdddgd}t |t t |j¡ d ¡|j|j¡ t d
|
dddgdddgd}t |t t |j¡ d ¡|j|j¡ t dddgddgd}t |t tj ddgddggt
|
d ddgddgd¡ d }t j t |d t dddg W 5 Q R X t j t |d t dddgd W 5 Q R X d}t j t|d t dddgddgt W 5 Q R X d S )Nç rL rK r4 r§ r¨ r© r r) r r0 r5 z*DataFrame constructor not properly called!r\ r zincompatible data and dtype)r r< r= r7 rõ r» r; rX r ra rm rc rd r« rx rµ )r> r@ re rB rB rC Ú test_constructor_single_valueê s6 ÿ ÿýþ z7TestDataFrameConstructors.test_constructor_single_valuec C sò t t j¡j}t t j¡j}t d¡j}t t j¡j}tdddtdtdddddd t d
|
¡d }|j
|
}t t d ¡gt |¡gd t |¡gd t d d }t ||¡ tdddddd|t jd|d|t jd|dit d
|
¡d }|j
|
}t t d¡gt d ¡g t d¡g t d¡g t |¡g ddd||gd }t ||¡ tdddddd|t jdgd
|
|d|t jdgd
|
|dit d
|
¡d }|j
|
}t t d¡gt d ¡g t d¡g t d¡g t |¡g ddd||gd }t ||¡ d S )Nrg rL r¤ rÑ Z20010101éÑ rK r )rÎ rk rÓ r+ ÚErÕ rW r0 ZABCDEr§ r® r¨ r© r5 r) rm )r7 r6 Úint_r` Zfloat_r¥ r r r r8 rà r rì r< rp ra )r> ZintnameZ floatnameZdatetime64nameZ
|
objectnamer@ rr rA rB rB rC Ú test_constructor_with_datetimes
|
s ûø
|
|
ÿþü ûø
|
|
|
ÿ
|
þ
|
ý
|
ü ú ûø
|
|
|
ÿ
|
þ
|
ý
|
ü úz9TestDataFrameConstructors.test_constructor_with_datetimesc C s6 t dddd}dd |D }t|}|jdks2td S )NrM r+ rÕ ©Ústartr rI c S s g | ] }| ¡ qS rB ©Ú to_pydatetime©r ÚtsrB rB rC r T s zNTestDataFrameConstructors.test_constructor_with_datetimes1.<locals>.<listcomp>rg )r r r6 rR )r> ÚindÚ datetimesZ
|
datetime_srB rB rC Ú test_constructor_with_datetimes1Q s z:TestDataFrameConstructors.test_constructor_with_datetimes1c C sv t dddd}dd |D }dd |D }t|dgd }||d
|
< |j}tt d ¡t d ¡gdd
|
gd }t ||¡ d S )NrM r+ rÕ r= c S s g | ] }| ¡ qS rB r? rA rB rB rC r [ s zNTestDataFrameConstructors.test_constructor_with_datetimes2.<locals>.<listcomp>c S s g | ] }| ¡ qS rB )r rA rB rB rC r \ s rD r Údatesúdatetime64[ns]rm rW )r r rà r r7 r6 r< rp )r> rC rD rF r@ rr rA rB rB rC Ú test_constructor_with_datetimes2X s þz:TestDataFrameConstructors.test_constructor_with_datetimes2c C s t d¡}| tddd¡}td|idgd}|jd |ks@tt |j t
|
ddi¡ td|ig}|jd |ksvtt |j t
|
ddi¡ d S ) Nr éÜ rL zEnd Dater rW r¸ údatetime64[ns, US/Eastern]) ÚpytzÚtimezoneZlocalizer r ÚiatrR r< rp rà r )r> rJ rV r@ rB rB rC Ú test_constructor_with_datetimes3f s
|
|
ÿ
|
ÿz:TestDataFrameConstructors.test_constructor_with_datetimes3c C s t ddd}td|i}|jd jd ks,tt dddd}td|i}t|jd jdks^tt dddd}td|i}t|jd jdkstd S ) NÚ20130101r4 rN rY r¸ ÚUTCrH r )r r rM rJ rR r: )r> Údrr@ rB rB rC Ú test_constructor_with_datetimes4x s z:TestDataFrameConstructors.test_constructor_with_datetimes4c C s^ t ddddd}td| ¡ jddi}t }||d< t ||¡ td|i}t ||¡ d S ) Nú1/1/2011r Ú10sr ©rI r rJ r§ T©Zdrop©r r Z to_seriesZ reset_indexr< r= )r> rú rA r@ rB rB rC Ú test_constructor_with_datetimes5
s z:TestDataFrameConstructors.test_constructor_with_datetimes5c C sV t ddddd}t dddd}t||d}t| ¡ jdd |d}t ||¡ d S )
|
NrS r rT r rU ©rI r rä TrV rW )r> rú Zi_no_tzr@ rA rB rB rC Ú test_constructor_with_datetimes6 s
|
z:TestDataFrameConstructors.test_constructor_with_datetimes6r? rt c C s* t |j}tt d¡g}t ||¡ d S )NrG )r rà r r7 r6 r< rp )r> r? rr rA rB rB rC Ú%test_constructor_datetimes_with_nulls s
|
z?TestDataFrameConstructors.test_constructor_datetimes_with_nullsÚorderÚKrÎ rÓ ÚFÚunitÚhr ÚmsÚusru c C s d| d}t jdddgdddgg||d }t|}t| d
|
¡}|d krztjtd d | |¡ W 5 Q R X | d¡}n |j|d}t ||¡ d S )Nz datetime64[ú]z
|
2015-01-01z
|
2015-01-02z
|
2015-01-03z
|
2017-01-01z
|
2017-01-02z
|
2017-02-03©r6 r\ rg )r} r+ r` r| ú Cannot castr\ z datetime64[s]r5 ) r7 ra r r; rc rd rx r< r= )r> r\ r_ r6 Únar@ rA rB rB rC Ú!test_constructor_datetimes_non_ns s þú z;TestDataFrameConstructors.test_constructor_datetimes_non_nsc C s´ d| d}t jt dd¡t dd¡gt dd¡t dd¡gg||d}t|}|d kr^d
|
}n|}t d | d¡}ttddtddgtddtddgg|d }t ||¡ d S ) Nz timedelta64[rc rL r+ rK r3 r rd )r+ r` r| r zm8[r5 )r7 ra r r r6 r r< r= ) r> r\ r_ r6 rf r@ Zexp_unitÚ exp_dtyperA rB rB rC Ú!test_constructor_timedelta_non_nsÉ s( þúþû z;TestDataFrameConstructors.test_constructor_timedelta_non_nsc C s2 t dd tdD }|j}tt d¡gd }t ||¡ t dd tdD }|j}tt d¡gd }t ||¡ t ddd gi}|jjd
|
t d ¡ks¢t t d d g}|jjd
|
t d ¡ksÈt t ddg}|jjd
|
t d¡ksît t dd d gi}|jjd
|
t d ¡kst t dddgi}|jjd
|
t d¡ksFt t dd itdd}|jjd
|
t d ¡ksvt t dditdd}|jjd
|
t d¡ks¦t t d d ddgddddgt
|
ddd tdD dd ddgd }|j}tt d ¡t d¡t d ¡t d ¡t d¡gt
|
d d}t ||¡ d S ) Nc S s g | ]}t d ¡qS )r )r7 r8 r rB rB rC r ô s zSTestDataFrameConstructors.test_constructor_for_list_with_dtypes.<locals>.<listcomp>r rÔ c S s g | ]}t jt d ¡ddqS )r r/ r5 )r7 ra r8 r rB rB rC r ù s r/ r§ l l r r0 rL rK r® r¯ r) r4 rW r3 rz g333333ó?gffffff@gffffff@g333333@rÒ c S s g | ]}t d ddqS )éÐ rL )r rù rB rB rC r s r_ )r§ r¨ r© rÈ r, rm rG Zabcde) r r rà r r7 r6 r< rp r¹ rR rì )r> r@ rr rA rB rB rC Ú%test_constructor_for_list_with_dtypesò sR
|
|
|
ûÿ ûø
|
z?TestDataFrameConstructors.test_constructor_for_list_with_dtypesc C s@ t |dd}d|d< |d dk ¡ s(t|d dk ¡ r<td S )NTrà r rÎ )r r rR )r> rð ÚcoprB rB rC Útest_constructor_frame_copy- s z5TestDataFrameConstructors.test_constructor_frame_copyc C s@ | ¡ }t|}|j|jk s tt t|¡|_t ||¡ d S r )
|
r´ r rP rR r7 r8 rO rX r< r= )r> rð Úorigrl rB rB rC Ú#test_constructor_frame_shallow_copy3 s
|
z=TestDataFrameConstructors.test_constructor_frame_shallow_copyc C sú |s||j ¡ }t|}d|d< |r:|j d dk ¡ rPtn|j d dk ¡ sPtt|dd}d|d< |j d dk ¡ rötnz|j ¡ }t|}|jjd jjs¢td|d< |j d dks¼tt|dd}|jjd jjrÜtd |d< |j d d ksötd S )
|
Nr Trà rF r r r¸ Fiè )
|
rº r´ r r rR rP r r¿ rÀ r¹ )r> rð rS r» r? r@ rB rB rC Ú test_constructor_ndarray_copy> s&
|
|
z7TestDataFrameConstructors.test_constructor_ndarray_copyc C sP |j }td|d idd}d|j|jd |jd
df< |d dk ¡ rLtd S )NrÎ Trà r r rë )r r rÌ rX r rR )r> rð rü r@ rB rB rC Ú test_constructor_series_copy\ s z6TestDataFrameConstructors.test_constructor_series_copyr@ rW gñ?g@r rz r7 r° c C sæ t t|jD ]}|jd d
|f qt t|j¡t|j }t|dkr|tjt dd |j
|
d d
tj f W 5 Q R X nft|dkr¸t |jd d
|d f |j
|
d d
tj f ¡ n*t |jd d
|f |j
|
d d
tj f ¡ d S )Nr z^nan$r\ rL )r rO r r¹ r7 r8 r! rc rd ÚKeyErrorrÌ rb r< rp r= )r> r@ rú ZindexerrB rB rC Útest_constructor_with_nasf s 0z3TestDataFrameConstructors.test_constructor_with_nasc C s: t dtjdgi}|d jtjks&t|d d r6td S )Nr§ FrL )r r7 rb r6 r¥ rR )r> rÈ rB rB rC Ú&test_constructor_lists_to_object_dtype s z@TestDataFrameConstructors.test_constructor_lists_to_object_dtypec C s\ t dddg}t |¡ dd¡}t |d¡}t||jd}t||||d}t ||¡ d S ) NrÎ rk rÓ rë rL r r5 r ) r r7 ra r9 Z broadcast_tor r6 r< r= )r> Úcatr? rr rA rB rB rC Ú*test_constructor_ndarray_categorical_dtype s zDTestDataFrameConstructors.test_constructor_ndarray_categorical_dtypec C sÔ t dtdidd}ttdddd}t |d |¡ ttddd}| ¡ }ttdddd}t |d |¡ |jdd}ttdddd}t |d |¡ t tddd}ttdddd}t |d |¡ d S ) NrÎ r r¯ r5 r_ r r¤ rl )r rì r r< rp r$ )r> r@ rA r rr rB rB rC Ú test_constructor_categorical s z6TestDataFrameConstructors.test_constructor_categoricalc C s6 t td}t|g}t| t¡g}t ||¡ d S )Nr )r rì r r; rm r< r= )r> ru r@ rA rB rB rC Ú-test_construct_from_1item_list_of_categorical s
|
zGTestDataFrameConstructors.test_construct_from_1item_list_of_categoricalc C sD t ttdttdg}t dddgdddgg}t ||¡ d S )Nr Zabdr§ r¨ r© rÈ ©r r rì r< r= rI rB rB rC Ú(test_construct_from_list_of_categoricals¶ s zBTestDataFrameConstructors.test_construct_from_list_of_categoricalsc C s@ t ttdtdg}t dddgdddgg}t ||¡ d S ) Nr Údefr§ r¨ r© rÈ r, Úfry rI rB rB rC Ú%test_from_nested_listlike_mixed_types¾ s z?TestDataFrameConstructors.test_from_nested_listlike_mixed_typesc C s@ t ttdttdg}t tdtdg}t ||¡ d S )Nr Zabdefgry rI rB rB rC Ú0test_construct_from_listlikes_mismatched_lengthsÆ s zJTestDataFrameConstructors.test_construct_from_listlikes_mismatched_lengthsc C sÒ ddddg}t | d¡}t |dd}t ||¡ ddddg}t | d¡}t |dd}t ||¡ td dd
|
}t ttjtjtjgdddgd }||_t d |i}t d t dddgddi|d }t
|
||¡ d S )NrL rK r4 r¯ r5 r§ r¨ r© Z20000101rN )rº Ú
|
categoriesr rW ) r r; r< rp r r r7 rb rX r r= )r> rV r ÚresrX rA r@ rB rB rC Ú#test_constructor_categorical_seriesË s ÿ z=TestDataFrameConstructors.test_constructor_categorical_seriesr6 c C s, t dddgidg|d}|jj|ks(td S )Nr§ rL rK r¨ r5 )r r¨ r6 rR )r> r6 r rB rB rC Ú%test_check_dtype_empty_numeric_columnâ s
|
z?TestDataFrameConstructors.test_check_dtype_empty_numeric_columnc C sF t dddgidg|d}|r0|tjkr0t |¡ |jjjdksBtd S )Nr§ rL rK r¨ r5 rm ) r r< Ú BYTES_DTYPESr Ú&mark_array_manager_not_yet_implementedr¨ r6 r` rR )r> Úrequestr6 rS r rB rB rC Ú$test_check_dtype_empty_string_columnð s
|
z>TestDataFrameConstructors.test_check_dtype_empty_string_columnc C sL t dtd ¡ j}t dti}t ||¡ tt dtdj}t ||¡ d S )Nr r )r rm r$ rà r< rp r r? rB rB rC Ú test_to_frame_with_falsey_namesþ s
|
z9TestDataFrameConstructors.test_to_frame_with_falsey_namesr* r¯ c C sb t ddddddgi|pdd}t td dg|d
|
}t ||¡ t dtd i|d}t ||¡ d S ) NrÎ r rL rK r4 r3 r0 r5 r r5 )r r r< r= )r> r6 rA rr rB rB rC Ú test_constructor_range_dtype s
|
z6TestDataFrameConstructors.test_constructor_range_dtypec C s\ G dd dt }tdddgdddgg}t||dddg|dddgg}t ||¡ d S ) Nc @ s e Zd ZdS )zETestDataFrameConstructors.test_frame_from_list_subclass.<locals>.ListNrþ rB rB rB rC ÚList s r rL rK r4 r3 r rF )rì r r< r= )r> r rA rr rB rB rC Ú test_frame_from_list_subclass s $z7TestDataFrameConstructors.test_frame_from_list_subclassÚ extension_arrZaabbcz1/1/2017z1/1/2018)r> Úendr c C s$ t t|}t |}t ||¡ d S r r
|
)r> r rA rr rB rB rC Ú%test_constructor_with_extension_array s z?TestDataFrameConstructors.test_constructor_with_extension_arrayc C s^ t ¡ }||f}t|ttdtddi|gd}tdddgtt|gd}t ||¡ d S )Nr4 rW r r rL rK )r Útodayr r r r r< r= )r> r\ Útuprr rA rB rB rC Ú*test_datetime_date_tuple_columns_from_dict, s
|
" zDTestDataFrameConstructors.test_datetime_date_tuple_columns_from_dictc C s t dddgtdddgd}t dd dd
|
d gtd dd ddgd}t||g}tt dddtjtjgdddddgg¡dddd d gd}t ||¡ d S )Né' rF r3 ZfemaleZmaleÚunknownrW rK é éò é r| r| g c@g @n@g Àb@r¯ r )r r r r7 ra rb r< r= )r> r" r' rr rA rB rB rC Ú/test_construct_with_two_categoricalindex_series4 s þ $ þzITestDataFrameConstructors.test_construct_with_two_categoricalindex_seriesc C s t tdd}t|d ¡ d¡}t|d ¡ d¡}t||d}ttdddd tdd
|
dd td
|
d dd td d dd td d dd td ddd tdddd tdddd tdddd tdddd g
|
dd}tdgd t j
|
gd t j
|
gd dgd d|d}t ||¡ d S )Nr r rÕ r r gòÒMbX¹¿gÍÌÌÌÌÌ#@r )ÚclosedgÍÌÌÌÌÌ3@g33333³=@gÍÌÌÌÌÌC@g ÀH@g33333³M@g33333SQ@gÍÌÌÌÌÌS@gfffffFV@r· T)ZorderedrW ) r r r Z value_countsÚheadÚtailr r r r7 rb r< r= )r> rÐ Úser1Úser2rr rX rA rB rB rC Ú1test_constructor_series_nonexact_categoricalindexB s. ö ó, ÿzKTestDataFrameConstructors.test_constructor_series_nonexact_categoricalindexc C s® t dddddt dddddfg}tj|ddgd }t|}|d
|
d |d d ksXt|d d |d d ksttt|d
|
}t|d tst|d |d d ksªtd S ) NrI r| r r{ rf rÕ )ÚDateúM8[us])Ú Forecastingr r5 r r rL )r r7 ra r rR r r
r )r> rF r? r@ r rB rB rC Útest_from_M8_structured\ s " z1TestDataFrameConstructors.test_from_M8_structuredc C s> G dd dt }td|dddddgi}|j jdks:td S )Nc @ s e Zd ZdS )zOTestDataFrameConstructors.test_from_datetime_subclass.<locals>.DatetimeSubclassNrþ rB rB rB rC ÚDatetimeSubclassj s r¡ r iä rL rG )r r r6 rR )r> r¡ r rB rB rC Útest_from_datetime_subclassh s z5TestDataFrameConstructors.test_from_datetime_subclassc C s@ t dddd}d}tjt|d t|tdd W 5 Q R X d S ) NrE r4 rG rH z@Shape of passed values|Passed arrays should have the same lengthr\ r3 rW )r rc rd r« r r )r> rT re rB rB rC Ú(test_with_mismatched_index_length_raisesp s zBTestDataFrameConstructors.test_with_mismatched_index_length_raisesc C sR t dddd}t |¡}ttj t|¡|d}t |d jt d¡¡sNt d S )Nz1/1/2000 00:00:00z1/1/2000 1:59:50rT r r rk rg )
|
r r7 r¬ r rÖ rg rO Z
|
issubdtyper6 rR )r> rñ rF r@ rB rB rC Ú!test_frame_ctor_datetime64_columnw s
|
z;TestDataFrameConstructors.test_frame_ctor_datetime64_columnc C s ddddgddddgg}t tj dd¡dd |D d}t|jtsHtt|jtrXtt tj dd¡|d }t|jts~td S )
|
Nr§ r¨ r r r3 c S s g | ]}t |¡qS rB rÎ r rB rB rC r s zYTestDataFrameConstructors.test_dataframe_constructor_infer_multiindex.<locals>.<listcomp>rW r ) r r7 rÖ rg r
rX r rR r )r> Z index_listsZmultirB rB rC Ú+test_dataframe_constructor_infer_multiindex~ s þzETestDataFrameConstructors.test_dataframe_constructor_infer_multiindexÚ
|
input_valsr rÊ rS ÚHrY rU c C s6 t d|i|d}t d|i d|i¡}t ||¡ d S )NrÎ r5 )r r; r< r= )r> r¦ ré rr rA rB rB rC Útest_constructor_list_str s z3TestDataFrameConstructors.test_constructor_list_strc C s< t dddd gi|d}t dddd gitd}t ||¡ d S )NrÎ r® r¯ r5 z1.0z2.0)r rm r< r= )r> ré rr rA rB rB rC Ú test_constructor_list_str_na s z6TestDataFrameConstructors.test_constructor_list_str_nar´ c sÖ |r"|s"|t jt j kr"t |¡ tjddg|d tjddg|djjdkrVd S t jddg|d
|
¡ }t d|dd d
|
dt d fd d }|s²| dj d< dj d< |sÒ|d t jddgjdj d d
df< jj d jks t|s |s |d |rº jjdkrn d j dd¡ksPtd j dd¡ks¢tn4 d j d¡kstd j d¡ks¢td |d ksÒtn|sÒd dksÒtd S )NrL rK r5 r4 r3 )ÚSÚU©r§ r¨ r© rà c S s0 t | tjr| jS t | jtjr(| jjS td S r )r
r7 ZndarrayÚbaser6 Z_ndarrayrx ©r rB rB rC Úget_base¿ s
|
z<TestDataFrameConstructors.test_dict_nocopy.<locals>.get_baseF©Zc_onlyc sv t fddjjD dks"t| r*d S t fddjjD dksNtt fddjjD dksrtd S )Nc 3 s | ]}| kV qd S r rB r )r© rB rC r Ï s zRTestDataFrameConstructors.test_dict_nocopy.<locals>.check_views.<locals>.<genexpr>rL c 3 s( | ] }t |jtjr| kV qd S r ©r
r6 r7 r )r§ r¯ rB rC r Ö s þc 3 s( | ] }t |jtjr| kV qd S r r± r )r¨ r¯ rB rC r Þ s þ)ÚsumrP r rR r° ©r§ r¨ r© r@ r¯ rB rC Ú check_viewsÈ s "þûÿ þûÿz?TestDataFrameConstructors.test_dict_nocopy.<locals>.check_viewsr r¸ rM Trý é. r} ru )F)r< Ú STRING_DTYPESr r r r7 ra r6 r{ r r´ r rô r¹ rà rR Útype) r> r
r´ Zany_numeric_ea_dtypeZany_numpy_dtyperS r» Zc_origr´ rB r³ rC Útest_dict_nocopy£ sF ÿþý
|
|
|
" z*TestDataFrameConstructors.test_dict_nocopyc C s2 t tddddgd}t dgd}t ||¡ d S )NrL r¤ rl rÑ r r
|
r? rB rB rC Ú'test_from_series_with_name_with_columns
|
s zATestDataFrameConstructors.test_from_series_with_name_with_columnsc C sd t dddgdddggdddgdd d
|
ggd }t dddgdddggt d d dg¡d }t ||¡ d S )NrL rK r4 r3 r rF rÎ r§ r¨ r© r )rÎ r§ )rÎ r¨ )rÎ r© )r r rB r< r= r? rB rB rC Útest_nested_list_columns
|
s ÿþz2TestDataFrameConstructors.test_nested_list_columnsc
|
C s t jddd}|j t¡ dd¡}t|}|jdks8t|j |j
|
k ¡ sLt||k ¡ ¡ s`tt j dddd g¡}|j t¡ dd¡}t|}|jdkst|j |j
|
k ¡ s²t||k ¡ ¡ sÆttj||||f j}t|}t||||d
|
} t || ¡ d S ) Nz
|
2016-04-05r4 rN rL rë ©rL r4 r3 r rF r )r r¢ rN r; rm r9 r r» rR rà r6 r Z IntervalIndexZ from_breaksr7 Zr_r r< r= )
|
r> Úpir r@ ÚiiZdata2rÁ Zdata3rò rA rB rB rC Ú1test_from_2d_object_array_of_periods_or_intervals
|
s zKTestDataFrameConstructors.test_from_2d_object_array_of_periods_or_intervalsz col_a, col_bc C s0 d}t jt|d t||d W 5 Q R X d S )Nz,Per-column arrays must each be 1-dimensionalr\ rä r' )r> Zcol_aZcol_bre rB rB rC Útest_error_from_2darray.
|
s z1TestDataFrameConstructors.test_error_from_2darrayc C s t dddgdddgdd}t |d j|d j¡r6td |jd
|
< t d tjtjgtjtjtjgtjtjtjgd dddgtd }t ||¡ d S ) NrL rK r4 r§ r¨ r© F)rX r r´ r r¸ r¬ ©rX r6 )
|
r r7 rÅ Z_valuesrR r¹ rb rm r< r= rI rB rB rC Ú&test_from_dict_with_missing_copy_false;
|
s
|
ýù z@TestDataFrameConstructors.test_from_dict_with_missing_copy_falsec C s8 d}t jt|d tt g ¡ddgd W 5 Q R X d S )NrK r\ r§ r¨ r= )rc rd r« r r7 ra r( rB rB rC Ú1test_construction_empty_array_multi_column_raisesL
|
s zKTestDataFrameConstructors.test_construction_empty_array_multi_column_raises)ßrã rä rå rD rU rZ rf rj rs rz rc ÚmarkÚ parametrizer r r r¡ r r£ r¦ r r¶ r½ r r Z skip_array_manager_invalid_testrÇ rÉ rÍ rÐ rã rç rê ró r÷ r r7 ra rm r r+ r, r- r r r r r r r# r% r) rb r, rµ r5 r: r> r@ rE rJ rU rY r^ rb re rh rj rl ro rq rt ru r r r r r r r r r r r r¡ r¥ r¨ r© r rµ r¶ rº r¼ rì r r¥ r6 rÔ r½ r Z BooleanDtypeZ Float64DtypeZ
|
Int64DtypeZ StringDtyperÀ 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 r r r r r r# r& r) r. r/ r1 r2 r3 r4 r6 r8 r< rE rH rN rR rX rZ r rh rw rt r[ rg ri rk rm ro rp rq r rs rt rv rw rx rz r} r~ r r< ZALL_NUMERIC_DTYPESZDATETIME64_DTYPESZTIMEDELTA64_DTYPESZ BOOL_DTYPESr r¶ r Z OBJECT_DTYPESr r Zarm_slowr 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 rB rB rB rC r1 M sò
|
! õþ
|
þþ
|
ýþ
|
|
|
|
úþ
|
|
úþ
|
|
|
|
úþ
|
&
|
|
|
|
4
|
|
|
|
üþ
|
|
þýþ
|
8G
|
"ûþ
|
|
üþ
|
|
'
|
|
|
üþ
|
|
|
|
2ùþ & ùþ
|
G øþ þ ùþ ;
|
(.8 ÿ4ùþ
|
ÿþýþ ÿ üþ
|
ûþ
|
c "ýþ r1 c @ s e Zd Zdd Zdd ZdS )Ú&TestDataFrameConstructorIndexInferencec C s| t jdddd}ttj t|¡|}t jdddd}ttj t|¡|}t||d}t jdddd}t |j
|
|¡ d S )Nz1/1/1999ú1/1/2012r} r z1/1/1980z 12/1/2001)r" r' ) r r¢ r r7 rÖ rg rO r r< rî rX )r> Zrng1r" Zrng2r' r@ r rB rB rC ÚAtest_frame_from_dict_of_series_overlapping_monthly_period_indexesT
|
s zhTestDataFrameConstructorIndexInference.test_frame_from_dict_of_series_overlapping_monthly_period_indexesc C sì t ddd}ttd|d}ttd| d¡d}ttd| d¡d}ttd}t|||d}tt|jt|j t|j td}t
|
|j|¡ t|||d }tt|jt|j t|j td} t
|
|j| ¡ t|||d
|
}
|
tt|jt|j t|j td} t
|
|
|
j| ¡ t|||d } tt|jt|j t|j td} t
|
| j| ¡ d } t j t | d t||||d W 5 Q R X t j t | d t|||d W 5 Q R X t j t | d t|||d W 5 Q R X d S )NrE r4 rN rW rP z
|
US/CentralrÒ r5 )rÎ rÓ rk )rk rÎ rÓ )rÓ rk rÎ z0Cannot join tz-naive with tz-aware DatetimeIndexr\ )rÎ rk rÓ r+ )rÎ rk r+ )r+ rÎ rk )r r r rn r r rì rX rm r< rî rc rd rx )r> rT r r Zser3Zser4r( Z exp_indexrÁ Z
|
exp_index3rò Zdf4Z
|
exp_index4re rB rB rC Ú/test_frame_from_dict_with_mixed_tzaware_indexes_
|
sH ÿ ÿ ÿ ÿ zVTestDataFrameConstructorIndexInference.test_frame_from_dict_with_mixed_tzaware_indexesN)rã rä rå rÇ rÈ rB rB rB rC rÅ S
|
s rÅ c @ s e Zd Zdd ZdS )Ú)TestDataFrameConstructorWithDtypeCoercionc C s t j dd¡}d}tjt|d t|dd W 5 Q R X t| ¡ dd}|jdk ¡ sZt
|
t j |d< d}tjt |d t|dd W 5 Q R X tjt |d t |d dd W 5 Q R X d}tjt |d t| d¡ W 5 Q R X tjt |d t |d d¡ W 5 Q R X d S )
|
NrÕ r z)Trying to coerce float values to integersr\ r^ r5 r¸ r[ r )r7 rÖ rg rc rd r« r Úroundrà r rR rb r r r; )r> r? re r@ rB rB rC Ú"test_floating_values_integer_dtype
|
s"
|
zLTestDataFrameConstructorWithDtypeCoercion.test_floating_values_integer_dtypeN)rã rä rå rË rB rB rB rC rÉ
|
s rÉ c @ sº e Zd Zej dddg¡dd Zej dddg¡d d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zej d eeg¡d d Zd d! Zd"d# Zd$d% Zd&S )'Ú&TestDataFrameConstructorWithDatetimeTZrJ r zdateutil/US/Easternc
|
C sÒ t dddd}| |¡}td|d|d}td|j}|d j|ksHtd
|
d |D }d d |D }td |i}||d< ||d< ||d< |j}tt d¡t|dt d¡t|dgd dddgd} t
|
|| ¡ d S )Nú2011/1/1ú2012/1/1úW-FRIr r¤ r rW ru rk c S s g | ] }| ¡ qS rB r? rA rB rB rC r ¸
|
s zeTestDataFrameConstructorWithDatetimeTZ.test_construction_preserves_tzaware_dtypes.<locals>.<listcomp>c S s g | ] }| ¡ qS rB r? rA rB rB rC r ¹
|
s rQ Údr_tzÚdatetimes_naiveÚdatetimes_with_tzrG r ) r rn r r Útzinfor6 rR rà r r7 r< rp )
|
r> rJ rQ rÐ r@ Z tz_expectedrÑ rÒ rr rA rB rB rC Ú*test_construction_preserves_tzaware_dtypes
|
s*
|
ü
|
ù zQTestDataFrameConstructorWithDatetimeTZ.test_construction_preserves_tzaware_dtypesÚpydtTFc C sf |}t d|d}|r | ¡ }d}tjt|d td|gidd W 5 Q R X d}tjt|d td|idgdd W 5 Q R X tjt|d t|gdd W 5 Q R X tjt|d ttj|gt ddd W 5 Q R X tjt|d t|dgdgdd
|
W 5 Q R X tjt|d tt
|
|ggdd W 5 Q R X tjt|d t|ggdgdd W 5 Q R X d S ) NÚ2019r úsCannot convert timezone-aware data to timezone-naive dtype. Use pd.Series\(values\).dt.tz_localize\(None\) instead.r\ r rG r5 ú/Cannot unbox tzaware Timestamp to tznaive dtyperÀ rÈ r5 ) r r@ rc rd r« r rx r7 ra rm r )r> Ztz_aware_fixturerÕ rJ rB re Zmsg2rB rB rC Ú'test_constructor_data_aware_dtype_naiveÊ
|
s* ÿ " zNTestDataFrameConstructorWithDatetimeTZ.test_constructor_data_aware_dtype_naivec C s t tdddddd}tddd}t||d }|d
|
jsBtd |d
|
jd
|
ksTtt |d
|
t|d
|
d¡ t |d t|d d¡ d S ) NrO r4 r rH r¤ rl Z20130110rN r rÎ zM8[ns, US/Easternrk ) r r r r6 rR r` r< rp r )r> rq rQ r@ rB rB rC Útest_from_dictì
|
s z5TestDataFrameConstructorWithDatetimeTZ.test_from_dictc C sª t ddddd}t|}t |d t|dd¡ tt|}t |d t|dd¡ t dddd}t|}t |d t|dd¡ tt|}t |d t|dd¡ d S ) NrO r4 r r¤ )rI rJ r` rl rH r )r r r< rp r )r> r+ rÁ rB rB rC Útest_from_indexú
|
s z6TestDataFrameConstructorWithDatetimeTZ.test_from_indexc C sB t ddd}t||d}td|d|d td|jd|d d S )NrÆ rÕ rN rW r¤ rä )r r r rº )r> rQ r rB rB rC Ú+test_frame_dict_constructor_datetime64_1680 s zRTestDataFrameConstructorWithDatetimeTZ.test_frame_dict_constructor_datetime64_1680c C s@ t dddd}t|}td|d|d}|d ¡ ¡ s<td S ) NrÍ rÎ rÏ r r¤ r rW rk )r r r r! r rR )r> rQ rB rÈ rB rB rC Ú+test_frame_datetime64_mixed_index_ctor_1681 s zRTestDataFrameConstructorWithDatetimeTZ.test_frame_datetime64_mixed_index_ctor_1681c C sX t ddddd}t|dgd}tdtdddtd ddtd
|
ddgi}t ||¡ d S ) Nz20130101T10:00:00r4 r r )r> rI r rJ Z
|
timestampsr r z20130101T10:01:00z20130101T10:02:00)r r r r< r= )r> rQ rr rA rB rB rC Ú test_frame_timeseries_column s
|
|
|
ýÿÿ zCTestDataFrameConstructorWithDatetimeTZ.test_frame_timeseries_columnc C sj ddg}ddddddd d
|
}t |d d d g|d}t ddtjtjfg|td d d gd}t ||¡ d S )NÚNevadaÚOhioç333333@ç333333@)r9 éÒ r ç333333û?çÍÌÌÌÌÌ @)rj r9 rã )rß rà r9 rã iÓ r )rá rä )râ rå rF )r r7 rb r r< r= )r> r Úpoprr rA rB rB rC Ú test_nested_dict_construction' s
|
þ ýzDTestDataFrameConstructorWithDatetimeTZ.test_nested_dict_constructionc C sd t dddd}|j t¡ dd¡}t|}|jdks8t|j|j k
|
¡ sLt||k
|
¡
|
¡ s`td S )Nz2016-04-05 04:30r4 rP rH rL rë r» ) r rN r; rm r9 r r» rR rà r6 r )r> rT r r@ rB rB rC Ú test_from_tzaware_object_array6 s zETestDataFrameConstructorWithDatetimeTZ.test_from_tzaware_object_arrayc C s t jtdtdtdgtdddtjtdddgtdd dtjtd
|
d dggtd j}t|d d dgd}dddg}|j|k ¡ st
|
d S )Nz2013-01-01 00:00:00z2013-01-02 00:00:00z2013-01-03 00:00:00z2013-01-01 00:00:00-0500r r z2013-01-03 00:00:00-0500z2013-01-01 00:00:00+0100ZCETz2013-01-03 00:00:00+0100r5 rÎ rk rÓ r rG rJ zdatetime64[ns, CET]) r7 ra r r rt rm r r rà r rR )r> r? r Zexpected_dtypesrB rB rC Ú$test_from_tzaware_mixed_object_array? s, ý
|
|
ý
|
|
ýõîýzKTestDataFrameConstructorWithDatetimeTZ.test_from_tzaware_mixed_object_arrayc C s: t d¡ d¡}t|dd}t| d¡}t ||¡ d S )NrÕ )r rK zdatetime64[ns, UTC]r5 )r7 r8 r9 r r; r< r= )r> Z
|
array_dim2r@ rA rB rB rC Ú test_from_2d_ndarray_with_dtype^ s zFTestDataFrameConstructorWithDatetimeTZ.test_from_2d_ndarray_with_dtyperá c C sh |dddh}d|j d}tjt|d td|i W 5 Q R X tjt|d t| W 5 Q R X d S )NrL rK r4 ú'z' type is unorderedr\ r§ )rã rc rd rx r r )r> rá rº re rB rB rC Ú!test_construction_from_set_raisesf s zHTestDataFrameConstructorWithDatetimeTZ.test_construction_from_set_raisesc C s> t jdddd dd¡}t|}tdd |jjD s:td S ) Nr r2 rG r5 r3 r4 c s s | ]}t |tV qd S r )r
r" )r r? rB rB rC r v s zeTestDataFrameConstructorWithDatetimeTZ.test_construction_from_ndarray_datetimelike.<locals>.<genexpr>)r7 r8 r9 r r rP r rR rÆ rB rB rC Ú+test_construction_from_ndarray_datetimelikeq s zRTestDataFrameConstructorWithDatetimeTZ.test_construction_from_ndarray_datetimelikec C s t j dd¡}t dg¡j}d}tjt|d t |dg|d W 5 Q R X t ddd g¡}tjt|d t |dd
|
gd W 5 Q R X d S ) NrÕ rK r¯ z'len\(arrays\) must match len\(columns\)r\ r¤ r5 r° r_ rÑ r )
|
r7 rÖ rg r ra r6 rc rd r« r )r> r? r6 re Zarr2rB rB rC Ú>test_construction_from_ndarray_with_eadtype_mismatched_columnsx s zeTestDataFrameConstructorWithDatetimeTZ.test_construction_from_ndarray_with_eadtype_mismatched_columnsc C sn dddgdddgg}t jtdd t|d d
|
hd W 5 Q R X t jtd d t|d d
|
d hd W 5 Q R X d S )NrL rK r4 r3 r rF zindex cannot be a setr\ r§ r¨ rW zcolumns cannot be a setr© r r' r¹ rB rB rC Ú"test_columns_indexes_raise_on_sets s
|
zITestDataFrameConstructorWithDatetimeTZ.test_columns_indexes_raise_on_setsN)rã rä rå rc rà rÄ rÔ rÙ rÚ rÛ rÜ rÝ rÞ rç rè ré rê ÚsetÚ frozensetrì rí rî rï rB rB rB rC rÌ ¬
|
s$
|
|
!
|
|
rÌ c C s" t | tr| jd S | jd S d S )Nr r¸ )r
r r¹ r® rB rB rC Úget1 s
|
|
rò c @ s e Zd Zejeedgddd Zejdd Zej
|
ddd g¡d
|
d Z d d Z dd Z dd Zej
|
dejejg¡dd Zej jddej
|
deejg¡dd Zdd Zej jddej
|
deejg¡d d Zej
|
dejejg¡d d Zd d! ZdS )"ÚTestFromScalarN©Úparamsc C s |j S r ©Úparam©r> r
rB rB rC Úbox s zTestFromScalar.boxc s dt di tkr dg d< |d kr4tjf S |tkrbtkrR fddS fddS n$tkrx fddS fd dS d S )
|
NrX rK rÎ r c s | | df |S )NrM rB ©r Úkwargs©Úextrary rB rC r £ s
|
ÿ ÿz,TestFromScalar.constructor.<locals>.<lambda>c s d| if |S ©NrÎ rB rú rü rB rC r § r c s | | gf |S r rB rú rü rB rC r ª r c s d| | gif |S rþ rB rú rü rB rC r ¬ s
|
|
ÿ ÿ)r r Ú functoolsÚpartialrß r )r> ry rù rB rü rC r s
|
zTestFromScalar.constructorr6 rg rv c C s8 |t j|d}t |j|k¡s"tt | ¡ ¡s4td S rÄ )r rt r7 r rà rR r! )r> r6 r r rB rB rC Útest_from_nat_scalar° s z#TestFromScalar.test_from_nat_scalarc C s( t d}||dd}t||ks$td S )NrL rv r5 ©r rò rR )r> r r r rB rB rC Ú*test_from_timedelta_scalar_preserves_nanos¶ s z9TestFromScalar.test_from_timedelta_scalar_preserves_nanosc C s, |t d }||dd}t||ks(td S )NrL rg r5 r )r> r Z fixed_now_tsrB r rB rB rC Ú*test_from_timestamp_scalar_preserves_nanos¼ s z9TestFromScalar.test_from_timestamp_scalar_preserves_nanosc C s4 t d}| ¡ }||td}tt|tjs0td S )NrL r5 )r Zto_timedelta64rm r
rò r7 r rR )r> r r Ztd64r rB rB rC Ú#test_from_timedelta64_scalar_object s z2TestFromScalar.test_from_timedelta64_scalar_objectÚclsc C s |dd}t jdt jdi| }|t jkr.d}nd}d d|g¡}tjt|d |||d
|
W 5 Q R X |d d}tjt|d |||d
|
W 5 Q R X d S ) Nrt ru rv rg z?dtype datetime64\[ns\] cannot be converted to timedelta64\[ns\]z?dtype timedelta64\[ns\] cannot be converted to datetime64\[ns\]ú|re r\ r5 r3 )r7 rw r Újoinrc rd rx )r> r r r* r6 Zmsg1re rB rB rC Ú(test_from_scalar_datetimelike_mismatchedÉ s
|
|
|
z7TestFromScalar.test_from_scalar_datetimelike_mismatchedziTimestamp constructor has been updated to cast dt64 to non-nano, but DatetimeArray._from_sequence has not)Úreasonc C s t ddd}d}|tjkr*t |d¡}d}||}t|}t|trJ|jn
|
|jjd }t |t
|
ksft |j j|ksvt ||kst d S )Né' rL r r+ úM8[s]r ) r r7 rw rò r
r r6 rà r¹ r· r rR Úasm8©r> r r r* rh rr Úitemr6 rB rB rC Ú#test_from_out_of_bounds_ns_datetimeÛ s
|
z2TestFromScalar.test_from_out_of_bounds_ns_datetimec C sd t t t j¡jd¡}||}t|}t|t jks8tt|t rH|j
|
n
|
|j j d }|t ks`td S ©Nr+ r )r7 rw Úiinfor0 Úmaxrò r· rR r
r r6 rà r¹ rm )r> r r* rr r r6 rB rB rC Ú test_out_of_s_bounds_datetime64ð s z.TestFromScalar.test_out_of_s_bounds_datetime64zoTimedeltaArray constructor has been updated to cast td64 to non-nano, but TimedeltaArray._from_sequence has notc C s t dddt ddd }d}|tjkr6t |d¡}d}||}t|}t|trV|jn
|
|jjd }t |t
|
ksrt |j j|kst ||kst d S )Nr rL i² zm8[us]r+ zm8[s]r ) r r7 r rò r
r r6 rà r¹ r· r rR r r rB rB rC Ú$test_from_out_of_bounds_ns_timedeltaø s
|
z3TestFromScalar.test_from_out_of_bounds_ns_timedeltac C s` |t t j¡jd}||}t|}t||ks4tt|trD|j n
|
|j
|
j d }|t ks\td S r ) r7 r r0 r rò r· rR r
r r6 rà r¹ rm )r> r r r* rr r r6 rB rB rC Ú test_out_of_s_bounds_timedelta64 s z/TestFromScalar.test_out_of_s_bounds_timedelta64c C sd d}t d|d}|d ks(|tkr2|tkr2d}t}nd}t}tj||d ||dd W 5 Q R X d S ) Nr rÖ r rØ r× r\ rg r5 )r r rß rx r« rc rd )r> r rù ry rJ rB re ÚerrrB rB rC Ú test_tzaware_data_tznaive_dtype s ÿz.TestFromScalar.test_tzaware_data_tznaive_dtype)rã rä rå rc Úfixturerì rß rù r rà rÄ r r r r r7 rw r r Zxfailr r r r r r r rB rB rB rC ró s2
|
|
|
|
ÿ ÿ
|
ró c @ sZ e Zd Zejddgddd Zejdd Zdd Zd
|
d Zd d Z dd Z
|
dd Z dS )ÚTestAllowNonNanoTFrô c C s |j S r rö rø rB rB rC Úas_td- s zTestAllowNonNano.as_tdc C sL t d¡ t j¡ d¡}|r8||d }tj||jdS tj||jdS d S )Nr r r r5 ) r7 r8 r; r0 r r& Z _simple_newr6 r" )r> r rº rB rB rC r? 1 s
|
zTestAllowNonNano.arrc C s t |}|j|jkstd S r )r r6 rR )r> r? rq rB rB rC Útest_index_allow_non_nano: s z*TestAllowNonNano.test_index_allow_non_nanoc C s, |rt |¡}nt|}|j|jks(td S r )r ZTimedeltaIndexr r6 rR )r> r? r rq rB rB rC Útest_dti_tdi_allow_non_nano> s z,TestAllowNonNano.test_dti_tdi_allow_non_nanoc C s t |}|j|jkstd S r )r r6 rR )r> r? rÐ rB rB rC Útest_series_allow_non_nanoE s z+TestAllowNonNano.test_series_allow_non_nanoc C s t |}|jd |jks td S r ©r rà r6 rR rÆ rB rB rC Útest_frame_allow_non_nanoI s z*TestAllowNonNano.test_frame_allow_non_nanoc C s$ t d|i}|jd |jks td S r r rÆ rB rB rC Ú#test_frame_from_dict_allow_non_nanoM s z4TestAllowNonNano.test_frame_from_dict_allow_non_nanoN) rã rä rå rc r r r? r r r r r! rB rB rB rC r ) s
|
|
r )Hra Ú collectionsr r r r Z dataclassesr r r r rÿ rÖ r Útypingr
|
r± Únumpyr7 r Znumpy.mar rc rK Z pandas.errorsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr Zpandas.core.dtypes.commonr Zpandas.core.dtypes.dtypesr r r r Zpandasr r r r r r r r r r r r r r r r! Zpandas._testingZ_testingr< Z pandas.arraysr" r# r$ r% r& rÜ rÛ r1 rÅ rÉ rÌ rò ró r rB rB rB rC Ú<module> sx D
|
ø 9 a
|
