U
|
ý°d´0 ã @ s d dl Zd dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z m Z d dlmZ d dlmZ ej dddg¡dd Zdd Zej d
|
d d g¡ej dddg¡ej d dejd dddg¡ej dejdddgddejdddgddg¡dd Zej dd d dg¡dd Zej d eddgee de dge ededge e d e d gg¡d d Zd d! Z ej d
|
d d g¡ej dddg¡ej d dejd dddg¡d"d# Z d$d% Z ej d&d'd( d)d( g¡ej d*dd+gg¡d,d- Z ej ddddg¡ej d.dd ddgg¡ej d*dd+d/gd+d/d0gg¡d1d2 Z ej dddg¡d3d4 Z!ej d5e
|
eg¡ej d6ddgdfddgdfd+d/gdfg¡d7d8 Z"ej d5e
|
eg¡d9d: Z#d;d< Z$ej d.dd ddgg¡d=d> Z%ej dd dd g¡d?d@ Z&dAdB Z'dS )Cé N) Ú DataFrameÚ DatetimeIndexÚIndexÚPeriodÚ PeriodIndexÚSeriesÚ TimedeltaÚTimedeltaIndexÚ Timestamp)Ú get_arrayÚdtypeÚint64c C s t dddgdd}t || d}t t|t|¡s6t|rP|jjd j ¡ sPt|rd|j d< |j d dksptt t|t|¡r¾tn4d|j d< |j d dks¦tt t|t|¡s¾tt || d}|rìd|j d< |j d dksêtn d|j d< |j d dks
|
td S )Né é é Úname©r ©r r )
|
r ÚnpÚ shares_memoryr ÚAssertionErrorÚ_mgrÚblocksÚrefsÚ has_referenceÚiloc)r Úusing_copy_on_writeÚserÚresult© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/copy_view/test_constructors.pyÚtest_series_from_series s$
|
|
|
|
r! c C sÔ t dddgdd}|j|j ¡ t|j|j d¡fD ]X}t ||d}t |j|j¡sXtd|j d< | rz|j d dkstq4|j d dks4tq4t |ddddgd}t |j|j¡r¶t| rÐ|j
|
j d j ¡ rÐtd S ) Nr r r r r Úidx)Úindexr )r r# ÚcopyÚlistÚrenamer r Úvaluesr r r r r r )r r r# r r r r Ú$test_series_from_series_with_reindex@ s
|
ü
|
r( ÚfastpathFTr" r r )ÚstartÚstopÚstepÚarrr r ZInt64c C s° |d ks|d k rd}t ||||d}| ¡ }t|d|}| rRt t||¡rftnt t||¡sftd|d< | rt ||¡ n,t dddg|d k r|n|j d}t ||¡ d S ) NF)r r# r) Ú_dataéd r r r r )
|
r r$ Úgetattrr r r r ÚtmÚassert_series_equalr ) r r" r r) r- r Úser_origÚdataÚexpectedr r r Útest_series_from_array] s r6 r$ c C s: t jdddgdd}t|d|d}t t||¡r6td S )Nr r r r r Úint32)r r$ )r Úarrayr r r r )r r$ r- r r r r Ú&test_series_from_array_different_dtypev s r9 ú
|
2019-12-31ú
|
2020-12-31z1 daysz2 daysc C sz t |}|jdd}| rBt t|t|¡s0t|j d¡rZtnt t|t|¡rZt|jd |jd< t
|
||¡ d S )NT)Údeepr r ) r r$ r r r r r Ú_has_no_referencer r1 Zassert_index_equal)r r" r r5 r r r Útest_series_from_index} s
|
r> c C sN t dddgdd}t|dd}t t|t|¡r6t| rJ|j d¡sJtd S ©Nr r r r r r7 r )r r r r r r r r= )r r" r r r r Ú'test_series_from_index_different_dtypes s
|
r@ c C s t dddgdd}| ¡ }t |j|||d}t t|t|¡sDt| rX|j d¡rXtd|jd< | rtt
|
||¡ nt dddg}t
|
||¡ d S ) Nr r r r r )r r) r# r r/ ) r r$ r r r r r r= r r1 r2 )r r" r r) r r3 Úser2r5 r r r Ú test_series_from_block_manager s
|
rB c C sP t dddgdd}t |jdd}t t|t|¡r8t| rL|j d¡sLtd S r? )r r r r r r r= )r r rA r r r Ú.test_series_from_block_manager_different_dtype s
|
rC Úfuncc C s | S ©Nr ©Úxr r r Ú<lambda>µ ó rH c C s | j S rE )r rF r r r rH µ rI ÚcolumnsÚac C s¦ t ddddgi}| ¡ }t ||}t t|dt|d¡sBtd|jd< | rzt t|dt|d¡rltt ||¡ n(t t|dt|d¡stt ||¡ d S )NrK r r r r/ r ) r r$ r r r r r r1 Úassert_frame_equal)r rJ rD ÚdfÚdf_origZnew_dfr r r Ú$test_dataframe_constructor_mgr_or_dfµ s
|
rO r# ÚbÚcc
|
C sF t dddg}t dddg}| ¡ }tdddgdddgd|||d}t||d|||d d
|
} t t| d t|¡sxtd | jd < |r®t t| d t|¡r tt ||¡ n|jd d ksÀtt dddg}t dddg}t||d|||d d
|
} d |jd< |r.t t| d t|¡r tt
|
| |¡ n| jd d ksBtd S )Nr r r é é é ©rK rP )r# rJ r F)r# rJ r r$ rK é
|
©r r r ) r r$ r r r r r r r1 r2 rL )
|
Úrequestr rJ r# r Ús1Ús2Zs1_origr5 r r r r Ú"test_dataframe_from_dict_of_seriesÈ sH ÿ ÿ
|
ÿ
|
r[ c C s~ t dddg}t dddg}t||ddddg| dd }t|d
|
}t |t|¡rVtd |jd < t|d
|
}t ||¡sztd S ) Nr r r rR rS rT rU F©r# r r$ rK r/ rW ©r r r r r r r )r rY rZ rM Ú
|
arr_beforeÚ arr_afterr r r Ú/test_dataframe_from_dict_of_series_with_reindexõ s
|
|
|
r` Úconsz data, dtypec C sp |||d}| ¡ }t||d}t t|t|d¡s:t| rN|j d¡rNt|d |jd< | rlt
|
||¡ d S )Nr r éÿÿÿÿrW ) r$ r r r r r r r= r r1 Z assert_equal)r r4 r ra ÚobjZobj_origrM r r r Ú#test_dataframe_from_series_or_index s rd c C sN |ddgdd}t |dd}t t|t|d¡r6t| rJ|j d¡sJtd S )Nr r r r r7 r )r r r r r r r= )r ra rc rM r r r Ú3test_dataframe_from_series_or_index_different_dtype s
|
re c C sR t tdtdgtd}t|}t t|t|d¡r:t| rN|j d¡sNtd S )Nr: r; r r )
|
r r
|
Úobjectr r r r r r r= )r r rM r r r Ú)test_dataframe_from_series_infer_datetime! s
|
rg c C sx t dddg}t dddg}t||d| dd d
|
}t|d }t |t|¡rPtd |jd < t|d }t ||¡sttd S )Ng ð?g @g @rR rS rT rU r Fr\ rK r/ rW r] )r# rY rZ rM r^ r_ r r r Ú-test_dataframe_from_dict_of_series_with_dtype) s
|
|
|
rh c C st t ddgddgg¡}t||d}| r.|dk sB|dksB|rZ|d krZt t|d|¡rptnt t|d|¡sptd S ) Nr r r rR )r$ FTr )r r8 r r r r )r r$ Zusing_array_managerr- rM r r r Útest_frame_from_numpy_array: s ÿþýüüri c C s t ddddgi}| ¡ }t |¡}| r8|j d¡r8tt t|dt|d¡sTtd|j d< | rpt
|
||¡ n t
|
||¡ d S )NrK r r r r r/ rW ) r r$ Z from_recordsr r= r r r r r r1 rL )r rM rN Zdf2r r r Ú*test_dataframe_from_records_with_dataframeJ s
|
|
rj )(Únumpyr ZpytestZpandasÚpdr r r r r r r r r
|
Zpandas._testingZ_testingr1 Zpandas.tests.copy_view.utilr ÚmarkZ parametrizer! r( Z
|
RangeIndexr8 r6 r9 r> r@ rB rC rO r[ r` rd re rg rh ri rj r r r r Ú<module> sp ,
|
( &ÿ
|
|
üþ
|
*
|
ÿ
|
|
|
|
