3
���h� �
@ s� d Z ddlZddlZddlZddljZddl m
Z
ddl m Z ej
ejddged�ejddged�ejddged�ejddDged�ejd
d ged�ejddgd d�ejddgd
d�gd�dd� �ZejjddEdFdGdHdIdJdKdLdMdNg
�dd � �ZejjddOdPdQdRdSdTdUdVdWdXg
�d d � �Zd d � Zd!d"� Zd#d$� Zd%d&� Z d'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z!d1d2� Z"d3d4� Z#d5d6� Z$d7d8� Z%d9d:� Z&d;d<� Z'ejjd=deg�d>d?� �Z(d@dA� Z)dBdC� Z*dS )YzN
Additional tests for PandasArray that aren't covered by
the interface tests.
� N)� PandasArray)� PandasDtype�a�b)�dtype� y @TF�datetime64[ns]ztimedelta64[ns])�paramsc C s | j S )zk
Parametrized fixture for NumPy arrays with different dtypes.
This excludes string and bytes.
)�param)�request� r �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/arrays/test_numpy.py�any_numpy_array s r zdtype, expected�bool�int�uint�float�complex�str�bytes�object�voidc C s t | �} | j|kst�d S )N)r Z _is_numeric�AssertionError)r �expectedr r r
�test_is_numeric& s r c C s t | �} | j|kst�d S )N)r Z _is_booleanr )r r r r r
�test_is_boolean: s r c C s" t tjd��} t| �dks t�d S )N�int64zPandasDtype('int64'))r �npr �reprr )r r r r
� test_reprN s r c C s( t jd�} t tjd��}| |ks$t�d S )Nr )r Zconstruct_from_stringr r r )�resultr r r r
� test_constructor_from_stringS s
r! c C s, t jtdd�� tdddg� W d Q R X d S )Nz NumPy array)�matchr � � )�pytest�raises�
ValueErrorr r r r r
� test_constructor_no_coercion] s r( c C s4 t jdddg�} tjt| �dd�}|j| k s0t�d S )Nr r# r$ T)�copy)r �array�pd�Seriesr �valuesr )�ndarray�serr r r
�!test_series_constructor_with_copyb s r0 c C sF t jdddg�} tjt| �dd�}tjdddgdd�}tj||� d S ) Nr r# r$ Zfloat64)r g �?g @g @)r r* r+ r, r �tm�assert_series_equal)r. r r r r r
�#test_series_constructor_with_astypei s r3 c C sJ t jdddgdd�} tj| dd�}tt jdddgdd��}tj||� d S )Nr r# r$ r )r Zuint64)r r* r Z_from_sequencer1 � assert_extension_array_equal)�arrr r r r r
�test_from_sequence_dtypep s r6 c C s4 t jddg�} t| dd�}t j|j| �dks0t�d S )Nr r T)r) F)r r* r Z
shares_memory�_ndarrayr )r5 r r r r
�test_constructor_copyw s r8 c C s" | }t |�}|jj|jks t�d S )N)r r Z numpy_dtyper )r �nparrr5 r r r
�test_constructor_with_data~ s r: c C st t tjdddg��} | j� }|| jks*t�| jdd�}|| jk sDt�| jdd�}tjdddgdd�}tj||� d S )Nr r# r$ T)r) Zf8)r )r r r* �to_numpyr7 r r1 �assert_numpy_array_equal)r5 r r r r r
�
test_to_numpy� s r= c C s: t jdddg�} d| jd< t jdddg�}tj| |� d S )Nr r# r$ �
r )r+ r, r* r1 r2 )r/ r r r r
�test_setitem_series� s
r? c C s< | }t |dd�}|d |d< |d |d< tj|j� |� d S )NT)r) r r )r r1 r< r; )r r9 r5 r r r
� test_setitem� s
r@ c C sH t jdddgdd�} t| �} d}tjt|d�� | j|� W d Q R X d S )Nr r# r$ r )r z)cannot perform not_a_method with type int)r" )r r* r r% r& � TypeErrorZ_reduce)r5 �msgr r r
�test_bad_reduce_raises� s
rC c
C sB t tjdddg��} d}tjt|d�� | jdd� W d Q R X d S )Nr r# r$ z/the 'keepdims' parameter is not supported .*all)r" T)Zkeepdims)r r r* r% r& r' �all)r5 rB r r r
�$test_validate_reduction_keyword_args� s rE c C s� t tjdddg��} tj| �}t tj| j��}tj||� tj| tj| d��\}}tj| jtj| jd��\}}t |�}t |�}tj||� tj||� d S )Ng �?g r# g �) r r r* �absr7 r1 r4 �divmod�add)r5 r r Zr1Zr2Ze1�e2r r r
�
test_ufunc� s
rJ c C s@ t tjdddg��} | | }t tjdddg��}tj||� d S )Nr r# r$ � � )r r r* r1 r4 )�xr r r r r
�test_basic_binop� s rN r c C sH t tjdddg| d��}d|d< t tjdddg| d��}tj||� d S )Nr r �c)r �tr )r r r* r1 r4 )r r5 r r r r
� test_setitem_object_typecode� s rQ c C sf t tjdddg��} tjtdd�� d| d< W d Q R X d| d< t| d ttjf�sbt t
| d ���d S ) Nr r# r$ r )r" r r g @) r r r* r% r& r' �
isinstancer �integerr �type)r5 r r r
�test_setitem_no_coercion� s
rU c C s� t tjdddg��} | j� }| d d � }tj| �}d| d< |d dksJt�|d dksZt�|d dksjt�d| d< d|d < | d
dks�t�d S ) Nr r# r$ � r g @� ���rX rX )r r r* �viewZasarrayr )r5 Zview1Zview2Zview3r r r
� test_setitem_preserves_views� s
rZ y �? @)r T)r T)r T)r T)r T)r F)r F)r F)r F)r F)r T)r F)r F)r F)r F)r F)r F)r F)r F)r F)+�__doc__Znumpyr r% Zpandasr+ Zpandas._testingZ_testingr1 Z
pandas.arraysr Zpandas.core.arrays.numpy_r Zfixturer* r r r r r r �markZ parametrizer r r r! r( r0 r3 r6 r8 r: r= r? r@ rC rE rJ rN rQ rU rZ r r r r
�<module> sp
|