U
|
®ý°d^ ã @ sÚ d dl m Z mZ d dlmZ d dlZd dlZd dlZd dlZd dl m
|
Z
|
d dl m m Z d dlmZmZmZmZmZmZmZmZmZmZmZmZ d dl m Z G dd dZ G dd d Z G d
|
d d Z!G d d d Z"dS )é )ÚdatetimeÚ timedelta)ÚreloadN)ÚiNaT) ÚNAÚ CategoricalÚCategoricalDtypeÚDatetimeTZDtypeÚIndexÚIntervalÚNaTÚSeriesÚ TimedeltaÚ TimestampÚcutÚ
|
date_rangec @ s6 e Zd Zdd Zdd Zej dee g¡dd Z
|
dS ) Ú TestAstypeAPIc C s¶ t dddgdd}| ¡ }d}tjt|d | tj¡ W 5 Q R X tjt|d | tj¡ W 5 Q R X tjt|d | d¡ W 5 Q R X tjt|d | d¡ W 5 Q R X d S )Nz
|
1970-01-01údatetime64[ns]©Údtypez8Casting to unit-less dtype 'datetime64' is not supported©ÚmatchÚ
|
datetime64)r Úto_frameÚpytestÚraisesÚ TypeErrorÚastypeÚnpr )ÚselfÚserZdfÚmsg© r" ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_astype.pyÚ test_astype_unitless_dt64_raises! s z.TestAstypeAPI.test_astype_unitless_dt64_raisesc C sP t dddg}d}tjt|d |jtjdd W 5 Q R X |jtjdd d S ) Né é é z_Expected value of kwarg 'errors' to be one of \['raise', 'ignore'\]\. Supplied value is 'False'r F©ÚerrorsÚraise)r r r Ú
|
ValueErrorr r Úfloat64Úint8)r r r! r" r" r# Ú test_arg_for_errors_in_astype0 s ÿz+TestAstypeAPI.test_arg_for_errors_in_astypeÚ dtype_classc C s4 t tddddd}|dti}| |¡}t dddd d
|
gdd}t ||¡ |dd i}| |¡}t d d dddgd dd}t ||¡ |ttd}d}tjt|d | |¡ W 5 Q R X |dti} tjt|d | | ¡ W 5 Q R X |t kr|i t d}
|
n|i }
|
tjt|d | |
|
¡ W 5 Q R X d S )Nr é
|
r& Úabc©ÚnameÚ0Ú2Ú4Ú6Ú8r, g g @g @g @g @©r r3 )r1 ÚdefzGOnly the Series name can be used for the key in Series dtype mappings\.r r )
|
r ÚrangeÚstrr ÚtmÚassert_series_equalr r ÚKeyErrorÚobject) r r/ r Zdt1ÚresultÚexpectedZdt2Zdt3r! Zdt4Zdt5r" r" r# Útest_astype_dict_like= s,
|
|
ÿ
|
z#TestAstypeAPI.test_astype_dict_likeN) Ú__name__Ú
|
__module__Ú __qualname__r$ r. r ÚmarkÚ parametrizeÚdictr rC r" r" r" r# r s r c @ s e Zd Zdd Zej dejd ¡dd Z ej de
|
ej g¡ej de e jd e d ¡e d
|
¡e d ¡ge e jd e d ¡e d
|
¡ejd gg¡d d Zdd Zdd Zej dejejg¡dd Zdd Zdd Zdd Zdd Zd d Z d d Z d!d" Z d#d$ Z ej d%d&d'd(gd)fej d&d'd(gd*e! "d+¡d,d&d'd(gd-fd.e#d/d0d1g d2fd.e$d3d4g d2fg¡ej d5d6d7g¡d8d9 Z%ej dej&ej'ej(g¡d:d; Z)ej d<d=ejd>fe*d?fg¡d@dA Z+ej ddBdCdDdEg¡dFdG Z,ej dHejej-g¡ej dej.ej/g¡dIdJ Z0ej de1ej2ej/g¡dKdL Z3dMdN Z4dOdP Z5dQdR Z6dSdT Z7dUdV Z8ej dej9ej: ¡dWdX Z;dYdZ Z<d2S )[Ú
|
TestAstypec C s\ t ddd}| d¡}t||gtd}| d¡}t| d¡| d¡gdd}t ||¡ d S )Nz2016-01-04 05:06:07ú
|
US/Pacific©Útzz
|
Asia/Tokyor z datetime64[ns, Europe/Brussels]zEurope/Brussels)r Ú
|
tz_convertr r@ r r= r> )r ÚtsZts2r ÚresrB r" r" r# Ú"test_astype_mixed_object_to_dt64tzd s
|
|
þz-TestAstype.test_astype_mixed_object_to_dt64tzr ZAllc C s2 |dkr.t g |d}t g |¡}t ||¡ d S )N)ÚSÚVÚMÚmr )r r r= r> )r r Z
|
init_emptyZ as_type_emptyr" r" r# Ú&test_astype_empty_constructor_equalityr s z1TestAstype.test_astype_empty_constructor_equalityÚseriesr0 é? é@ éè g ð?c C s$ | |¡}| t¡}t ||¡ d S ©N)r Úmapr< r= r> )r r rW rA rB r" r" r# Útest_astype_str_map s
|
|
|
z TestAstype.test_astype_str_mapc C s0 t tjg d¡}t tgdd}t ||¡ d S )Nz period[D]r )r r Únanr r r= r> ©r rA rB r" r" r# Útest_astype_float_to_period s z&TestAstype.test_astype_float_to_periodc C s. t ddgdd}| |jj¡}t ||¡ d S )Nr% r& Úint64r )r r Úarrayr r= r> )r r rA r" r" r# Útest_astype_no_pandas_dtype s z&TestAstype.test_astype_no_pandas_dtypec C sv dg}t |}t |¡jdkr8tjjdd}|j |¡ d|j d|j d}tj
|
t |d | |¡ W 5 Q R X d S ) Nr% )Ú timedelta64r z GH#33890 Is assigned ns unit©ÚreasonzThe 'z&' dtype has no unit\. Please pass in 'z\[ns\]' instead.r ) r r r r3 r rG ÚxfailÚnodeÚ
|
add_markerrD r r+ r )r r ÚrequestÚdatar rG r! r" r" r# Ú*test_astype_generic_timestamp_no_frequency s ÿz5TestAstype.test_astype_generic_timestamp_no_frequencyc C s< t ddd}t| t¡}tdddgtd}t ||¡ d S )Nú
|
2012-01-01r' ©Úperiodsz
|
2012-01-02z
|
2012-01-03r ©r r r r< r@ r= r> )r ZdtirA rB r" r" r# Útest_astype_dt64_to_str¬ s z"TestAstype.test_astype_dt64_to_strc C s> t dddd}t| t¡}tdddgtd}t ||¡ d S ) Nrm r' ú
|
US/Eastern©ro rM ú2012-01-01 00:00:00-05:00z2012-01-02 00:00:00-05:00z2012-01-03 00:00:00-05:00r rp )r Zdti_tzrA rB r" r" r# Útest_astype_dt64tz_to_str³ s ýúz$TestAstype.test_astype_dt64tz_to_strc C s¨ t tdtdd}| d¡}|jtjks,tt tdddddg}| d¡}|jtjks\tt d d
|
td D }tj |d< |jdkst| d¡}|jtjks¤td S ) NzM8[ns]é )r ÚindexÚOéÑ r% r& r c S s g | ]}t d ddddqS )ry r% r& r )r ©Ú.0Úir" r" r# Ú
|
<listcomp>Ì s z3TestAstype.test_astype_datetime.<locals>.<listcomp>r' )
|
r r r; r r r Úobject_ÚAssertionErrorr r^ ©r r r" r" r# Útest_astype_datetimeÁ s
|
|
|
|
z TestAstype.test_astype_datetimec C s t tdddd}| t¡}t | t¡td}t ||¡ t |jj d¡j |jj
|
¡}t ||¡ t | t¡}| t¡}t ||¡ d}t j t |d t |j d ¡ W 5 Q R X t j t |d t |j |j¡ W 5 Q R X | d
|
¡}t td dd d}t ||¡ d S ) NÚ20130101r' rr rs r ÚUTCz1Cannot use .astype to convert from timezone-naiver údatetime64[ns, US/Eastern]zdatetime64[ns, CET]z20130101 06:00:00ZCET)r r r r@ r= r> ÚvaluesÚdtZ tz_localizerN rM r r r r )r r rA rB r! r" r" r# Útest_astype_datetime64tzÔ s"
|
|
|
z#TestAstype.test_astype_datetime64tzc C sd t tdg}| t¡}t dg}t ||¡ t tdddg}| t¡}t dg}t ||¡ d S )Nz2010-01-04 00:00:00z
|
2010-01-04rr rL z2010-01-04 00:00:00-05:00)r r r r< r= r> )r rO rP rB r" r" r# Útest_astype_str_cast_dt64ò s
|
|
|
|
z$TestAstype.test_astype_str_cast_dt64c C s6 t tdddg}| t¡}t dg}t ||¡ d S )Nr% Úd)Úunitz1 days)r r r r< r= r> )r Útdr rB r" r" r# Útest_astype_str_cast_td64 s
|
|
z$TestAstype.test_astype_str_cast_td64c C sB t tddd}| t¡}t|jd ts.t|jt j
|
ks>td S )Nr r' rn r ) r r r r@ Ú
|
isinstanceÚilocr r r r r~ )r Zdt64serrA r" r" r# Ú test_dt64_series_astype_object s
|
z)TestAstype.test_dt64_series_astype_objectc C sD t dddgdd}| t¡}t|jd ts0t|jtj ks@td S )Nú59 Daysr útimedelta64[ns]r r )
|
r r r@ r r r r r r r~ )r ZtdserrA r" r" r# Ú test_td64_series_astype_object s
|
z)TestAstype.test_td64_series_astype_objectú data, dtypeÚxÚyÚzzstring[python]zstring[pyarrow]Zpyarrow)ZmarksÚcategoryr' z
|
2020-01-01r rL Nr r% r) r* Úignorec C sh t ||d}|dkr4|}|jtdd}t ||¡ n0d}tjttf|d |jt|d W 5 Q R X d S )Nr r r( z!(Cannot cast)|(could not convert)r ) r r Úfloatr= r> r r r+ r )r rk r r) r rB rA r! r" r" r# Ú/test_astype_ignores_errors_for_extension_dtypes s z:TestAstype.test_astype_ignores_errors_for_extension_dtypesc C s2 t dg|d}| t¡}t dg}t ||¡ d S )Ng¹?r z0.1)r r r< r= r> ©r r r rA rB r" r" r# Ú test_astype_from_float_to_str0 s
|
|
z(TestAstype.test_astype_from_float_to_strzvalue, string_value)NÚNoner^ z<NA>c C s> t dd|gtd}| t¡}t dd|gtd}t ||¡ d S )NÚaÚbr )r r@ r r< r= r> )r ÚvalueZ string_valuer rA rB r" r" r# Ú test_astype_to_str_preserves_na8 s
|
|
z*TestAstype.test_astype_to_str_preserves_naÚfloat32r, ra Úint32c C s@ t tj d¡dd}| |¡}|j|ks,t|j|jks<td S )Nrv Úfoor2 )r r ÚrandomZrandnr r r r3 )r r r Zas_typedr" r" r# Ú test_astypeG s
|
zTestAstype.test_astyper c C s6 d}t |g}tjt|d | |¡ W 5 Q R X d S )Nz9Cannot convert non-finite values \(NA or inf\) to integerr ©r r r r+ r )r r r r! r r" r" r# Ú test_astype_cast_nan_inf_intO s
|
z'TestAstype.test_astype_cast_nan_inf_intc C s< t ddddg}d}tjt|d | |¡ W 5 Q R X d S )NZcarZhouseÚtreeÚ1z/invalid literal for int\(\) with base 10: 'car'r r§ )r r Úarrr! r" r" r# Ú test_astype_cast_object_int_failY s z+TestAstype.test_astype_cast_object_int_failc C s¾ t d¡ |¡d }t|}d}tjt|d | |¡ W 5 Q R X tjt|d | ¡ |¡ W 5 Q R X tjt|d t| |¡ W 5 Q R X tjt|d |j |¡ W 5 Q R X d S )Nrv r' z$Cannot losslessly cast from .* to .*r )
|
r Úaranger r r r r+ r r
|
rb )r Zfloat_numpy_dtypeZ any_unsigned_int_numpy_dtyper« r r! r" r" r# Ú)test_astype_float_to_uint_negatives_raise` s z4TestAstype.test_astype_float_to_uint_negatives_raisec C s: t ddddgtd}| t¡}t |t t dd¡¡ d S )Nrª r5 Ú3r6 r r% rv )r r@ r Úintr= r> r r )r r« rA r" r" r# Útest_astype_cast_object_intv s
|
z&TestAstype.test_astype_cast_object_intc
|
C sÈ t j}t|d t d¡t d¡t d¡gtdgg}d }t ¡ dkrxd}t| ¡ g}| d¡}t|g}t ||¡ |D ]$}| d¡}|
|
t ¡} t || ¡ q||d k rÄ|dkrÄt t t |¡ d S ) Nr0 rX rY rZ u< ãã¼ã¿ã¼ãµã¤ã¨ã³ã¹ããåã¯ããæ»ãã§ããzutf-8u éèé£ã¹ãªãã¨ãã°ãÚunicode)ÚstringÚdigitsr r= ÚrandsÚsysÚgetdefaultencodingÚencoder r> r\ r< r Zsetdefaultencoding)
|
r r´ Z test_seriesZformer_encodingÚitemr rA rB rP Zexpecr" r" r# Útest_astype_unicode| s$ $þ
|
|
|
|
z TestAstype.test_astype_unicodec C s, t dddg t¡}|jt d¡ks(td S )Nr¤ ÚbarZbazZS3)r r ÚbytesÚdtypesr r r )r rA r" r" r# Útest_astype_bytes s z TestAstype.test_astype_bytesc C s4 t tjdd}| d¡}t ddd}t ||¡ d S )Nr@ r ÚboolT)r r r^ r r= r> ©r r rA rB r" r" r# Útest_astype_nan_to_bool s
|
z"TestAstype.test_astype_nan_to_boolc C sZ t dddg|d}| tdd¡}t tdddtdddtd ddd
|
}t ||¡ d S ) Né r é r rK rL z#1969-12-31 16:00:00.000000004-08:00z#1969-12-31 16:00:00.000000000-08:00z#1969-12-31 16:00:00.000000009-08:00)r r% r& )r r r r r= r> r r" r" r# Ú!test_astype_ea_to_datetimetzdtype¦ s
|
|
|
ýÿz,TestAstype.test_astype_ea_to_datetimetzdtypec C s< t ddddg}d|jd< | |¡j}|j}t ||¡ d S )Nr r% r& r' ZMichiganZLocation)r Úattrsr r= Zassert_dict_equal)r Zany_numpy_dtyper rA rB r" r" r# Útest_astype_retain_Attrs¹ s
|
|
z#TestAstype.test_astype_retain_Attrs)=rD rE rF rQ r rG rH r Z typecodesrV r< Zstr_r r³ r´ r= rµ r^ r] r` rc r rd rl rq ru r r r r r r Úparamr Z
|
skip_if_nor r r Úfloat16r¢ r, r r r¡ r¦ Úinfr£ ra r¨ r° r- r¬ r® r± rº r¾ rÁ ZALL_INT_EA_DTYPESZFLOAT_EA_DTYPESrÄ rÆ r" r" r" r# rJ c s
|
&$þþ
|
ý ÷þ
|
ýþ
|
|
|
|
þ
|
rJ c @ s e Zd Zej ddegdfdegdfddgdfddegdfdegd fd
|
d gd fd dgdfd degdfddegdfg ¡dd Zd S )ÚTestAstypeStringr TÚbooleanÚAr z
|
2020-10-10r rt r r% NZUInt16z1/1/2021z2/1/2021z period[M]z1 Dayr r c C s` |dks|dkr2t |kr2tjjdd}|j |¡ t||d}| |¡ |j¡}t
|
||¡ d S )NrË r z6TODO StringArray.astype() with missing values #GH40566re r ) r r rG rg rh ri r r r r= r> )r rk r rj Znullable_string_dtyperG r rA r" r" r# Ú/test_astype_string_to_extension_dtype_roundtripÅ s ÿ z@TestAstypeString.test_astype_string_to_extension_dtype_roundtrip) rD rE rF r rG rH r r rÍ r" r" r" r# rÊ Ä s
|
|
|
þ
|
|
ôþrÊ c @ s¬ e Zd Zdd Zdd Zdd Zej ddd g¡ej d
|
d d g¡ej d d d g¡dd Z dd Z
|
dd Z ej dddddgddddgg¡dd Z d d Z d d ZdS )!ÚTestAstypeCategoricalc
|
C sú t dd tdddD }ttj d¡ ddd¡ ¡ }t|tdddd|d }|}t
|
| d
|
¡|¡ t
|
| t ¡|¡ d }t jt|d | d ¡ W 5 Q R X tt ddddddddg}tddddddddg}t
|
| d¡|¡ tt ddddg}tddddg d¡}t
|
| d¡|¡ dd }tt |j¡d d }|| d ¡| || tj¡| t t |¡t |j¡¡ t
|
| d
|
¡|¡ t
|
| t ¡|¡ |j |jj ¡ ¡j ¡ } | d ¡ d
|
¡}
|
t
|
|
|
| ¡ | d ¡ t ¡}
|
t
|
|
|
| ¡ d S ) Nc S s g | ]}| d |d qS ©z - ió r" rz r" r" r# r} ë s zJTestAstypeCategorical.test_astype_categorical_to_other.<locals>.<listcomp>r é' éô éd é) F©ÚrightÚlabelsr z#Cannot cast object dtype to float64r r, r r Úcr< rª r5 r¯ r6 r% r& r' r r° c S s( t t t | ¡¡t t |¡¡¡ d S r[ )r= Úassert_almost_equalr ÚsortÚunique)r r r" r" r# Úcmpÿ s zCTestAstypeCategorical.test_astype_categorical_to_other.<locals>.cmpZ value_groupr2 r@ )r r; r r r¥ Z RandomStateÚrandintÚ sort_valuesr r= r> r r r r r+ rb r
r~ rØ ÚcatZset_categoriesÚ
|
categoriesZremove_unused_categories) r rÞ r rB r! ÚexpÚs2Zexp2rÛ Zroundtrip_expectedrA r" r" r# Ú test_astype_categorical_to_otherê s:
|
ÿ
|
z6TestAstypeCategorical.test_astype_categorical_to_otherc C s t dd tdddD }ttj ddd¡ ¡ }t|tdddd|d }d
|
}tj t
|
|d | t ¡ W 5 Q R X tj t
|
|d | d ¡ t ¡ W 5 Q R X d S ) Nc S s g | ]}| d |d qS rÏ r" rz r" r" r# r} s zUTestAstypeCategorical.test_astype_categorical_invalid_conversions.<locals>.<listcomp>r rÐ rÑ rÒ rÓ FrÔ zKdtype '<class 'pandas.core.arrays.categorical.Categorical'>' not understoodr r@ ) r r; r r r¥ rÜ rÝ r r r r r )r rÞ r r! r" r" r# Ú+test_astype_categorical_invalid_conversions s ÿzATestAstypeCategorical.test_astype_categorical_invalid_conversionsc C sÞ t dddg}| tddgdd¡}t tdddgdd}t ||¡ | tddgdd¡}t tdddgdd}t ||¡ | tdddgdd¡}t tdddgdddgdd}t ||¡ t |jjt dddg¡ d S )Nr r T©ÚorderedFr× ©rß rå )
|
r r r r r= r> Zassert_index_equalrÞ rß r
|
rÀ r" r" r# Ú test_astype_categoricaldtype# s ÿ z2TestAstypeCategorical.test_astype_categoricaldtyper3 Nr¤ Ú dtype_orderedTFÚseries_orderedc C s¸ t d}tt d|d}t|||d}t|d}| |¡}t|j|} t||| d}
|
t ||
|
¡ tt d|}| |¡}t|||d}
|
t ||
|
¡ |dkr´|}
|
| d¡}t ||
|
¡ d S ) NZ abcaacbabZbacrä r9 )r3 r ZadcFr )Úlistr r r rß r= r> ) r r3 rè ré Zs_dataZs_dtyper r rA Z exp_dtyperB r" r" r# Ú&test_astype_categorical_to_categorical4 s
|
|
|
|
z<TestAstypeCategorical.test_astype_categorical_to_categoricalc C s` t ddtjg}|jtjks t| tddgd¡}t tddtjgddgd}t
|
||¡ d S )NTF)rß ) r r r^ r½ r~ r r r r r= r> rÀ r" r" r# Ú'test_astype_bool_missing_to_categoricalR s
|
z=TestAstypeCategorical.test_astype_bool_missing_to_categoricalc C s@ t dddg}tjtdd |jdddgdd W 5 Q R X d S )Nr r zgot an unexpectedr r Træ )r r r r r r r" r" r# Ú test_astype_categories_raises[ s z3TestAstypeCategorical.test_astype_categories_raisesÚitemsr r r× r% r& r' c C s. t |}t t|}| d¡}t ||¡ d S )Nr )r r r r= r> )r rî r rà rP r" r" r# Ú test_astype_from_categoricala s
|
z2TestAstypeCategorical.test_astype_from_categoricalc C s~ ddddg}t |}t t|dd}| td dd¡}t ||¡ t t|tddd}| ttddd¡}t ||¡ d S )Nr r r× Trä Zabcdefræ )r r r r r= r> rê )r Úlstr rà rP r" r" r# Ú*test_astype_from_categorical_with_keywordsh s z@TestAstypeCategorical.test_astype_from_categorical_with_keywordsc C s: t tdtjgdd}t tdtgdd}t ||¡ d S )Nr% r r )r r r r^ r r= r> r_ r" r" r# Ú#test_astype_timedelta64_with_np_nant s z9TestAstypeCategorical.test_astype_timedelta64_with_np_nan)rD rE rF râ rã rç r rG rH rë rì rí rï rñ rò r" r" r" r# rÎ é s *
|
rÎ )#r r Ú importlibr r³ r¶ Únumpyr r Zpandas._libs.tslibsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr Zpandasr r r r r
|
r r r r r r r Zpandas._testingZ_testingr= r rJ rÊ rÎ r" r" r" r# Ú<module> s 8 C c%
|
