3
���h91 � & @ sB d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlZ d dl
j Z d dl
mZ d dlmZ d dlmZmZmZmZmZmZmZ d dlmZmZm Z d dl m Z m Z m Z ej!j"dd d
gdej#d d
g�fd d
ge$eej%d d
ge$d ��fd d
gej&d �eej%d
dgej&d �d ��fej%d d
gdd �dej#d d
g�fd d
gd eej%d d
gd d ��fe j'dd�e j'dd�gde ddgdd�fe j'dd�ge j(d�e dgdd�fd d
gej&d�ej#ej%d d
gdd ��fej%d d
gdd �dej#ej%d d
gdd ��fe j)ddg�ej&d�ej#ddg�fe j)ddg�dej#ddg�fddgej&d�ej#ddg�fddge j*dd�ej#ddge j*dd�d �fddgej&d�ej#ddg�fe j+ddg�ej&d�ej#ddg�fe j+ddg�dej#ddg�fdd gd e j,dd g�fdd ge j-dd d �e j,dd gd d �fe j.d d
�e j.d d!�gd"ej/dRdSg�fd d gd#ed d gdd �fd dgd$ed dgd$d �fe j0d d
g�deej%d d
gej1d ��fddgd%ej#ddg�fddge j2� ej#ddg�fd dgd&ej#d dg�fd dge j3� ej#d dg�fe j4d d
g�deej%d d
gej1d ��fe j0e j,dd gdd d'gd(��de j,dd gdd d'gd(�fej5d �ej5d �gd)e d d g�fe ddgdd�d e j,e j'dd�e j'dd�g�fg �d*d+� �Z6d,d-� Z7ej8d�Z9ej!j"d.e j'dd�e j'dd�ge ddgdd�fe j.d d �e j.d d
�gej:d d d
g�fe j;d�e j;d�gej#ddg�fe j d/d d �e j d0d d �gej#ddg�fej%d d
gd1d �eej%d d
gd1d ��fej%d d
gd2d �eej%d3d/gd1d ��fe j;ddd�e j;ddd�gej#ddge j*dd�d �fe j d/d d e9d4�e j d0d d e9d4�gej#ddge9d�fe j<d�e j<d�gej#ddg�fej%d d
gd5d �eej%d d
gd5d ��fej%d d
gd6d �eej%d3d/gd5d ��fd d
gej#d d
g�fd dgej#d dg�fd e j=gej#d e j=g�fd ej>gej#d ej>g�fdd gej#dd g�fddgej#ddg�fd d7gej#d d7g�fd dgej#d dg�fg�d8d9� �Z?ej!j"d:e j'dd�e j'dd;�ge j.d d d<d=�e j.d d
d>d=�ge j;ddd�e j;dd?d�ge j;ddd�e j;d�gej%e j;d�e j;ddd�g�g�d@dA� �Z@ej!j"d:ej%d d
gd d!gg�d d
gd d!ggg�dBdC� �ZAdDdE� ZBeG dFdG� dGe ��ZCG dHdI� dIe �ZDdJdK� ZEejFdLdM� �ZGdNdO� ZHG dPdQ� dQ�ZIdS )T� N)�registry)�register_extension_dtype)� is_scalar)� BooleanArray�
DatetimeArray� IntegerArray�
IntervalArray� SparseArray� StringArray�TimedeltaArray)� PandasArray�
integer_array� period_array)� DecimalArray� DecimalDtype�
to_decimalzdata, dtype, expected� � )�dtypeZfloat32g �?g @�int64�2000�DZ2001z Period[D])�freqzdatetime64[ns]ZCET)�tzZ1HZ2Hztimedelta64[ns]�a�b�categoryT)�ordered� � �intervalz
Sparse[int64]ZInt16�string�boolean�c)�
categories�decimalc C s t j| |d�}tj||� d S )N)r )�pd�array�tm� assert_equal)�datar �expected�result� r- �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/arrays/test_array.py�
test_array s wr/ c C s� t jddg�} tj| | jd�}t j| |j�dks4t�tj| | jdd�}t j| |j�dks\t�tj| | jdd�}t j| |j�dks�t�d S )Nr r )r FT)r �copy)�npr' r& r Z
shares_memoryZ_ndarray�AssertionError)r r r- r- r. �test_array_copy� s r3 zdata, expectedi� i� zM8[ns]zM8[us]i� )�tzinfozm8[ns]zm8[us]Fc C s t j| �}tj||� d S )N)r& r' r( r) )r* r+ r, r- r- r. �test_array_inference� s D
r5 r* �A�left)�closed�right�UTCc C s, t j| �}ttj| td��}tj||� d S )N)r )r& r' r r1 �objectr( Z assert_extension_array_equal)r* r, r+ r- r- r. �test_array_inference_fails� s
r< c C s, t jtdd�� tj| dd� W d Q R X d S )Nz!PandasArray must be 1-dimensional)�matchr )r )�pytest�raises�
ValueErrorr& r' )r* r- r- r. �test_nd_raises s rA c C s( t jtdd�� tjd� W d Q R X d S )NzCannot pass scalar '1')r= r )r> r? r@ r& r' r- r- r- r. �test_scalar_raises
s rB c @ s e Zd ZdZedd� �ZdS )�
DecimalDtype2�decimal2c C s t S )zq
Return the array type associated with this dtype.
Returns
-------
type
)�
DecimalArray2)�clsr- r- r. �construct_array_type s z"DecimalDtype2.construct_array_typeN)�__name__�
__module__� __qualname__�name� classmethodrG r- r- r- r. rC s rC c s" e Zd Zed� fdd� �Z� ZS )rE NFc s, t |tjtjf�rtd��t� j|||d�S )Nz3scalars should not be of type pd.Series or pd.Index)r r0 )�
isinstancer& �Series�Index� TypeError�super�_from_sequence)rF Zscalarsr r0 )� __class__r- r. rR % s z DecimalArray2._from_sequence)NF)rH rI rJ rL rR �
__classcell__r- r- )rS r. rE $ s rE c C sj | }|t jd�t jd�g�}tjtdd�� tj|� W d Q R X tj|dd�}tj|j �}t
j ||� d S )N�1�2z3scalars should not be of type pd.Series or pd.Index)r= rD )r ) r% �Decimalr> r? rP rE rR r& r' �valuesr( r) )Zindex_or_seriesZboxr* r, r+ r- r- r. �test_array_unboxes- s rY c c s. t jjt�} t jj| � d V t jjt� d S )N)r Zdtypes�indexr �pop�append)�idxr- r- r. �registry_without_decimal< s r^ c C sN t jd�d kst�tjd�tjd�g}tj|td�}tj |�}t
j ||� d S )Nr% rU rV )r ) r �findr2 r% rW r& r' r r rR r( r) )r^ r* r, r+ r- r- r. �test_array_not_registeredD s
r` c @ s� e Zd Zdd� Zdd� Zdd� Zejjde j
dd d
d �e j d �ge j
dd d
d
d�e j d d
d�ge j ddd d�e j
d�gg�dd� �Zdd� ZdS )�TestArrayAnalyticsc C sd t jdddg|d�}|jddd�}t|�s.t�|dks:t�|jddd�}t|�sTt�|d ks`t�d S )
Nr r r# )r r7 )Zsider r9 r )r& r' � searchsortedr r2 )�selfZ string_dtype�arrr, r- r- r. �test_searchsortedO s z$TestArrayAnalytics.test_searchsortedc C sd t jdddg|d�}|jd�}t|�s*t�|dks6t�|jdg�}tjdgtjd�}tj||� d S )Nr r �Z )r � r ) r& r' rb r r2 r1 �intpr( �assert_numpy_array_equal)rc �any_real_dtyperd r, r+ r- r- r. �'test_searchsorted_numeric_dtypes_scalarZ s
z:TestArrayAnalytics.test_searchsorted_numeric_dtypes_scalarc C sF t jdddg|d�}|jddg�}tjddgtjd�}tj||� d S )Nr r rf )r r rg )r& r' rb r1 rh r( ri )rc rj rd r, r+ r- r- r. �'test_searchsorted_numeric_dtypes_vectord s z:TestArrayAnalytics.test_searchsorted_numeric_dtypes_vectorzarr, valZ20120101�
Z2D)�periodsr Z20120102zAsia/Hong_Kong)rn r r )r z1 dayz10 days)�start�endrn z2 daysc C s0 t j|�}|j|�}t|�s t�|dks,t�d S )Nr )r& r' rb r r2 )rc rd �valr, r- r- r. �$test_search_sorted_datetime64_scalarj s
z7TestArrayAnalytics.test_search_sorted_datetime64_scalarc C sP t jdddg|d�}|jddgtj|�d�}tjddgtjd�}tj||� d S )Nr r r )r r )Zsorter)r& r' rb r1 Zargsortrh r( ri )rc rj rd r, r+ r- r- r. �test_searchsorted_sorter� s z+TestArrayAnalytics.test_searchsorted_sorterN)rH rI rJ re rk rl r> �mark� parametrizer& Z
date_range� TimestampZtimedelta_range� Timedeltarr rs r- r- r- r. ra N s
ra )r r )r r )J�datetimer% Znumpyr1 r> ZpytzZpandas.core.dtypes.baser Zpandasr& Zpandas._testingZ_testingr( Zpandas.api.extensionsr Zpandas.api.typesr Z
pandas.arraysr r r r r r
r Zpandas.core.arraysr r
r Z pandas.tests.extension.decimalr r r rt ru rR r; r' r ZPeriodZ PeriodDtypeZ
DatetimeIndexZDatetimeTZDtypeZTimedeltaIndexZ CategoricalZCategoricalDtypeZInterval� from_tuplesrN r Z StringDtypeZ BooleanDtyperO rW r/ r3 �timezoneZcetZ from_breaksrv rw ZNA�nanr5 r< rA rB rC rE rY Zfixturer^ r` ra r- r- r- r. �<module> s
$
&& ,
&" * 8
|