U
|
ý°d¤D ã @ s¤ d dl Zd dlZd dlmZmZ d dlmZ d dlZ d dlm
|
Z
|
m Z m Z m Z mZ d dlmZ d dlmZmZmZmZmZmZ G dd dZej dej d d gej d
|
ej d fe d d g¡ej dfe d d g¡e j dfe j ddgddedfe j!ddgddee j"j#j# $d¡fe j% &d d dg¡edfe ddg¡ed fe 'd g¡ed fg¡d d Z(ej d!e d dd"g¡g¡d#d$ Z)d%d& Z*ej d'e d d g¡d(fe j"j+j,d)d*gd+dd,fe j d ej-gd-d
|
d.fe &d d g¡d/fed d gd0feej d dgd d
|
d,feej d1d2gd3d
|
eddd
|
d,fg¡d4d5 Z.d6d7 Z/ej d8ej d dgej d
|
ej d dgej d
|
fe d d g¡ej d d ge0d
|
fe j"j+j,d)d*gd+de e j1d)d+de j1d*d+dg¡fe j d ej-gd-d
|
ej d e j2ge0d
|
fe &d d dg¡ej e 3d d ¡e 3d d¡ge0d
|
fed d gej d d gej d
|
feej d)d*gd3d
|
ej d)d*gd3d
|
feej d9d:gd3d
|
eddd
|
e e d;dde d<ddg¡feej d d=gd>d
|
d?dej d d=gd d
|
fe ed@ddAdB¡e e d@dAde dCdAdg¡fg
|
¡dDdE Z4ej dFdGdHg¡ej d!ej d dd"gd d
|
ej d d dIge0d
|
g¡dJdK Z5ej dFdGdHg¡dLdM Z6ej dNd ddgdOd dPdQdRgfe d)e d)e j7gde d)e 8dS¡gd" fg¡dTdU Z9ej dVd dddWgdXdYdZd[ge:ddPdQej-d\gfd dddWgdXdYdZd[ge:ej-dPdQej-d\gfdPdQej-d\gd]d^d_d`ge;d d dd dWgfe d)e d)e j7gd e dafd e dbfd e d)fgde d)e 8dS¡gd" fg¡dcdd Z<dedf Z=ej dgd dd"gd ddgdhe d dd"g¡e d dej-g¡dhe d dd"g¡e d ddg¡dhg¡ej die:ej-fe0dfg¡djdk Z>ej dld e d ddg¡iej dPgdQgej-gge:d
|
fd dd"gd dd"gdhej d d gddgd"d"gge:d
|
fg¡dmdn Z?dodp Z@G dqdr drZAdS )sé N)Úis_datetime64_dtypeÚis_timedelta64_dtype)ÚDatetimeTZDtype)ÚCategoricalIndexÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_range)Ú DatetimeArrayÚ IntervalArrayÚ PandasArrayÚ PeriodArrayÚ SparseArrayÚTimedeltaArrayc @ s e Zd Zdefdefdefdefdefdefdefdefd efd
|
efd efd efd efdefgZej
|
de¡ej j
|
ddd dd dd dd gddddgddd Z ej
|
d d e d fd ed fd!e d fd!ed fg¡ej j
|
dd"d d#d d$d d%d gddddgdd&d' Z ej
|
de¡d(d) Zej
|
ded efd!efg ¡d*d+ Zej j
|
dd,d d-d d.d d/d gddddgdd0d1 Zd2d3 Zd4S )5ÚTestToIterableZint8Úint16Úint32Úint64Zuint8Zuint16Zuint32Zuint64Úfloat16Úfloat32Úfloat64údatetime64[ns]údatetime64[ns, US/Eastern]útimedelta64[ns]z dtype, rdtypeÚmethodc C s | ¡ S ©N©Útolist©Úx© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/base/test_conversion.pyÚ<lambda>5 ó zTestToIterable.<lambda>c C s | ¡ S r ©Úto_listr r r r! r" 6 r# c C s t | S r ©Úlistr r r r! r" 7 r# c C s t | ¡ S r ©r' Ú__iter__r r r r! r" 8 r# r r% r' Úiter)Úidsc C sp |}|dkrDt |tjrDtjtdd |dg|d W 5 Q R X d S |dg|d}||d }t||sltd S ©Nr zfloat16 indexes are not ©Úmatché ©Údtyper )Ú
|
issubclassÚpdÚIndexÚpytestÚraisesÚNotImplementedErrorÚ
|
isinstanceÚAssertionError)ÚselfÚindex_or_seriesr r1 ÚrdtypeÚtypÚsÚresultr r r! Ú test_iterable1 s z TestToIterable.test_iterablezdtype, rdtype, objÚobjectÚar/ Úcategoryc C s | ¡ S r r r r r r! r" U r# c C s | ¡ S r r$ r r r r! r" V r# c C s t | S r r& r r r r! r" W r# c C s t | ¡ S r r( r r r r! r" X r# c C s0 |}||g|d}||d }t ||s,td S )Nr0 r )r8 r9 ) r: r; r r1 r< Úobjr= r> r? r r r! Ú!test_iterable_object_and_categoryI s z0TestToIterable.test_iterable_object_and_categoryc C sV t dg|d}t| ¡ d \}}t||s0tt| ¡ d \}}t||sRtd S )Nr/ r0 r )r r' Úitemsr8 r9 )r: r1 r< r> Ú_r? r r r! Útest_iterable_itemsg s
|
z"TestToIterable.test_iterable_itemsc C s |}|dkrDt |tjrDtjtdd |dg|d W 5 Q R X d S |dg|d}| t¡d }t|t sp|f}||ks|t
|
d S r, ) r2 r3 r4 r5 r6 r7 ÚmapÚtyper8 Útupler9 )r: r; r1 r< r= r> r? r r r! Útest_iterable_maps s
|
z TestToIterable.test_iterable_mapc C s | ¡ S r r r r r r! r" r# c C s | ¡ S r r$ r r r r! r" r# c C s t | S r r& r r r r! r" r# c C s t | ¡ S r r( r r r r! r" r# c C s2 t tdtdg}||d }t|ts.td S )Nz
|
1999-12-31z
|
2000-12-31r )r r r8 r9 )r: r Úir? r r r! Ú test_categorial_datetimelike s z+TestToIterable.test_categorial_datetimelikec C s¢ t dt dg}t|}|jdks&tt||D ]0\}}t|t sFt|jd ksTt||ks0tq0t dddt dddg}t|}|jdkstt||D ]2\}}t|t s°t|j|jksÀt||kstqtdtdg}t|}|jd ksôtt||D ]$\}}t|tst||ksþtqþtj dd
|
d tj dd
|
d g}t|}|jd ksXtt||D ]:\}}t|tj s|t|j
|
d
|
kst||ksbtqbd S ) Nz
|
2011-01-01z
|
2011-01-02r ú
|
US/Eastern©Útzr z1 daysz2 daysr ÚM©Úfreqz Period[M]) r r r1 r9 Úzipr8 rQ r r3 ÚPeriodrT )r: Úvalsr> ÚresÚexpr r r! Ú test_iter_box s:
|
|
þ z TestToIterable.test_iter_boxN)Ú__name__Ú
|
__module__Ú __qualname__ÚintÚfloatr r Údtypesr5 ÚmarkÚ parametrizer@ rA rE rH rL rN rZ r r r r! r s| ò ü
|
ø
|
üþ ü
|
ø
|
|
ÿ
|
ü
|
ø
|
|
r zarr, expected_type, dtyper/ r0 r rB ÚbrA rC Z2017Z2018ú
|
US/CentralrP zdatetime64[ns, US/Central]iâ iã ÚArS zA-DECé Úintervalr l d( zm8[ns]c C sJ t | j}t | ¡j}t||ks&tt|t|ks:tt ||¡ d S r )r Z_valuesr3 r4 rJ r9 ÚtmZ assert_equal)ÚarrÚ expected_typer1 Zl_valuesZr_valuesr r r! Útest_values_consistent» s
|
|
rk ri é c C s& t | }|j}t| }t ||¡ d S r )r Úarrayr rh Z assert_extension_array_equal)ri Úserr? Úexpectedr r r! Útest_numpy_arrayá s rp c C sR t | d}|j}t| r(t|tsNtn&t| r@t|tsNtnt|tsNtd S )Nr0 ) r rm r r8 r
|
r9 r r r )Zany_numpy_dtypern r? r r r! Útest_numpy_array_all_dtypesé s
|
rq z arr, attrÚ_codesÚ2000Ú2001ÚDZ_ndarrayÚInt64Ú_dataÚ_leftZ_sparse_valuesz2000-01-01T12:00:00z2000-01-02T12:00:00úM8[ns]c C s: |}|| ddj }|r*t| |} t||}|| ks6td S )NF©Úcopy)rm Úgetattrr9 )ri Úattrr; ÚrequestÚboxr? r r r! Ú
|
test_arrayô s
|
|
r c C s> t j dgddgg¡} d}tjt|d | j W 5 Q R X d S )Nre rB rc z&MultiIndex has no single backing arrayr- )r3 Ú
|
MultiIndexZ from_productr5 r6 Ú
|
ValueErrorrm )ÚidxÚmsgr r r! Ú test_array_multiindex_raises s r
z arr, expectedz2000-01-01T06:00:00z2000-01-02T06:00:00ú
|
2000-01-01ú
|
2000-01-02l qa Úi8ÚHz
|
2016-01-01z
|
US/Pacific©ÚperiodsrQ z
|
2016-01-02c C s |}t d ¡ || }W 5 Q R X | jjdkrR|tjkrRtjjdd}|j
|
|¡ | ¡ }t ||¡ t |¡}t ||¡ d S )Nr z,thing is Int64 and to_numpy() returns object)Úreason)rh Úassert_produces_warningr1 Únamer3 rm r5 ra ZxfailÚnodeZ
|
add_markerÚto_numpyÚassert_numpy_array_equalÚnpÚasarray)ri ro Zindex_or_series_or_arrayr~ r Úthingra r? r r r! Ú test_to_numpy s 5
|
r Ú as_seriesTFÚcc C s t j| dd}|r t|jdd}| ¡ }t | |¡dks<t|jdd}t | |¡dks\t|jdd}t | |¡dks|td S )NFrz T)r3 r4 r Úvaluesr r Z shares_memoryr9 )ri r rD r? r r r! Útest_to_numpy_copya s r c C s d}t jddg|d}| r"t|}| ¡ }tjtd|dtd|dgtd}t ||¡ |jdd}t ||¡ |jdd}tjdd gdd}t ||¡ d S )
|
NrO rs rt rP r0 rA ry z 2000-01-01T05z 2001-01-01T05)
|
r3 Ú DatetimeIndexr r r rm r rA rh r )r rQ rD r? ro r r r! Útest_to_numpy_dtypev s ÿ r z!values, dtype, na_value, expectedr ç ð?ç @ç z 2000-01-01T00:00:00.000000000c C s0 | |}|j ||d}t |¡}t ||¡ d S ©N©r1 Úna_value)r r rm rh r )r; r r1 r¡ ro rD r? r r r! Ú"test_to_numpy_na_value_numpy_dtype s
|
r¢ z+data, multiindex, dtype, na_value, expectedé )r rB )r rc )r/ rc )r/ r g @)rB r )rB r/ )rB rf )rc r Z2021Z2022c C s@ t j |¡}t| |d}|j||d}t |¡}t ||¡ d S )N)Úindexr ) r3 r Ú from_tuplesr r r rm rh r )ÚdataZ
|
multiindexr1 r¡ ro r¤ Zseriesr? r r r! Ú(test_to_numpy_multiindex_series_na_value¡ s
|
$
|
r§ c C st t dddg} d}tjt|d | jdd W 5 Q R X t dddgdd } tjt|d | jdd W 5 Q R X d S )
|
Nr/ rf rl z5to_numpy\(\) got an unexpected keyword argument 'foo'r- T)Zfoorv r0 )r r5 r6 Ú TypeErrorr )r> r r r r! Útest_to_numpy_kwargs_raisesÌ s r© r¦ )rB rc zdtype, na_valuec C sH t | ¡}|j||d}tjddgddgd|gg|d}t ||¡ d S )Nr r/ rf rl r0 )r3 Ú DataFramer r rm rh r )r¦ r1 r¡ Údfr? ro r r r! Ú test_to_numpy_dataframe_na_valueÙ s
|
r¬ zdata, expectedc C s* t | ¡}|jttjd}t ||¡ d S r )r3 rª r r_ r Únanrh r )r¦ ro r« r? r r r! Ú$test_to_numpy_dataframe_single_blockê s
|
r® c C sL t t ddtjg¡¡} t t ddtjg¡¡}| jdd t | |¡ d S )Nr r r )r¡ )r3 rª r rm r r rh Zassert_frame_equal)r? ro r r r! Ú.test_to_numpy_dataframe_single_block_no_mutateþ s r¯ c @ s6 e Zd Zej dddg¡dd Zdd Zdd ZdS )
|
Ú TestAsArrayrQ Nrd c C sd t tdd|d}t d ¡ tj|td}W 5 Q R X t td|dtd|dg¡}t ||¡ d S )Nrs rf r r0 r rP r )
|
r r rh r r r rA rm r r )r: rQ rn r? ro r r r! Útest_asarray_object_dt64 s ÿz$TestAsArray.test_asarray_object_dt64c C s< t tddd}tjddgdd}t |¡}t ||¡ d S )Nrs rf )r r r ry r0 ©r r r rm r rh r )r: rn ro r? r r r! Útest_asarray_tz_naive s
|
z!TestAsArray.test_asarray_tz_naivec C s` d}t tdd|d}tjddgdd}tj|d d}t ||¡ tj|dd}t ||¡ d S )
|
Nrd rs rf r z 2000-01-01T06z 2000-01-02T06ry r0 r r² )r: rQ rn ro r? r r r! Útest_asarray_tz_aware s z!TestAsArray.test_asarray_tz_aware) r[ r\ r] r5 ra rb r± r³ r´ r r r r! r° s
|
r° )BÚnumpyr r5 Zpandas.core.dtypes.commonr r Zpandas.core.dtypes.dtypesr Zpandasr3 r r r r r Zpandas._testingZ_testingrh Zpandas.core.arraysr
|
r r r r r r ra rb rm r ZndarrayZ Categoricalr Z PeriodIndexÚcorer` Z PeriodDtypeZ IntervalIndexZ from_breaksZTimedeltaIndexrk rp rq ZarraysZ period_arrayr r r
rA rV ZNAZIntervalr r r ZNaTZ
|
datetime64r¢ r_ r^ r§ r© r¬ r® r¯ r° r r r r! Ú<module> sT
|
ýý ý
|
ýìþ
|
|
ÿüùøþ
|
& þ& þ þ ÿü
|
|
þÿùþ
|
|
þÿþÙþ4
|
&ÿ
|
üþþ
|
|
|
û
|
|
û
|
|
û ûêþ!
|
|
" ýþ þ þûþ
|
|
