U
|
ý°d` ã @ sn d dl Z d dlm m mZ d dlZd dlmZ d dl m
|
Z dd Z e j dddg¡dd Zd
|
d ZdS ) é N)ÚSeriesc C sP t dddgdd} t g dd}t | j|jg¡}t | |g¡j}t ||¡ d S )NÚaÚbÚcÚcategory)Zdtype)r Ú_concatÚ concat_compatZ_valuesÚpdÚconcatÚtmZassert_categorical_equal)Zser1Zser2ÚresultÚexpected© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/dtypes/test_concat.pyÚ.test_concat_mismatched_categoricals_with_empty
|
s
|
r ÚcopyTFc C sB t dt jdddgi¡}| ¡ }t j|g| d}t ||¡ d S )NÚ timestampz 2020-04-08 09:00:00.709949+0000ÚUTC)Útz)r )r Z DataFrameÚ Timestampr r
|
r Zassert_frame_equal)r Zdfr r r r r Ú%test_concat_single_dataframe_tz_aware s ÿr c C s t jdddd} | j dd¡}tj|d d
|dd
gdd}t ||¡ tj|d d
d d
f |d d
dd
f gd d}t ||¡ d
|
}tj t
|
|d 8 tj|d d
d d
f |d d
dd
f gdd W 5 Q R X tj t
|
|d ( tj|d d
|dd
gd d W 5 Q R X d S ) Nz
|
2016-01-01é$ ÚD)ZperiodsÚfreqé é r )Zaxisé zNall the input array dimensions.* for the concatenation axis must match exactly)Úmatch) r Z period_rangeÚ_dataZreshaper r r Zassert_period_array_equalÚpytestZraisesÚ
|
ValueError)ÚpiZarrr Úmsgr r r Útest_concat_periodarray_2d s " 2 ÿ<r# )r Zpandas.core.dtypes.concatÚcoreZdtypesr
|
r Zpandasr r Zpandas._testingZ_testingr r ÚmarkZ parametrizer r# r r r r Ú<module> s
|
|
