U
|
ý°dl" ã @ sH d dl Zd dlZd dlmZ d dlm mZ d dl Z
|
d dl m Z m Z m Z mZ d dlmZ d dlmZ dd Zej ddd g¡ej d
|
dd d g¡d d Zej dddddg¡dd Zejdd Zej dddg¡dd Zej dddg¡dd Z d d Z d d Z d!d" Z d#d$ Z ejj!ed%d&d'd( Z"d)d* Z#d+d, Z$dS )-é N)Úpa_version_under7p0)Ú DataFrameÚSeriesÚ TimestampÚ
|
date_range)Ú get_arrayc C s t dddgdddgdd}| ¡ }| d ¡}| rlt t|d
|
t|d
|
¡sNtt t|d t|d ¡r¤tn8t t|d
|
t|d
|
¡rtt t|d t|d ¡r¤td |jd < | rÎt t|d
|
t|d
|
¡rÎtt ||¡ | d ¡}d |jd < t || d ¡¡ d S )Né é é é é é ç ø?©ÚaÚbÚcÚfloat64r r ç @©r r ©
|
r ÚcopyÚastypeÚnpÚ shares_memoryr ÚAssertionErrorÚilocÚtmÚassert_frame_equal©Úusing_copy_on_writeÚdfÚdf_origÚdf2© r$ úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/copy_view/test_astype.pyÚtest_astype_single_dtype s
|
|
|
|
r& ÚdtypeÚint64ÚInt64Ú new_dtypeúint64[pyarrow]c C sÜ |dkrt rt d¡ tddddgi|d}| ¡ }| |¡}| r`t t|dt|d¡s|t n t t|dt|d¡r|t d|j
|
d < | r¦t t|dt|d¡r¦t t ||¡ | |¡}d
|
|j
|
d < t || |¡¡ d S ) Nr+ úpyarrow not installedr r r r
|
©r' é
|
©r r éd ) r ÚpytestÚskipr r r r r r r r r r ©r r' r* r! r" r# r$ r$ r% Útest_astype_avoids_copy* s
|
|
|
|
|
r4 r Úint32ZInt32úint32[pyarrow]c C sª |dkrt rt d¡ tddddgi}| ¡ }| |¡}t t|dt|d¡rVt | rj|j
|
d¡sjt d|j d < t ||¡ | |¡}d
|
|j d < t || |¡¡ d S ) Nr6 r, r r r r
|
r r r/ r0 )r r1 r2 r r r r r r r Ú_mgrÚ_has_no_referencer r r )r r' r! r" r# r$ r$ r% Ú"test_astype_different_target_dtypeD s
|
|
|
|
|
r9 c C sL t dddg} t dd¡ | d¡}W 5 Q R X t t| t|¡sHtd S )Nr r r
|
zmode.copy_on_writeTr) )r ÚpdZoption_contextr r r r r )ÚserÚresultr$ r$ r% Útest_astype_numpy_to_eaY s r= zdtype, new_dtype)ÚobjectÚstring)r? r> c C s t ddddgi|d}| ¡ }| |¡}| rJt t|dt|d¡sftn t t|dt|d¡rftd|jd< t ||¡ d S ©Nr r r r- Úxr/ r r3 r$ r$ r% Ú test_astype_string_and_objecta s
|
|
rB c C s t ddddgi|d}| |¡}| ¡ }| rJt t|dt|d¡sftn t t|dt|d¡rftd|jd< t ||¡ d S r@ )
|
r r r r r r r r r r )r r' r* r! r# r" r$ r$ r% Ú-test_astype_string_and_object_update_originalr s
|
|
rC c C s\ t dddgdddgtdddgdd d
|
}| ¡ }| ddd ¡}| rt t|d t|d ¡sbtt t|d t|d ¡s~tt t|dt|d¡rðtnTt t|d t|d ¡r¸tt t|d t|d ¡rÔtt t|dt|d¡rðtd|jd< | r t t|d t|d ¡r td|jd< | rLt t|d t|d ¡rLtt
|
||¡ d S )Nr 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% Útest_astype_dict_dtypes
s& $ÿ
|
|
rD c C sT t dtddddi}| d¡}t t|dt|d¡r<t| rP|j d¡sPtd S )Nr ú
|
2019-12-31r ÚD)ÚperiodsÚfreqzdatetime64[ms]r ) r r r r r r r r7 r8 ©r r! r< r$ r$ r% Ú$test_astype_different_datetime_resos s
|
|
rJ c C sV t dtdddddi}| d¡}| rR|j d¡r6tt t|dt|d¡sRtd S ) Nr rE r rF ú
|
US/Pacific©rG rH Útzz datetime64[ns, Europe/Berlin]r © r r r r7 r8 r r r r rI r$ r$ r% Ú test_astype_different_timezones© s ÿ
|
rO c C sV t dtdddddi}| d¡}| rR|j d¡s6tt t|dt|d¡rRtd S ) Nr rE r rF rK rL z datetime64[ms, Europe/Berlin]r rN rI r$ r$ r% Ú.test_astype_different_timezones_different_reso³ s ÿ
|
rP r, )Úreasonc C s\ t dtdtdgidd}| d¡}| rX|j d¡r:tt t|dt|dj ¡rXtd S )Nr z2020-01-01 01:01:01.000001zM8[ns]r- ztimestamp[ns][pyarrow]r )
|
r r r r7 r8 r r r r Ú_datarI r$ r$ r% Útest_astype_arrow_timestamp½ s þÿù
|
rS c C sx t dddg}| ¡ }|jddddd}| rFt t|t|¡s^tnt t|t|¡r^td|jd< t ||¡ d S )Nr r r F)Zconvert_integerZconvert_booleanZconvert_floatingZconvert_stringrA r )
|
r r Úconvert_dtypesr r r r r r Zassert_series_equal)r r; Zser_origr< r$ r$ r% Ú!test_convert_dtypes_infer_objectsÐ s ü
|
rU c C s4 t ddgddgddgddgd }| ¡ }| ¡ }| r¨t t|dt|d¡sRtt t|d
|
t|d
|
¡sntt t|dt|d¡stt t|d t|d ¡s¦tnrt t|dt|d¡rÄtt t|dt|d¡ràtt t|d t|d ¡rütt t|d
|
t|d
|
¡rtd |jd < t ||¡ d S )Nr r r r r g @TF)r r r ÚdrV r rA r/ )
|
r r rT r r r r r r r r r$ r$ r% Útest_convert_dtypesã s "
|
rW )%Únumpyr r1 Z pandas.compatr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr: r r r r Zpandas._testingZ_testingr Zpandas.tests.copy_view.utilr r& ÚmarkZ parametrizer4 r9 Z skip_array_manager_invalid_testr= rB rC rD rJ rO rP ZskipifrS rU rW r$ r$ r$ r% Ú<module> sB
|
|
ÿ
|
ÿ
|
|
|
|
|
