3
���h�L � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z m Z d dl
Zd dl
mZmZmZmZmZmZ d dljZG dd� d�ZdS )� )�deepcopyN)�PY37)�
async_mark�
skip_if_no)� Categorical� DataFrame�Series�compat�
date_range�timedelta_rangec @ s� e Zd Zejjdddg�dd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z!d<d=� Z"d>d?� Z#d@dA� Z$dBdC� Z%dDdE� Z&dFdG� Z'dHdI� Z(dJdK� Z)dLdM� Z*dNdO� Z+dPdQ� Z,dRdS� Z-e.� dTdU� �Z/dVdW� Z0dXdY� Z1e2dZ�d[d\� �Z3d]S )^�TestDataFrameMisc�attr�index�columnsc C s< t ||�}d |_|j� }dt ||�_t ||�jd ks8t�d S )N�foo)�getattr�name�copy�AssertionError)�self� float_framer
�ind�cp� r �=/tmp/pip-build-5_djhm0z/pandas/pandas/tests/frame/test_api.py� test_copy_index_name_checking s
z/TestDataFrameMisc.test_copy_index_name_checkingc C sn |d }|j dkst�|jd�}|j dks.t�|jd d �df }|j dksNt�|jd d � }|j dksjt�d S )N�A�B)r r �pop�loc)r r �s�s2r r r � test_getitem_pop_assign_name s
z.TestDataFrameMisc.test_getitem_pop_assign_namec C sH xB|j D ]8}x2|jD ](}|j||�}|| | }tj||� qW qW d S )N)r r Z
_get_value�tm�assert_almost_equal)r r �idx�col�result�expectedr r r �test_get_value) s
z TestDataFrameMisc.test_get_valuec C s� |j d�}tjdd� |jD ��}tj|j|� |jd�}tjdd� |jD ��}tj|j|� |j d�}tjdd� |jD ��}tj|j|� |jd�}tjdd� |jD ��}tj|j|� d S ) Nzfoo#c S s g | ]}d |� ��qS )zfoo#r )�.0�cr r r �
<listcomp>2 s z<TestDataFrameMisc.test_add_prefix_suffix.<locals>.<listcomp>z#fooc S s g | ]}|� d ��qS )z#foor )r* r+ r r r r, 6 s �%c S s g | ]}d |� ��qS )r- r )r* r+ r r r r, : s c S s g | ]}|� d ��qS )r- r )r* r+ r r r r, > s )Z
add_prefix�pd�Indexr r# Zassert_index_equalZ
add_suffix)r r Z with_prefixr( Z with_suffixZwith_pct_prefixZwith_pct_suffixr r r �test_add_prefix_suffix0 s
z(TestDataFrameMisc.test_add_prefix_suffixc C st |}|j d�dkst�|j d�dks(t�|j d�dks:t�|j d�dksLt�|j d�dks^t�|jd�dkspt�|jd�dks�t�|jd�dks�t�|jd�dks�t�|jd�dks�t�|jd�|jks�t�|jd�|jks�t�tjtdd�� |j d� W d Q R X tjtd d�� |jd
� W d Q R X tjtd d�� |jd � W d Q R X tjtdd�� |j d � W d Q R X d S ) Nr � r Zrowsr z
No axis named)�match� z No axis.*foor z
No axis.*None) Z_get_axis_numberr Z_get_axis_nameZ _get_axisr r �pytest�raises�
ValueError)r r �fr r r �
test_get_axisA s* z TestDataFrameMisc.test_get_axisc C s |j }|� |jkst�d S )N)�keysr r )r r Zgetkeysr r r � test_keys^ s zTestDataFrameMisc.test_keysc C s( t jtdd�� |j|k W d Q R X d S )Nzunhashable type: 'Index')r2 )r4 r5 � TypeErrorr )r r r r r �test_column_contains_raisesb s z-TestDataFrameMisc.test_column_contains_raisesc C s� t jtd�td�gtd�d�}x td�D ]}|t|�ks(t�q(W t|jd�t j�sVt�t jtd�td�gt jj tt
dd���d�}x td�D ]}|t|�ks�t�q�W x td�D ]}|t|�ks�t�q�W t|jd�t j�s�t�d S )NZabcdZefgh�ABCD)r r ZEFGH) r. r �list�dirr �
isinstance� __getitem__r Z
MultiIndex� from_tuples�zip)r �df�keyr r r �test_tab_completionf s z%TestDataFrameMisc.test_tab_completionc C s\ t � }t dg�}d}tjt|d�� t|� W d Q R X tjt|d�� t|� W d Q R X d S )Nr1 z;'DataFrame' objects are mutable, thus they cannot be hashed)r2 )r r4 r5 r; �hash)r � empty_framerD �msgr r r �test_not_hashabley s
z#TestDataFrameMisc.test_not_hashablec C s6 d}t |g i�}|t|�ks t�|jd |ks2t�d S )Nu �r )r r? r r )r ZcolnamerD r r r �+test_column_name_contains_unicode_surrogate� s z=TestDataFrameMisc.test_column_name_contains_unicode_surrogatec C s@ t tjjdd��}t tjjdd��}d|j_|jjd ks<t�d S )Nr � r )r �np�random�randnr r r )r Zdf1Zdf2r r r �test_new_empty_index� s z&TestDataFrameMisc.test_new_empty_indexc C sl t jdd�� t j|�}W d Q R X t|t|��s4t�|j|jksDt�|j|jksTt�tj ||j
t j�� d S )N�ignore)�all) rM Zerrstate�sqrtr@ �typer r r r# �assert_frame_equal�apply)r r r' r r r �test_array_interface� s z&TestDataFrameMisc.test_array_interfacec C s\ |j d�}||jkst�|j d�}||jks0t�d}tjt|d�� |j d� W d Q R X d S )Nr r1 z Axis must be 0 or 1 \(got 2\))r2 r3 )Z
_get_agg_axisr r r r4 r5 r6 )r r �colsr% rI r r r �test_get_agg_axis� s
z#TestDataFrameMisc.test_get_agg_axisc C sb t � }|jst�|j s t�|j s(t�t dddgdddgd�tjd�d �}|d
= |j s^t�d S ) Ng �?g @g @�a�br+ )r r rL )r r )r �emptyr rM �arange)r r �float_string_framerH rD r r r � test_nonzero� s
$z TestDataFrameMisc.test_nonzeroc C sL t dddgdddggdddgd �}x$|j� D ]\}}t|t j�s,t�q,W d S )
Nr1 r3 rL � � � rZ r[ )r )r �itemsr@ �_constructor_slicedr )r rD �k�vr r r �test_iteritems� s "z TestDataFrameMisc.test_iteritemsc C sx dddg}t dddgddd gg|d
�}xLt||j� �D ]:\}\}}||ksNt�t|t�s\t�|| |kj� s6t�q6W d S ) NrZ r[ r+ r1 r3 rL r` ra rb )r )r rC rc r r@ r rR )r rX rD r+ re rf r r r �
test_items� s
z TestDataFrameMisc.test_itemsc C s t jt|�|j�st�d S )N)r# Z
equalContentsr> r r )r r r r r � test_iter� s zTestDataFrameMisc.test_iterc C s\ x*|j � D ] \}}|j| }tj||� q
W x*|j � D ] \}}|j| }tj||� q6W d S )N)�iterrowsr r# �assert_series_equal)r r r^ re rf �expr r r �
test_iterrows� s
z TestDataFrameMisc.test_iterrowsc C sP t ddddgtdddd�d ��}x*|j� D ] \}}|j| }tj||� q*W d S )
NZM1701ZM1802ZM1903ZM2004z
2000-01-01r` �M)�periods�freq)Z non_iso8601Ziso8601)r r
rj r r# rk )r r re rf rl r r r �test_iterrows_iso8601� s
z'TestDataFrameMisc.test_iterrows_iso8601c C s� t tjddd�gd gd gdgg gt� gi gd��}ttjddd�d d dg t� i gtd�ddd�}t|j� �\}}tj||� d S ) Ni� r1 � )rZ r[ r+ �d�er7 �gZabcdefgr �object)r r �dtype) r �datetime�setr r> �nextrj r# rk )r rD r( �_r' r r r �test_iterrows_corner� s z&TestDataFrameMisc.test_iterrows_cornerc C s� xZt |j� �D ]J\}}tj|dd � �}|d |_|j|d d �f jdd�}tj||� qW tt j
j d�t d�d�ddgd �}x&|jd
d �D ]}t
|d t�s�t�q�W tdd d
gdddgd�d�}|ddg }t|j� �d$d%d&gks�t�tj� � p�tj� �stt|jd d���dk�st�t|jdd��}|jd'k�s:t�|j|j|jf|k�sTt�t|�jdk�sht�ddg|_ t|jdd��}|d(k�s�t�|jd)k�s�t�td d � t d �D ��} t| j� �}
t
|
t ��s�t�t �r�t |
d ��s�t�nt |
d � �s�t�td d � t d!�D �g�} t| jd
d ��} t
| t ��s4t�t | d ��sDt�td"d � t d#�D �g�}
t|
jd
d ��}t
|t ��s|t�t �r�t |d ��s�t�nt |d � �s�t�d S )*Nr1 r T)�dropra )�floats�intsr~ r )r F)r r3 rL r` rb )rZ r[ )�datarZ )r z![(0, 1, 4), (1, 2, 5), (2, 3, 6)]ZTestNamer/ r[ �def�return�_1�_2c S s i | ]}|gd t |� �qS )r7 )�str)r* �ir r r �
<dictcomp> s z5TestDataFrameMisc.test_itertuples.<locals>.<dictcomp>i �_fieldsc S s i | ]}d |� �d|� ��qS )�bar_�foo_r )r* r� r r r r� s � c S s i | ]}d |� �d|� ��qS )r� r� r )r* r� r r r r� s � )r r1 r1 )r1 r3 r3 )r3 rL rL )r/ rZ r[ )r r1 r` )r/ r� r� ) � enumerate�
itertuplesr rd r Ziloc� reset_indexr# rk rM rN rO �ranger@ �intr r> r Zis_platform_windowsZis_platform_32bit�reprrz r� r/ rZ r[ rT �__name__r �tupler �hasattr)r r r� �tupr r( rD ZdfaaZtup2Zdf3Ztup3Zdf_254_columnsZresult_254_columnsZdf_255_columnsZresult_255_columnsr r r �test_itertuples� sN
z!TestDataFrameMisc.test_itertuplesc C s� t ddddddgdddddd gd
��}t|d �|d < t|jj�}tj|jj�j� }tj ||� x |j
d
d�D ] }t |� qnW x |j � D ]\}}t |� q�W x |j
� D ]\}}t |� q�W d S )Nr1 r3 rL r` ra rb rZ r[ rt )�id� raw_grader� �gradeF)r )r r r> r� �valuesrM �array�tolistr# r$ r� r� rj rc ) r rD r' r( �t�rowr r+ r& r r r �#test_sequence_like_with_categorical* s $ z5TestDataFrameMisc.test_sequence_like_with_categoricalc C s t |�t |j�kst�d S )N)�lenr r )r r r r r �test_lenB s zTestDataFrameMisc.test_lenc
C s� |}|j }|j}xht|�D ]\\}}xRt|�D ]F\}} || }
tj| �r^tj||
| �srt�q,| ||
| ks,t�q,W qW |ddg j }|d dks�t�tdddgdd d
gd ��} | j }|d dks�t�|dd g j }|jdd gd
�j } tj || � d S )Nr r r �bary �?y @y @r1 r3 rL )�complex�realr )r )r r )r r )
r� r r� rM �isnanr r Zreindexr# r$ )
r r r^ �frame�arrZ
frame_colsr� r� �j�valuer& rD r( r r r �test_values_mixed_dtypesE s"
z*TestDataFrameMisc.test_values_mixed_dtypesc C sF t jddgddgd��}tjddgddgg�}|j� }tj||� d S )Nr1 r3 rL g @)r r )r. r rM r� �to_numpyr# �assert_numpy_array_equal)r rD r( r' r r r �
test_to_numpy_ s z TestDataFrameMisc.test_to_numpyc C sN t jddgddgd��}tjddgddggdd�}|jdd�}tj||� d S ) Nr1 r3 rL g @)r r r` Zint64)rw )r. r rM r� r� r# r� )r rD r( r' r r r �test_to_numpy_dtypee s z%TestDataFrameMisc.test_to_numpy_dtypec C sX t jjdd�}tj|�}|jj|ks(t�|jdd�j|ks>t�|jdd�j|k sTt�d S )Nr` rL F)r T) rM rN rO r. r r� �baser r� )r r� rD r r r �test_to_numpy_copyk s
z$TestDataFrameMisc.test_to_numpy_copyc C sF t jt jd�dgg�}|jtd�}tjddggtd�}tj||� d S )Nz2020-01-01 00:00:00g Y@)rw z100.0) r. r � Timestampr� r� rM r� r# r� )r rD r' r( r r r � test_to_numpy_mixed_dtype_to_strr s z2TestDataFrameMisc.test_to_numpy_mixed_dtype_to_strc
C s� t tjjdd��}tj|j|jdd�� tj|j|jdd�� tj||jdd�� d}tj t
|d�� |jdd� W d Q R X d S )N�
ra r r1 z)No axis named 2 for object type DataFrame)r2 r3 ) r rM rN rO r# rU �TZswapaxesr4 r5 r6 )r rD rI r r r �
test_swapaxesy s z TestDataFrameMisc.test_swapaxesc C sP |}|j dd�}|j dd�}tj||� |j dd�}|j dd�}tj||� d S )Nr )Zaxisr r1 r )�sumr# rk )r r r7 r( r' r r r �test_axis_aliases� s z#TestDataFrameMisc.test_axis_aliasesc C s$ t jtj�st�t jtj�s t�d S )N)�pydocZgetdocr r r r )r r r r �test_class_axis� s z!TestDataFrameMisc.test_class_axisc C s" |j }|jd t|j�ks t�d S )Nr1 )r� �shaper� r r )r r^ r� r r r �test_more_values� s z"TestDataFrameMisc.test_more_valuesc C sD t dddgitjtjd�gddggd�}t|�}d}||ks@t�d S ) N�Xr1 r3 �20130101rZ r[ )r z/ X
NaT a 1
2013-01-01 b 2)r r. ZNaTr� r� r )r r^ rD r' r( r r r �test_repr_with_mi_nat� s
&z'TestDataFrameMisc.test_repr_with_mi_natc C s( x"|j � D ]\}}|j|ks
t�q
W d S )N)rc r r )r r^ re rf r r r �test_items_names� s z"TestDataFrameMisc.test_items_namesc C s. |j }x"|j� D ]\}}|j|kst�qW d S )N)Z_seriesrc r r )r r^ �seriesre rf r r r �test_series_put_names� s z'TestDataFrameMisc.test_series_put_namesc C s� t dddg�}|j st�t dgdgd�}|j s6t�t ddgddgd�j� }|jsXt�|jjsdt�t � t dgd �t dgd
�t dg i�g}x"|D ]}|js�t�|jjs�t�q�W d S ) Nr1 r3 rL )r r rZ r[ r+ rs )r )r )r r\ r Zdropnar� )r rD Z empty_framesr r r �test_empty_nonzero� s
z$TestDataFrameMisc.test_empty_nonzeroc C sP t tddd�tddd�d��}|j}|jj� }ttjd�di�}t j
||� d S )Nr� r� )ro z1 day)r r rv ) r r
r r� ZdtypesZ value_countsr rM rw r# rk )r rD r� r' r( r r r �test_with_datetimelikes� s
z)TestDataFrameMisc.test_with_datetimelikesc C s4 d|j d d �df< |j d d �df dkj� s0t�d S )Ng @r ra )r� rR r )r r r r r � test_values� s z TestDataFrameMisc.test_valuesc C sJ t |�}|d }d|d d �< x(|j� D ] \}}|d | |ks&t�q&W d S )Nr r� )r rc r )r r r r� r% r� r r r �
test_deepcopy� s
z TestDataFrameMisc.test_deepcopyc C sP t ddddgddddgdddd gd
��}d d � }d
d� }||j� |� dd� }||jd�|� dd� }||j� |� dd� }||j� |� dd� }||j� |� dd� }||j� |� dd� }||j� |� dd� }||j� |� |j� d }dd� }||jd�d |� dd� }||j� |� dd� }||j� |� dd� }||j� |� d S ) Nr r� ZbazZquxr r1 r3 rL r` )rZ r[ r+ c S s || �}|d kst �d S )N)r )r� r7 r' r r r �_check_f� s z<TestDataFrameMisc.test_inplace_return_self.<locals>._check_fc S s | j ddd�S )NrZ T)�inplace)� set_index)�xr r r �<lambda>� s z<TestDataFrameMisc.test_inplace_return_self.<locals>.<lambda>c S s | j dd�S )NT)r� )r� )r� r r r r� � s rZ c S s | j dd�S )NT)r� )Zdrop_duplicates)r� r r r r� � s c S s | j ddd�S )Nr[ T)r� )Z sort_values)r� r r r r� � s c S s | j dd�S )NT)r� )Z
sort_index)r� r r r r� � s c S s | j ddd�S )Nr T)r� )�fillna)r� r r r r� � s c S s | j dddd�S )Nr1 r T)r� )�replace)r� r r r r� � s c S s | j ddidd�S )Nr1 r T)r� )�rename)r� r r r r� � s r+ c S s | j ddd�S )NT)r� r} )r� )r� r r r r� s c S s | j ddd�S )Nr T)r� )r� )r� r r r r� s c S s | j dddd�S )Nr1 r T)r� )r� )r� r r r r�
s c S s | j ddidd�S )Nr1 r T)r� )r� )r� r r r r� s )r r r� )r r� r� r7 rs r r r �test_inplace_return_self� s8 & z*TestDataFrameMisc.test_inplace_return_selfc � s� t jddd� ddlm} d}|j|�I d H dd l}|jdk rLtjd �}ntjt dd �}|�, |d
�� t
|j j d d �� W d Q R X W d Q R X d S )
NZIPythonz6.0.0)Z
minversionr )�provisionalcompleterz(import pandas as pd; df = pd.DataFrame()z0.17.0F)Zcheck_stacklevelrQ zdf.r1 )
r4 Z importorskipZIPython.core.completerr� Zrun_code�jedi� __version__r# Zassert_produces_warning�DeprecationWarningr> Z CompleterZ completions)r �ipr� �coder� �warningr r r �test_tab_complete_warning s
z+TestDataFrameMisc.test_tab_complete_warningc C sL t jdddgi�}|ji ks t�d|jd< |jtd�}|jddiksHt�d S )Nr r3 rL r1 �version)r )r. r �attrsr r� r� )r rD r' r r r �
test_attrs+ s
z TestDataFrameMisc.test_attrsc C s� t ddgi�}dg|d< |d |j� d|d jd< tj|t d gdgd��� dg|d< |d jd d
kspt�tj|t d gdgdgd��� d S ) NrZ r1 r r� )rZ r� �y)rZ r� r� ���r� r� r� )r r r� r# rU r )r rD r r r �test_cache_on_copy3 s
z$TestDataFrameMisc.test_cache_on_copyZjinja2c C sH t � }tj|� tjtdd�� |jtjd�j ddd�� W d Q R X d S )Nz Not supported for DataFrames!)r2 � rL )
r �inspect�
getmembersr4 r5 �NotImplementedErrorZ_constructor_expanddimrM r] Zreshape)r rD r r r �!test_constructor_expanddim_lookupE s
z3TestDataFrameMisc.test_constructor_expanddim_lookupN)4r� �
__module__� __qualname__r4 �markZ parametrizer r" r) r0 r8 r: r< rF rJ rK rP rW rY r_ rg rh ri rm rq r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r r� r r r r r sX
= @r )r r rx r� r� ZnumpyrM r4 Z
pandas.compatr Z pandas.util._test_decoratorsr r Zpandasr. r r r r r
r Zpandas._testingZ_testingr# r r r r r �<module> s
|