U
|
¬ý°d^ ã @ s d dl Zd dlZd dlmZ d dlZd dlmZ d dl
|
m Z ej deejejddg¡dd Zd d
|
Zd d Zd d Zej dddg¡dd ZdS )é N)Ú PeriodDtype)Ú period_arrayÚdtypeZuint32Zuint64c C sp t ddd gdd}t | ¡tjkrJtjtdd | | ¡ W 5 Q R X d S | | ¡}|j d¡}t
|
||¡ d S )NÚ2000Ú2001ÚD©ÚfreqzDo obj.astype\('int64'\)©ÚmatchÚi8) r Únpr Úint64ÚpytestÚraisesÚ TypeErrorÚastypeÚ_ndarrayÚviewÚtmÚassert_numpy_array_equal)r ÚarrÚresultÚexpected© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/period/test_astype.pyÚtest_astype_int s
|
r c C sh t ddd gdd} | jtjdd}|j| jks2t| jtjdd}|| jk sPtt || j d¡¡ d S ) Nr r r r F)ÚcopyTr )
|
r r r r Úbaser ÚAssertionErrorr r r )r r r r r Útest_astype_copies s r c C sX t dddd gdd} | d¡}tjddgdd}tjjddddg|d }t ||¡ d S )
|
Nr r r r Úcategoryr é éÿÿÿÿ)Ú
|
categories)r r ÚpdZ PeriodIndexZ CategoricalZ
|
from_codesr Zassert_categorical_equal)r r r$ r r r r Útest_astype_categorical( s
|
|
r& c C sB t ddd gdd} | td¡}t ddd gdd}t ||¡ d S )Nr r r r ÚM)r r r r Zassert_period_array_equal)r r r r r r Útest_astype_period0 s r( Úotherzdatetime64[ns]útimedelta64[ns]c C sv t ddd gdd}| dkrHtjt| d d
d | | ¡ W 5 Q R X n*| | ¡}t ddtjg¡j}t
|
||¡ d S )Nr r r r r* éüÿÿÿr
|
) r r r r r r% Z DatetimeIndexZNaTÚ_datar Zassert_datetime_array_equal)r) r r r r r r Útest_astype_datetime7 s
|
r- )Únumpyr r Zpandas.core.dtypes.dtypesr Zpandasr% Zpandas._testingZ_testingr Zpandas.core.arraysr ÚmarkZ parametrizeÚintÚint32r r r r& r( r- r r r r Ú<module> s
|
|
