Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/__pycache__/test_loc.cpython-37.pyc
Назад
B �5�g�u � @ s d Z ddlmZ ddlmZmZmZmZ ddlmZ ddl Z ddl m Z ddl Z ddlZddlm mZ ddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$ ddl%m&Z' ddl(m)Z) dd l*m+Z+ G d d � d e+�Z,G d d � d �Z-G dd� d�Z.G dd� d�Z/G dd� d�Z0G dd� d�Z1G dd� d�Z2G dd� de+�Z3G dd� d�Z4G d d � d �Z5ej6�7d d d d!gd d gdd"gfd d d#gd d gdd"gfd d d#gd d#gdd$gfg�d%d&� �Z8d'd(� Z9d)d*� Z:d+d,� Z;d-d.� Z<d/d0� Z=d1d2� Z>d3d4� Z?d5d6� Z@G d7d8� d8�ZAdS )9z$ test label based indexing with loc � )� namedtuple)�date�datetime�time� timedelta)�StringION)�gettz)� Categorical�CategoricalIndex� DataFrame� DatetimeIndex�Index� IndexSlice� MultiIndex�Period�Series� SparseDtype� Timedelta� Timestamp� date_range�timedelta_range� to_datetime� to_timedelta)� is_scalar)�Basec @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�TestLocc C s | j dddgtd� d S )N�loc� �labels)�typs�fails)� check_result�KeyError)�self� r$ �T/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/test_loc.py�test_loc_getitem_int+ s z TestLoc.test_loc_getitem_intc C s | j dddgtd� d S )Nr �c�empty)r r )r! r" )r# r$ r$ r% �test_loc_getitem_label0 s z TestLoc.test_loc_getitem_labelc C s� | j dddddddgtd� | j ddd gtd� | j ddd gtd� | j dd dddgtd� | j dd dgtd� | j dd dgd td � | j dd d gd td � d S ) Nr �f�ints�uintsr �mixed�ts)r r �floats� r )r �axesr )r! r" )r# r$ r$ r% �#test_loc_getitem_label_out_of_range5 s z+TestLoc.test_loc_getitem_label_out_of_rangec C s@ | j ddddgdddgtd� | j ddd d gdddgtd� d S ) Nr r � r r+ r, r/ )r r g @�A)r! r" )r# r$ r$ r% �test_loc_getitem_label_listB s z#TestLoc.test_loc_getitem_label_listc C sz | j ddddgdgtd� | j ddddgdd d gdtd � | j dd d dgdd d gdtd � | j ddddgdgdtd � d S )Nr r r3 r r( )r r � r+ r, r/ )r r1 r � � � )r3 r7 )r3 � )r � �multi)r! r" )r# r$ r$ r% �(test_loc_getitem_label_list_with_missingK s z0TestLoc.test_loc_getitem_label_list_with_missingc C s" | j ddddgddgdtd� d S ) Nr r0 � �( r+ r, r3 )r r1 r )r! r" )r# r$ r$ r% �!test_loc_getitem_label_list_failsZ s z)TestLoc.test_loc_getitem_label_list_failsc C s d S )Nr$ )r# r$ r$ r% �!test_loc_getitem_label_array_like` s z)TestLoc.test_loc_getitem_label_array_likec C s$ ddddg}| j d|dgtd� d S )NTFr r( )r r )r! � IndexError)r# �br$ r$ r% �test_loc_getitem_boole s z TestLoc.test_loc_getitem_boolc C s� | j dtdd�dddddgtd � | j dtd d �dgdtd � | j dtd d�dgdtd � | j dtd d�dgdtd � | j dtd dd �dgdtd � d S )Nr r3 r7 r r- r( r. r/ )r r Z20130102Z20130104)r r1 r r � r r: )r! �slice� TypeErrorr" )r# r$ r$ r% � test_loc_getitem_label_slicek s z$TestLoc.test_loc_getitem_label_slicec C st t ddgddgddggddgdddgd�}t�d �|jd < t ddgd dgddggddgdddgd�}t�||� d S ) Nr0 �a�� �col1�col2r6 r3 )�columns�indexr )r3 rK r )r �np�aranger �tm�assert_frame_equal)r# �df�expectedr$ r$ r% � test_setitem_from_duplicate_axis� s (z(TestLoc.test_setitem_from_duplicate_axisN) �__name__� __module__� __qualname__r& r) r2 r5 r= r@ rA rD rH rU r$ r$ r$ r% r * s r c @ s e Zd Zej�dded�ddffded�ddffded�dd ffd ed �ed �dffd ed �ddffded�ded�ffded�d dffg�dd� �Zdd� Zdd� Z dd� Z dd� Z d d � Z d d � Z d d!� Zej�d"d#d$gd#d$d#d$gg�d%d&� �Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zejd1d2� �Zej�d3d4ejd4ejd5�ejd4d4d4d4d4gejd5�g�d6d7� �Zd8d9� Zd:d;� Z d<d=� Z d>d?� Z d@dA� Z ej�dBdCdDdEdFdGdHdIdJg�dKdL� �Z dMdN� Z!dOdP� Z"dQdR� Z#dSdT� Z$e%j&dUdV� �Z'dWdX� Z(dYdZ� Z)d[d\� Z*d]d^� Z+ej�d_d`ge,dad`da�e�d`g�g�ej�dbdcge�dcg�g�ddde� ��Z-ej�dfd4dggd`dhdidEgfdje.djdjdjdjgdkdlej/ej/gdjdjdjdjggd`dhdidEgdm�fdndidEgfdjdoge.dndgej/ej/gdkdldjdogdpdqej/ej/ggd`dhdidEgdm�fdnd`dhdigfejdjdodrgejd5�e.dndgej/gdjdodrgdpdqej/ggd`dhdigdm�fe,dndkda�dhdidEgfdjdodrgdsdtdugge.dndgej/ej/gdkdjdodrgdpdsdtduggd`dhdidEgdm�fe,dndkda�did`dEgfejdjdodrgdsdtduggejd5�e.dndgej/ej/gdodldjdrgdtdqdsduggd`dhdidEgdm�fe,dadada�d`digfe.djdogdrdsgdtduggd`digdm�e.djdgdogdrdldsgdtdqduggd`dhdigdm�fg�dvdw� �Z0dxdy� Z1dzd{� Z2d|d}� Z3ejj4d~d� �Z5d�d�� Z6d�d�� Z7d�d�� Z8ejj9d�d��d�d�� �Z:d�d�� Z;d�d�� Z<d�d�� Z=d�d�� Z>d�d�� Z?e%j&d�d�� �Z@d�d�� ZAd�d�� ZBd�d�� ZCd�d�� ZDd�d�� ZEej�d�d�d�d�g�ej�d�ejejFeGg�e%jHd�d�� ���ZId�d�� ZJe%jHd�d�� �ZKd�d�� ZLej�d�eMejeNeOg�d�d�� �ZPd�d�� ZQej�d�d�eNdng�fd�eNdgg�fg�d�d�� �ZRd�d�� ZSej�d�d4d�dndgdkdldpdqdjdodrg fe,dldo�d4dndgdkd�d�d�d�dodrg fdkdpgd4dndgd�dld�dqdjdodrg fg�d�d�� �ZTd�d�� ZUd�d�� ZVd�d ZWej�d�d�d�gg�d�dƄ �ZXd�dȄ ZYej�d�d`d`gd`e,da�fg�d�d˄ �ZZd�d̈́ Z[d�dτ Z\d�dф Z]d�dӄ Z^d�dՄ Z_ej�d�d�d�g�d�dڄ �Z`d�d܄ Zad�dބ Zbd�d� Zcd�d� ZddaS )��TestLoc2zmsg, keyz'Period\('2019', 'A-DEC'\), 'foo', 'bar'i� �foo�barz&Period\('2019', 'A-DEC'\), 'y1', 'bar'�y1z&Period\('2019', 'A-DEC'\), 'foo', 'z1'�z1z;Period\('2018', 'A-DEC'\), Period\('2016', 'A-DEC'\), 'bar'i� i� z&Period\('2018', 'A-DEC'\), 'foo', 'y1'z;Period\('2017', 'A-DEC'\), 'foo', Period\('2015', 'A-DEC'\)i� i� z&Period\('2017', 'A-DEC'\), 'z1', 'bar'c C sz t td�ddgtd�td�dgtd�dtd �gd d d gd ddgd���dddg�}tjt|d�� |j| W dQ R X dS )a parse_time_string return parameter if type not matched. PeriodIndex.get_loc takes returned value from parse_time_string as a tuple. If first argument is Period and a tuple has 3 items, process go on not raise exception i� �x1Zx2i� i� r\ i� r] i� r3 r r7 r6 r0 r> )r4 �B�CZV1ZV2r4 r_ r` )�matchN)r r � set_index�pytest�raisesr" r )r# �msg�keyrS r$ r$ r% �;test_contains_raise_error_if_period_index_is_in_multi_index� s zDTestLoc2.test_contains_raise_error_if_period_index_is_in_multi_indexc C s> t ddgi�}tjtdd�� |jd d �df W d Q R X d S )NrI r3 u א)ra )r rc rd r" r )r# rS r$ r$ r% �$test_loc_getitem_missing_unicode_key� s z-TestLoc2.test_loc_getitem_missing_unicode_keyc C sT t tj�d�dd� td�D �d�}|jd }|jd d �df jd }t�||� d S ) N)r0 r; c S s g | ]}d |d �qS )ZABCDEr; r$ )�.0�xr$ r$ r% � <listcomp>� s z2TestLoc2.test_loc_getitem_dups.<locals>.<listcomp>r0 )rN )r4 r r r4 )r rO �randomZ random_sample�ranger rQ �assert_series_equal)r# rS rT �resultr$ r$ r% �test_loc_getitem_dups� s z TestLoc2.test_loc_getitem_dupsc C s� t ddddtd�ggdddddgdgd�}tddddtd�gdddddgdd�}|jd }t�||� |jd }t�||� d S ) Nr3 r rZ r[ �20130101rI )rM rN )rN �namer )r r r �ilocrQ rn r )r# rS rT ro r$ r$ r% �test_loc_getitem_dups2� s z TestLoc2.test_loc_getitem_dups2c C s t td�td�tjddd�d d tjddd�d d d ���d �}d d d gf}|�� }|j| d9 < t�|j| d|j| � d}|�� }|j| d9 < |j| d|j| ks�t �dd d gf}|�� }|j| d9 < t� |j| d|j| � d S )NZrtttiZaaader; �float64)�dtypegq= ףp�?r g�(\��տ)�merZ r[ �bar2rw �rr[ rx g @)ry r[ �t) r �listrO rP rb �copyr rQ rn �AssertionErrorrR )r# Zdf_orig�indexerrS r$ r$ r% �test_loc_setitem_dups� s( z TestLoc2.test_loc_setitem_dupsc C s� t dddgtdddgdd�d��}|d dk}|j|d f d }||j|d f< t dddgtdd d gdd�d��}t�||� t dddgdddgd�d d�}|d dk}|j|d f }||j|d f< t dddgdddgd�d d�}t�||� d S )Nr r3 �d rJ i, Zuint32)rv )rI rC rI rC �� i- �uint64)r r r rQ rR )r# Zdf1�ixZnewb1rT �df2Znewb2r$ r$ r% �test_loc_setitem_slice s " z TestLoc2.test_loc_setitem_slicec C s� t dgdgdgdgd��}dddg}|jd d �|f �d �|jd d �|f< t dgtjdgd d �tjdgd d �tjdgd d �d��}t�||� d S ) Nr4 g333333�?g g �)�idrI rC r' rI rC r' Zfloat32)rv )r r �astyperO �arrayrQ rR )r# rS �colsrT r$ r$ r% �test_loc_setitem_dtype s &z TestLoc2.test_loc_setitem_dtypec C sr t td�dddgd�}tjtdd�� |ddg W d Q R X t td��}tjtdd�� |d dg W d Q R X d S ) Nr7 rI rC r' )rN z not in index)ra �dr )r rm rc rd r" )r# �sr$ r$ r% �$test_getitem_label_list_with_missing) s z-TestLoc2.test_getitem_label_list_with_missingrN TFc C sN t dddg�}dt|�� dt|�� �}tjt|d�� |j| W d Q R X d S )Nr3 r r7 z Boolean index has wrong length: z instead of )ra )r �lenrc rd rB r )r# rN r� re r$ r$ r% � test_loc_getitem_bool_diff_len4 s z'TestLoc2.test_loc_getitem_bool_diff_lenc C s d S )Nr$ )r# r$ r$ r% �test_loc_getitem_int_slice< s z#TestLoc2.test_loc_getitem_int_slicec C s� t tj�d�dddgdddgd�}d }tjt|d �� |jd d gd d gf W d Q R X ttd �}d |jd < d |jd< tjtdd �� |jd W d Q R X d}tjt|d �� |jddg W d Q R X d}tjt|d �� |jdg W d Q R X d|jd< tjtdd �� |jddg W d Q R X d |d< d}tjt|d �� |jdg W d Q R X |d= tjt|d �� d|jdg< W d Q R X t dgdggd d gdgd�}d}tjt|d �� |jdgd d �f W d Q R X tjt|d �� |jdg W d Q R X d S )N)r7 r7 rI rC r' �er* �g)rN rM zJ\"None of \[Int64Index\(\[1, 2\], dtype='int64'\)\] are in the \[index\]\")ra r3 r )rv z^-1$���zL\"None of \[Int64Index\(\[-1, -2\], dtype='int64'\)\] are in the \[index\]\"���zE\"None of \[Index\(\['4'\], dtype='object'\)\] are in the \[index\]\"�4r7 z not in indexzH\"None of \[Int64Index\(\[-2\], dtype='int64'\)\] are in the \[index\]\"r �valuezG\"None of \[Int64Index\(\[3\], dtype='int64'\)\] are in the \[index\]\") r rO rl rc rd r" r r �object)r# rS re r� r$ r$ r% �test_loc_to_fail@ s@ zTestLoc2.test_loc_to_failc C sr t dddg�}|jdg tjtt�d�d�� |jdg W d Q R X tjtdd�� |jddg W d Q R X d S )Nr3 r r7 z="None of [Int64Index([3], dtype='int64')] are in the [index]")ra z not in index)r r rc rd r" �re�escape)r# r� r$ r$ r% � test_loc_getitem_list_with_fail� s z(TestLoc2.test_loc_getitem_list_with_failc C s� t tjjdd�dddddgd�}|j�d d � �}|jt�|� }|j| }t�||� |j|j }t�||� |jt j|d d � }t�||� d S ) N)r; r6 )�sizeZalpha_0Zalpha_1Zalpha_2Zbeta_0Zbeta_1)rN c S s d| kS )N�alphar$ )rj r$ r$ r% �<lambda>� � z)TestLoc2.test_loc_index.<locals>.<lambda>�boolean)rv ) r rO rl rN �mapr r� rQ rR �values�pd)r# rS �maskrT ro r$ r$ r% �test_loc_index� s zTestLoc2.test_loc_indexc C s� t tj�dd�ddddgddddgd�}|jd d �dd�f jdd�d d �f }|jddgk�� sft�|j ddgk�� s|t�t t d �gd gd ��jd }t t d �d gd d gdd�}t � ||� |jtks�t�d S )Nr: r4 r_ r` �D)rM rN r r rq r3 )rI rC rI rC )rN rr )r rO rl �randr rs rM �allr} rN r r rQ rn rv r� )r# rS ro rT r$ r$ r% �test_loc_general� s ( zTestLoc2.test_loc_generalc C s" t tdd�ttd�tjd�d��S )Nz 2000-01-01z 2000-01-5r; )rv )r �val)r r r rm rO �int64)r# r$ r$ r% �frame_for_consistency� s z TestLoc2.frame_for_consistencyr� r )rv c C sV t tdtd�tjd�ttd�tjd�d��}|�� }||jd d �df< t�||� d S )Nr r; )rN rv )rv )r r� r ) r r rm rO r� r| r rQ rR )r# r� r� rT rS r$ r$ r% � test_loc_setitem_consistency� s z%TestLoc2.test_loc_setitem_consistencyc C sR t tdtd�d�ttd�tjd�d��}|�� }d|jd d �df< t�||� d S )NrZ r; )rN )rv )r r� r ) r r rm rO r� r| r rQ rR )r# r� rT rS r$ r$ r% �(test_loc_setitem_consistency_dt64_to_str� s z1TestLoc2.test_loc_setitem_consistency_dt64_to_strc C sR t tdtd�d�ttd�tjd�d��}|�� }d|jd d �df< t�||� d S )Ng �?r; )rN )rv )r r� r ) r r rm rO r� r| r rQ rR )r# r� rT rS r$ r$ r% �*test_loc_setitem_consistency_dt64_to_float� s z3TestLoc2.test_loc_setitem_consistency_dt64_to_floatc C sJ t dttd�g�i�}d|jd d �df< t dtdg�i�}t�||� d S )Nr Z20180101�string)r r r r rQ rR )r# rS rT r$ r$ r% �'test_loc_setitem_consistency_single_row� s z0TestLoc2.test_loc_setitem_consistency_single_rowc C st t ddgd�}|d �tj�|d< t ddgd�}d|jd d �df< t�||� t ddgd�}d|d< t�||� d S )Nrj �y)rM r3 )r r� rO r� r rQ rR )r# rT rS r$ r$ r% �"test_loc_setitem_consistency_empty� s z+TestLoc2.test_loc_setitem_consistency_emptyc C s� d}t jt|�ddgdddgd�}t|jd d �df �|jd d �df< t|jd d �df �|jd d �df< |jd d �df |jd d �df |jd d �df< |jd d �df �d �|jd d �df< td d d d g|jdd�}t� |d |� d S )Na� Level_0,,,Respondent,Respondent,Respondent,OtherCat,OtherCat Level_1,,,Something,StartDate,EndDate,Yes/No,SomethingElse Region,Site,RespondentID,,,,, Region_1,Site_1,3987227376,A,5/25/2015 10:59,5/25/2015 11:22,Yes, Region_1,Site_1,3980680971,A,5/21/2015 9:40,5/21/2015 9:52,Yes,Yes Region_1,Site_2,3977723249,A,5/20/2015 8:27,5/20/2015 8:41,Yes, Region_1,Site_2,3977723089,A,5/20/2015 8:33,5/20/2015 9:09,Yes,Nor r3 r )�headerZ index_col)� RespondentZ StartDate)r� ZEndDate)r� ZDurationztimedelta64[s]id i� iH g �@)rN rr ) r� Zread_csvr r r r� r rN rQ rn )r# �datarS rT r$ r$ r% �-test_loc_setitem_consistency_slice_column_len s """ z6TestLoc2.test_loc_setitem_consistency_slice_column_len�unit�Y�Mr� �h�mr� �ms�usc C s� t dt�d�t�d�gi�}|jd d �df j�d|� d��|jd d �|f< |jd d �df j�d|� d��|d< t|jd d �df |d�}t�|jd d �|f |� d S )N� timestampz2017-02-11 12:41:29z1991-11-07 04:22:37z datetime64[�]rT )rr ) r rO � datetime64r r� r� r rQ rn )r# r� rS rT r$ r$ r% � test_loc_assign_non_ns_datetime' s 0&z(TestLoc2.test_loc_assign_non_ns_datetimec C s� t �dddddgi�}t|d ddd�|d < |jd d �d f |jd d �d f< |jd d gd f |jd d gd f< t dd d gdddgdddgdddggdd d gd�}d d g}|| �t�||< t�||� d S )Nr l �CO!gl �s~!gl 4 8�l A{�?�r� T)r� �cacheZdate_dtZ date_dt_cpr r7 z2017-01-24 13:26:12.711z2017-01-24 13:52:05.110z2018-10-22 13:44:05.888z2018-10-23 08:08:41.025)rM )r � from_dictr r �applyrQ rR )r# rS rT rM r$ r$ r% �test_loc_modify_datetime9 s z!TestLoc2.test_loc_modify_datetimec C s� t dddgdgtd�}tjdddgdd �|jdddgdf< tdddgdddgtd �}|rrtdddgdddgdd �}t d|i�}t�||� d S ) Nr7 r; r: r4 )rN rM rv r3 r r� )rv )rN rv )r �floatrO r� r r rQ rR )r# �using_array_managerrS �serrT r$ r$ r% �#test_loc_setitem_frame_with_reindexS s $ z,TestLoc2.test_loc_setitem_frame_with_reindexc C s� t dddgddgtd�}d|d< tjdd dgd d �|jdddgdf< td ddgdddgd d �}t d|i�}d|d< t�||� d S ) Nr7 r; r: r4 r_ )rN rM rv r� r3 r r� )rv )rN rv )r r� rO r� r r rQ rR )r# rS r� rT r$ r$ r% �)test_loc_setitem_frame_with_reindex_mixedb s $ z2TestLoc2.test_loc_setitem_frame_with_reindex_mixedc C st t dddgddgtd�}d|d< tjdddgdd �|jtdd d �df< t dddgdd �dddgd �}t�||� d S )Nr3 r r7 r4 r_ )rN rM rv r� r� )rv r r� )r4 r_ )rN )r r� rO r� r rF rQ rR )r# rS rT r$ r$ r% �*test_loc_setitem_frame_with_inverted_slicel s & z3TestLoc2.test_loc_setitem_frame_with_inverted_slicec C s� dd� t d�D �}tjddd�}dd� t d�D �}tjddd�}tt|��|��}t|d�}tj|d < ||j|d f< tj|d < ||j|d f< t ||tj d �}t ||d�}t||d ��j |d�} t � || � d S ) Nc S s g | ]}d t |� �qS )�@)�str)ri �ir$ r$ r% rk y s z9TestLoc2.test_loc_setitem_empty_frame.<locals>.<listcomp>r; r� )rv c S s g | ]}d t |� �qS )r� )r� )ri r� r$ r$ r% rk | s r: )rN r4 r_ )rN rv )r4 r_ )rm rO rP r{ �set�unionr �nanr r ru �reindexrQ rR ) r# Zkeys1Zval1�keys2Zval2rN rS ZseraZserbrT r$ r$ r% � test_loc_setitem_empty_framev s z%TestLoc2.test_loc_setitem_empty_framec C s� t tj�dd�td�td�d�}|jd }d|jd< |jd }|dksJt�|jd }|dks`t�d|jd d �d d �f< |jd d �d d �f }|jd d �dd �f }t� ||� d S ) Nr: �abcd�ABCD)rN rM )r r r3 )rI r4 r r_ r� ) r rO rl �randnr{ rs r r} rQ rR )r# rS ro rT r$ r$ r% �test_loc_setitem_frame� s z TestLoc2.test_loc_setitem_framec C sT t dddgtjd��}|j|j|j|jkdf< t dddgtjd��}t�||� d S )Nr3 r r7 )r4 r_ r_ )r rO r� r4 r r_ rQ rR )r# rS rT r$ r$ r% �/test_loc_setitem_frame_nan_int_coercion_invalid� s z8TestLoc2.test_loc_setitem_frame_nan_int_coercion_invalidc C s� t ddgddgddgd��}|jdddgf }tddgddgtdd �}t�||� t d dgd dgddgd��}d d g|jdddgf< t�||� d S ) Nr3 r r7 r: rI rC )r3 r rI r )rN rv rr r; r8 )r r r r� rQ rn rR )r# rS ro rT r$ r$ r% �#test_loc_setitem_frame_mixed_labels� s z,TestLoc2.test_loc_setitem_frame_mixed_labelsc C s t dddgttd�tjd�d��}|jdd� }|jd d� |_||jd d�< t dddgtdddgtjd�d��}t�||� t t d d �ttd �tjd�d ��}t t d�t d�t d�t d�t d�gtd dd ddgtjd�d ��}|jd d� }|jdd � |_||jdd�< t�||� d S )NrZ r[ �bazr7 )rv )r4 r_ r3 r r z 2000-01-01z 2000-01-5r; )r r� Z20000101Z20000102Z20000103r: ) r r rm rO r� r rN rQ rR r r )r# rS �rhsrT r$ r$ r% � test_loc_setitem_frame_multiples� s, " z)TestLoc2.test_loc_setitem_frame_multiplesr~ r4 Nr� �Zc C sL t ddgddggddgd�}||jd|f< |jd }t|�rD|d ksHt�d S ) Nr3 r r7 r: r4 r_ )rM r )r r4 r� )r r r r} )r# r~ r� rS ro r$ r$ r% �"test_loc_setitem_with_scalar_index� s z+TestLoc2.test_loc_setitem_with_scalar_indexzindex,box,expectedr r_ r` r9 r7 r: )rM r3 rE r; r8 � r6 � � c C s< t ddgddgddggddgd �}||j|< t�||� d S ) Nr3 r r7 r: r; r8 r4 r_ )rM )r r rQ rR )r# rN �boxrT rS r$ r$ r% � test_loc_setitem_missing_columns� s 5" z)TestLoc2.test_loc_setitem_missing_columnsc C s t dtd��d�tjgi�}|j}|jdg }t�|j|� |jdg }t�|j|� dd l }t d|� ddd�|� ddd�gi�}|j}|jdg }t�|j|� |jdg }t�|j|� t d d gd gd i�}|j}|jdd� }t�|j|� |jd d � }t�|j|� d S ) Nr rq �UTCr r3 i� i� r �textz some wordsr� r7 ) r r � tz_localizer� �NaT�dtypesrs rQ rn r )r# rS rT ro r r$ r$ r% �test_loc_coercion! s( " zTestLoc2.test_loc_coercionc C sf t d��d�t d��d�g}t|ddgd�}ttd�}|d ||�d< |d ||�d< t�||� d S ) N�* r� i� rZ r[ )rN )rv r r3 )r r� r r� rQ rn )r# Z indexer_sl�valsrT r� r$ r$ r% �test_setitem_new_key_tzE s z TestLoc2.test_setitem_new_key_tzc C s� t ddddddgddddddgd �d dd dddgd �}d }tjt|d �� |jdd � W d Q R X d}tjt|d �� |jd d � W d Q R X d }tjt|d �� |jdd� W d Q R X t ddddddgddddddgd �d dd dddgd �jd d�}|jdd � }t ddddgddddgd �ddddgd �}t�||� |jd d � }t�||� |jdd� }t dddgdddgd �dddgd �}t�||� d S )Nr3 r r7 r: r; r8 r9 rE )r4 r_ r )rN z5'Cannot get left slice bound for non-unique label: 1')ra z5'Cannot get left slice bound for non-unique label: 0')�axis)r rc rd r" r � sort_indexrQ rR )r# rS re ro rT r$ r$ r% �test_loc_non_uniqueS s, 44* $z TestLoc2.test_loc_non_uniquec s� t d�� � fdd�}� fdd�}|dd�}|jjdks:t�t�d�}|j| }|||�}t�||� |d d �}|jjdks~t�t�d �}|j| }|||�}t�||� d S ) NZABCDEFGc sJ t �ttj�| t� ��t�| �� d�tt�|t� �f�dg| � d�g�S )N)rN rM r ) r� �concatr rO rl r� r� rP �ones)�length�l2)rM r$ r% �gen_test} s z;TestLoc2.test_loc_non_unique_memory_error.<locals>.gen_testc sL t |�}t�| �dg�tt�|t � �f�dg| � d�| �|dd � �g�S )Nr )rN rM r3 )r� r� r� �taker rO r� )rS r� Zlen_mask)rM r$ r% � gen_expected� s z?TestLoc2.test_loc_non_unique_memory_error.<locals>.gen_expectedi� r� Fi�� i�� ) r{ rN Z is_uniquer} rO rP r rQ rR )r# r� r� rS r� ro rT r$ )rM r% � test_loc_non_unique_memory_erroru s z)TestLoc2.test_loc_non_unique_memory_errorc C s\ t ddgddgg�}d|j_|jddg jj}|dks:t�|jddg jj}|dksXt�d S )Nr3 Z index_namer )r rN rr rs r} r )r# rS ro r$ r$ r% � test_loc_name� s zTestLoc2.test_loc_namec C s� t �dd�}t j|jd d �g f |jd d �d d�f ddd� t j|jg d d �f |jd d�d d �f ddd� t j|jg |jd d�d d �f ddd� d S )Nr; r r T)�check_index_type�check_column_type)rQ ZmakeCustomDataframerR r rs )r# rS r$ r$ r% �!test_loc_empty_list_indexer_is_ok� s ..z*TestLoc2.test_loc_empty_list_indexer_is_okc C sD t ddddgi�}|jd d � }||k s,t�|d d � |k s@t�dddg|d< |rh|d dk�� r|t�n|d dk�� s|t�||jd d �d d �f ks�t�t tj�dd��}|d |jd d �df ks�t�tdddddd g�}|jd d � }||k s�t�|d d � |k �st�d d d g|d d�< t|d d� d d d gk��s@t�d S ) NrI r3 r r7 r: r6 r r; r8 r9 rE r� )r r r} r� rO rl r� r )r# r� Z original_dfZ sliced_dfrS Zoriginal_seriesZ sliced_seriesr$ r$ r% �&test_identity_slice_returns_new_object� s" z/TestLoc2.test_identity_slice_returns_new_objectz!accidental fix reverted - GH37497)�reasonc C s| t ttd�td��ddgd�}|�� }|jd d �df }|d7 }t�||� |�� }|j|jdf }|d7 }t�||� d S )Nr7 rI rC )rM r )r �ziprm r| r rQ rR rN )r# rj r� �q�zr$ r$ r% �test_loc_copy_vs_view� s z TestLoc2.test_loc_copy_vs_viewc C s� t �d�j}tddg|d |gd�}|j|d }|jd }||ksHt�|j|d g }|jdg }t�||� |j|d |g }t�||� d S )Nr� r3 r )rN r ) rO �iinfo�maxr r rs r} rQ rn )r# �umaxr� ro rT r$ r$ r% �test_loc_uint64� s zTestLoc2.test_loc_uint64c C sr t �d�j}tddg|d |gd�}tjtdd�� |jd W d Q R X tjtdd�� |jdg W d Q R X d S )Nr� r3 r )rN z-1)ra r� )rO r r r rc rd r" r )r# r r� r$ r$ r% �!test_loc_uint64_disallow_negative s z*TestLoc2.test_loc_uint64_disallow_negativec C sR dddg}t |d gt|� d��}t ddgd�}||jd d �df< t�||� d S )Nr3 r r7 )rj r� rj r� )rM )r r� r rQ rR )r# r� rT rS r$ r$ r% �*test_loc_setitem_empty_append_expands_rows s z3TestLoc2.test_loc_setitem_empty_append_expands_rowsc C sf dddg}t |d gt|� d��}t ddgd�}|d �tj�|d< ||jd d �df< t�||� d S )Nr3 r r7 )rj r� rj r� )rM )r r� r� rO r� r rQ rR )r# r� rT rS r$ r$ r% �6test_loc_setitem_empty_append_expands_rows_mixed_dtype s z?TestLoc2.test_loc_setitem_empty_append_expands_rows_mixed_dtypec C sD t dgtjgd��}t ddgtd�}|jd |jd< t�||� d S )Ng �?)rj r� rj r� )rM rv )r rj )r rO r� r� r rQ rR )r# rT rS r$ r$ r% �*test_loc_setitem_empty_append_single_value& s z3TestLoc2.test_loc_setitem_empty_append_single_valuec C s� ddg}t ddgd�}|j�tj�|_d}tjt|d�� ||jddgdf< W d Q R X d � d d d g�}tjt |d�� ||jdd�df< W d Q R X d S ) Nr3 r rj r� )rM zFNone of \[Int64Index\(\[0, 1\], dtype='int64'\)\] are in the \[index\])ra r �|z?cannot copy sequence with size 2 to array axis with dimension 0zCcould not broadcast input array from shape \(2,\) into shape \(0,\)z@Must have equal len keys and value when setting with an iterable) r rN r� rO r� rc rd r" r �join� ValueError)r# r� rS re r$ r$ r% �$test_loc_setitem_empty_append_raises/ s z-TestLoc2.test_loc_setitem_empty_append_raisesc C sD t ddgddgg�}|jt�d� }tddgdd�}t�||� d S )Nr3 r r7 r: r )rr )r r rO r� r rQ rn )r# rS ro r� r$ r$ r% � test_indexing_zerodim_np_arrayG s z'TestLoc2.test_indexing_zerodim_np_arrayc C s, t ddg�}|jt�d� }|dks(t�d S )Nr3 r r )r r rO r� r} )r# r� ro r$ r$ r% �%test_series_indexing_zerodim_np_arrayN s z.TestLoc2.test_series_indexing_zerodim_np_arrayc C sl ddddddgd gd }t |tdd�d �}t tdd�tjd �}ddddddg|jd dd �< t�||� d S ) Nr3 r r7 r: r; r8 i� i� )rN )rN rv i� r� )r rm rO ru r rQ rn )r# r� rT ro r$ r$ r% �test_loc_reverse_assignmentT s z$TestLoc2.test_loc_reverse_assignmentc C s� t j�d� dd� td�D �}t|dgd�}t|dgtd�}t�||� dd� |D �|j|j df< t|dgt d�� t�}t�||� d d� |D �|d< t|dgt d�}t�||� d S ) N� c S s g | ]}t tj�� d ��qS )g�-���q=)r� rO rl )ri �_r$ r$ r% rk a s zPTestLoc2.test_loc_setitem_str_to_small_float_conversion_type.<locals>.<listcomp>r; r4 )rM )rM rv c S s g | ] }t |��qS r$ )r� )ri rj r$ r$ r% rk h s c S s g | ] }t |��qS r$ )r� )ri rj r$ r$ r% rk n s ) rO rl �seedrm r r� rQ rR r rN r� r� )r# Zcol_dataro rT r$ r$ r% �3test_loc_setitem_str_to_small_float_conversion_type^ s z<TestLoc2.test_loc_setitem_str_to_small_float_conversion_typec C s� t dddd�}|jdk|jdk@ }ttj�t|�d�|d�}|tkrL|d }|j t dd� }|j | }t � ||� |j d d � }|j t dd� }|d d � }|j �d �|_ |j �d �|_ t � ||� d S ) Nz1/1/2000z1/5/2000Z5min)�freqr� r> r7 )rN r z1/4/2000r� )r �hour�minuter rO rl r� r� r r r rQ � assert_equalrN � _with_freq) r# �frame_or_series�rngr� �objro �exp�chunkrT r$ r$ r% � test_loc_getitem_time_objectr s z%TestLoc2.test_loc_getitem_time_object� spmatrix_tZ coo_matrixZ csc_matrixZ csr_matrixrv c C s� dd l }t|j|�}d\}}|tj|||d�|d�}tj�|�}td|�}|j| j } |� � | } t � | | � |j| j j } t�|t|dd��} t � | | � d S )Nr )r; r9 )rv r )Z fill_value)� scipy.sparse�getattr�sparserO �eyer � from_spmatrixrm r r� ZtoarrayrQ Zassert_numpy_array_equalr� �fullr ) r# r" rv Zscipy�rowsr� ZspmatrixrS Zitr_idxro rT r$ r$ r% �$test_loc_getitem_range_from_spmatrix� s z-TestLoc2.test_loc_getitem_range_from_spmatrixc C s< t dtjddgtd�d�i�}|jddg }t�||� d S )Nr4 r r� )rv r3 )r r� r� r r rQ rR )r# rS ro r$ r$ r% �,test_loc_getitem_listlike_all_retains_sparse� s z5TestLoc2.test_loc_getitem_listlike_all_retains_sparsec C s� ddl m} tj�|d��}|jtd� }tdddddgdddddggtdd�d�}t� ||� |jtd� jtd � }tdddddggtdd�d�}t� ||� d S ) Nr )r& r; r g �?g ru )rv r3 ) r# r& r r% r' r rm r rQ rR )r# r& rS ro rT r$ r$ r% � test_loc_getitem_sparse_frame� s z&TestLoc2.test_loc_getitem_sparse_framec C s� t dddddgtdd�d�}|jtd� }t ddgtdd�d�}t�||� |jtd� jtd� }t ddgtdd�d�}t�||� d S )Ng �?g ru )rv r r7 )r r r rm rQ rn )r# r� ro rT r$ r$ r% � test_loc_getitem_sparse_series� s z'TestLoc2.test_loc_getitem_sparse_series�key_typec C sH |dddg�}|j d d �|f }|j d d �dddgf }t�||� d S )Nr4 r_ r` )r rQ rR )r# � float_framer. �idxro rT r$ r$ r% �test_loc_getitem_iterable� s z"TestLoc2.test_loc_getitem_iterablec C sb t tjjdd��}tdddd�|_|jtd�d �d d �f }|jdd �d d �f }t� ||� d S )N)r6 r: )r� Z0sr6 r� )�start�periodsr ) r rO rl �normalr rN r r rQ rR )r# rS rT ro r$ r$ r% �#test_loc_getitem_timedelta_0seconds� s z,TestLoc2.test_loc_getitem_timedelta_0secondsz val,expectedl ���� l c C s4 t ddgddgd�}|j| }||_t�||� d S )Nr3 r l ���� l )rN )r r rr rQ rn )r# r� rT rS ro r$ r$ r% � test_loc_getitem_uint64_scalar� s z'TestLoc2.test_loc_getitem_uint64_scalarc C sD t dddgdddgd�}|�� }d|jd< d|jd < t�||� d S ) NrI rC r' r g �?r3 )rN Zzoor )r r| r rs rQ rn )r# r� rT r$ r$ r% �,test_loc_setitem_int_label_with_float64index� s z5TestLoc2.test_loc_setitem_int_label_with_float64indexzindexer, expectedr0 c C s\ t td�dd�}tdtd�id|d�}d|j|j| df< t||dgdd�}t�||� d S ) Nr6 r� )r� rj r� )rv rN r0 )rN rM rv )r rm r r rN rQ rR )r# r~ rT �tdirS r$ r$ r% �/test_loc_setitem_listlike_with_timedelta64index� s z8TestLoc2.test_loc_setitem_listlike_with_timedelta64indexc C s� t dddddgtd�d��}t dddddgtd�d��}tddgddgd�|jdd�df< tddgddgd�|jdd �df< t�||� d S ) Nr3 Zaaaaa)rI rC rC ZaabbarI )� categoriesr r7 )r r{ r r rQ rR )r# rS r r$ r$ r% �8test_loc_setitem_categorical_values_partial_column_slice s ""zATestLoc2.test_loc_setitem_categorical_values_partial_column_slicec C sb t dgdgd��}t|d dddgd�}||jd d �df< |d }t||jdd�}t�||� d S ) NrI r )�AlphaZNumericr< rC r' )r: )rN rr )r r r r rN rQ rn )r# rS r: ro rT r$ r$ r% �'test_loc_setitem_single_row_categorical s z0TestLoc2.test_loc_setitem_single_row_categoricalc C s� t dtd�gd i�}t�d�|jdd�df< td�|jd ksDt�td�|jd ksZt�td d d �|jd < t�t �� td �|jd ks�t�W d Q R X td ��� |jd ks�t�d S ) Nr' z 2010-10-01r7 z 2008-08-08r r3 )r r' )r3 r' i� r; )r r' z 2005-05-05) r r rO r� r r} r rQ �assert_produces_warning� FutureWarning)r# rS r$ r$ r% �"test_loc_setitem_datetime_coercion s z+TestLoc2.test_loc_setitem_datetime_coercion�idxer�varc C sX |}t ddd|d�}td|dgd�}t|dgtjd�}||jd d �|f< t�||� d S ) Nz 2015-07-12r7 �H)r2 r3 r �tzg333333�?rB )rN rM )rN rM rv )r r rO ru r rQ rR )r# rA Ztz_naive_fixturerD r0 rT ro r$ r$ r% �!test_loc_setitem_datetimeindex_tz" s z*TestLoc2.test_loc_setitem_datetimeindex_tzc C s| t dddd�}ttj�t|�d�|d�}tddd�}ttd dd�td dd��}d d d dg}ddddddddddddg }|�� }d|j |< |j | }|j | �� } d| j d d �< |r�| � t �} t � || � |�� }d|j |< |j| |j |< t � ||� |�� }d|j |< |j | }|j | �� } d| j d d �< |�r>| � t �} t � || � |�� }d|j |< |j| |j |< t � ||� d S )Nz 2012-01-01z 2012-01-05Z30min)r r; )rN r� r r � � �H �x � � � � �J �K �L �z �{ �| � � � )r r rO rl r� r� r rF r| r r� r� rQ rR rs ) r# r� rN rS ZakeyZbkeyZaindsZbindsro rT r$ r$ r% �test_loc_setitem_time_key. s< z"TestLoc2.test_loc_setitem_time_keyrf c C s� t �dddg�}tdddgddd gg|d �}|�� }tjd td �|jd d �|f< td dd gd dd gg|d �}t� ||� |j dd�}tjd td �|jd d �|f< |j dd�}t� ||� d S )N)r4 r: )r_ �3)r4 �2r3 r r7 r: r; r8 )rM )r r )rv r )r� ) r � from_tuplesr r| rO Zzeros�intr rQ rR r� )r# rf �mirS r rT r$ r$ r% �,test_loc_setitem_unsorted_multiindex_columnsT s z5TestLoc2.test_loc_setitem_unsorted_multiindex_columnsc C s> t dddg|d�}d|jd< t dddg|d�}t�||� d S )Nr3 r r7 )rv r: r )r r rQ rn )r# Z any_int_dtypeZseriesrT r$ r$ r% �test_loc_setitem_uint_dropc s z#TestLoc2.test_loc_setitem_uint_dropc C sp t dt�dd�g �}t�dd�|jdddg< t dt�dd�g �}tt�dd��|jdddg< t�||� d S )Nr6 r� r0 r3 r r7 )r rO � timedelta64r r rQ rn )r# r� rT r$ r$ r% � test_loc_setitem_td64_non_nanok s z'TestLoc2.test_loc_setitem_td64_non_nanoc C s� t j�dd�}ttd��}d�ddg�}tjt|d�� ||j td�< W d Q R X d}tjt|d�� ||j d d �< W d Q R X d S )Nr r z,shape mismatch: value array of shape \(2,2\)z0cannot reshape array of size 4 into shape \(2,\))ra zEcould not broadcast input array from shape \(2,2\) into shape \(2,?\)) rO rl r� r rm r rc rd r r )r# r� r� re r$ r$ r% � test_loc_setitem_2d_to_1d_raisess s z)TestLoc2.test_loc_setitem_2d_to_1d_raisesc C sX t jddd�}tdddgdddgd d d gg|d d dgd�}d}|jd }t�||� d S )Nr r7 )r2 r3 r3 r r: r; r8 r9 rE r� r4 r_ r` )rN rM )g �?r4 )r� �interval_ranger r rQ �assert_almost_equal)r# rN rS rT ro r$ r$ r% � test_loc_getitem_interval_index� s * z(TestLoc2.test_loc_getitem_interval_indexc C sz t jdddd�}tdddgddd gd d d gg|d ddgd�}t jddddd�}tddg|d d�}|jd }t�||� d S )Nr r7 Zboth)r2 r3 �closedr3 r r: r; r8 r9 rE r� r4 r_ r` )rN rM )r2 r3 r re )rN rr )r3 r4 )r� rb r r r rQ rn )r# rN rS Z index_exprT ro r$ r$ r% � test_loc_getitem_interval_index2� s * z)TestLoc2.test_loc_getitem_interval_index2�tpl)r3 )r3 r c C sR t ddgddd�}t|d�}|j|g }t |gddd�}t|d�}t�||� d S )N)r3 )r3 r r4 F)rr � tupleize_cols)rN )r r r rQ rR )r# rg r0 rS ro rT r$ r$ r% �+test_loc_getitem_index_single_double_tuples� s z4TestLoc2.test_loc_getitem_index_single_double_tuplesc C sn t dddg�}|dd�}|dd�}t||gdd d �}td d g|d dgd�}|j|dd� d }|dksjt�d S )N� IndexTyperI rC rZ r[ r� ZbofZcomposite_indexF)rr rh )r3 r )r7 r: r4 r_ )rN rM r3 )r r r r r} )r# rj Zidx1Zidx2rN rS ro r$ r$ r% �!test_loc_getitem_index_namedtuple� s z*TestLoc2.test_loc_getitem_index_namedtuplec C s� t tj�dd�dddddgdd d gd �}d |d < tj|j|jd d d� d f< tjtjd tjd tjgtd�}t � |d j |� d S )Nr; r7 rI rC r' r� r� rZ r[ r� )rN rM �quxr� r )rv ) r rO rl r� r� r rN r� r� rQ rc r� )r# rS rT r$ r$ r% �$test_loc_setitem_single_column_mixed� s z-TestLoc2.test_loc_setitem_single_column_mixedc C sp t tj�dd�td�d�}tj|d< d|jd< |j}tt� d�gd t� d �g d d d dgd �}t � ||� d S )Nr> r7 �ABC)rM �eventrZ )r6 ro ru r� r4 r_ r` )rN ) r rO rl r� �tupler� r r� r rv rQ rn )r# rS ro rT r$ r$ r% �test_loc_setitem_cast2� s z TestLoc2.test_loc_setitem_cast2c C sb t dtjdtjd�i�}d|jd< |jjt�tj�ks:t�t�d�|_|jjt�tj�ks^t�d S )N�oner8 )rv )r3 rr r9 ) r rO rP Zint8r r� rr rv r} )r# rS r$ r$ r% �test_loc_setitem_cast3� s z TestLoc2.test_loc_setitem_cast3)erV rW rX rc �mark� parametrizer rg rh rp rt r r� r� r� r� r� r� r� r� r� Zfixturer� rO r� r� r� r� r� r� r� r� r� r� r� r� r� �tdZ&skip_array_manager_not_yet_implementedr� r� r� r� r� rF r� r r� r� r� r� r� Zarm_slowr� r� r� r� Zxfailr r r r r r r r r r r r! ru �complexZskip_if_no_scipyr* r+ r, r- �iterr r r1 r5 r6 r7 r9 r; r= r@ rE rW r] r^ r` ra rd rf ri rk rm rq rs r$ r$ r$ r% rY � s� $ &C 6 & % & $ ( ,$ $ 6 $"4 ( , && rY c @ s� e Zd Zej�dddgddddgddddggfddgddddgddddggfddgddgfddddgddddggfddgddgfddddgddddggfddgddgfddddgddddggfddgddgfddddgddddggfddgddgfdddgdddggfg�ej�ddd g�d d � ��Zd d � Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zej�d d d g�d d � �Zd!S )"�TestLocWithMultiIndexzkeys, expectedrC rI r3 r r' �dimrN rM c C s� |dddddgdddddggi}t t�d��dd�f|�}t�|�}|dkrn|j|d d �f }t�|j |� n(|d kr�|jd d �|f }t�|j |� d S ) Nr' rI rC r3 r � r; rN rM ) r rO rP �reshaper � from_arraysr rQ �assert_index_equalrN rM )r# rz �keysrT �kwargsrS � exp_index�resr$ r$ r% �'test_loc_getitem_multilevel_index_order� s z=TestLocWithMultiIndex.test_loc_getitem_multilevel_index_orderc C s� |}|j d }|d j d }|jj|jjdd � ks8t�|jj|jjdd � ksTt�|j d }|d j d }|jj|jjd ks�t�|jj|jjd ks�t�d S )Ni� r4 r3 )i� r r )r rN �namesr} rr )r# Z/multiindex_year_month_day_dataframe_random_dataZymdro �result2r$ r$ r% �test_loc_preserve_names� s z-TestLocWithMultiIndex.test_loc_preserve_namesc C sf t �dgddgg�}td|d�}|jg }|d d� }t�||� |j|jdd� }t�||� d S )Nr r3 )rN )r � from_productr r rQ rn rs )r# r\ r� r� rT Zres2r$ r$ r% �.test_loc_getitem_multiindex_nonunique_len_zero� s zDTestLocWithMultiIndex.test_loc_getitem_multiindex_nonunique_len_zeroc C s� t d gt�dgdgg��}|jd }|d ks0t�t�dgddgg�}t d gt|� t|d�}|jd }|d ksrt�t dgt|� t|d�}|jd }|dks�t�d S ) N�Level1�Level2)r� r� �Level2_aZLevel2_b)rv rN )r� r� r3 )r r r} r r} r� r� r� )r# r� ro Zmidxr$ r$ r% �0test_loc_getitem_access_none_value_in_multiindex s zFTestLocWithMultiIndex.test_loc_getitem_access_none_value_in_multiindexc C s� t jtddddddddgddddddddg�ddgd �}td d d d d d d d g|d �}d |jd d�< td d d d d d d d g|d �}t�||� d S )Nr[ r� rZ rl rr �two�first�second)r� r3 )rN r� )r� rr )rZ r� )r rZ r r r rQ rn )r# rN ro rT r$ r$ r% �!test_loc_setitem_multiindex_slice s z7TestLocWithMultiIndex.test_loc_setitem_multiindex_slicec C sJ t dddd�}ttd�|�}|jtddd�tddd�� }t�||� d S )Nz 2000-01-01Z10mini�� )r r3 il r3 i4 )r r rm r r rQ rn )r# �timesr� ro r$ r$ r% �7test_loc_getitem_slice_datetime_objs_with_datetimeindex- s zMTestLocWithMultiIndex.test_loc_getitem_slice_datetime_objs_with_datetimeindexc C sh t td�td�d�tdd�d�}|jddgddgf }t d d gd d gd�tddg�d�}t�||� d S ) Nr6 )rI rC z 2010-01-01z 2010-01-10)rN z 2010-01-05rI rC r r: )r rm r r r rQ rR )r# rS ro rT r$ r$ r% �3test_loc_getitem_datetime_string_with_datetimeindex3 s zITestLocWithMultiIndex.test_loc_getitem_datetime_string_with_datetimeindexc C s� t jddddddgddgd�}tddgdd gd d gd d gddgddgg|ddgd�}|jdd�}tddgdd gddggddg|�ddd g�d�}|jd }t�||� d S )N)rZ r[ )ZbahZbamr4 r_ )r� g �?r3 g @r g @r7 g @r: g @r; g @r8 r` r� )rN rM r )�level)rM rN )r rZ r r� r� r rQ rR )r# r\ rS rT ro r$ r$ r% �3test_loc_getitem_sorted_index_level_with_duplicates@ s, . zITestLocWithMultiIndex.test_loc_getitem_sorted_index_level_with_duplicatesc C s� t t�ddd�ttddg�td�gdgd dgd td�gd d gd �d �}tddgddgd d dd�}|jjd }t� ||� |j dg jjd }t� ||� d S )Nr � rI rC r6 r r; r3 ZIndex1ZIndex2)�levels�codesr� )r� rN F�category)r: �orderedrr rv ) r rO rP r r rm rN r� rQ r~ r )r# rS rT ro r$ r$ r% �5test_loc_getitem_preserves_index_level_category_dtypeb s zKTestLocWithMultiIndex.test_loc_getitem_preserves_index_level_category_dtype�lt_valuer> r6 c C sp t dddddgitd�ddd d ggd �}tjtd d ��0 |j|d |k d d �f jdgd d �f W d Q R X d S )NrI r� � �" �- �aabbr r3 r r7 )rN z\['b'\] not in index)ra rC )r r{ rc rd r" r )r# r� rS r$ r$ r% �>test_loc_multiindex_levels_contain_values_not_in_index_anymore{ s (zTTestLocWithMultiIndex.test_loc_multiindex_levels_contain_values_not_in_index_anymoreN)rV rW rX rc rt ru r� r� r� r� r� r� r� r� r� r� r$ r$ r$ r% ry � s& &&&&& "ry c @ s� e Zd Zejjdd� �Zdd� Zdd� Zdd� Z d d � Z d d � Z d d� Z dd� Z ej�d�dd� �Zej�dddddddg�dd � �Zd S ) �TestLocSetitemWithExpansionc C sN t dtd�idd�}t|�d |jt|�< t dtd�idd�}t�||� d S )Nrj i@B r� )rv r3 iAB )r rm r� r rQ rR )r# ro rT r$ r$ r% �/test_loc_setitem_with_expansion_large_dataframe� s zKTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_large_dataframec C s$ t td�}d|jd< t�|t dgdgd�� d|jd< t�|t ddgddgd�� t td�}d|jd< t�|t dgdgd�� d|jd< t�|t ddgddgd�� t td�}d|jd< t�|t dgdgd�� d|jd< t�|t ddgddgd�� d |jd< t�|t ddd gdddgd�� d S ) N)rv r3 )rN r7 g �?g @rZ r[ r: )r r� r rQ rn )r# r� r$ r$ r% � test_loc_setitem_empty_series� s" z9TestLocSetitemWithExpansion.test_loc_setitem_empty_seriesc sr t dddtd�d�� � fdd�td�D �}td g|d gd �}x|D ]}d|j|< qDW td|d �}t�||� d S ) Ni� r� r3 z US/Pacific)�tzinfoc s g | ]}� t |d d� �qS )i� )�seconds)r )ri r� )�baser$ r% rk � s zUTestLocSetitemWithExpansion.test_loc_setitem_incremental_with_dst.<locals>.<listcomp>� r )rN )r r rm r r rQ rn )r# Zidxsro r. rT r$ )r� r% �%test_loc_setitem_incremental_with_dst� s zATestLocSetitemWithExpansion.test_loc_setitem_incremental_with_dstc C s� t d�}t d�}xrdd� dd� dd� dd� gD ]R}t� }d|j||�d f< d |j||�d f< td d d gi||gd �}t�||� q.W d S )Nz20130101 09:00:00z20130101 10:00:00c S s | S )Nr$ )rj r$ r$ r% r� � r� zQTestLocSetitemWithExpansion.test_loc_setitem_datetime_keys_cast.<locals>.<lambda>c S s | � � S )N)Z to_datetime64)rj r$ r$ r% r� � r� c S s | � � S )N)Z to_pydatetime)rj r$ r$ r% r� � r� c S s t �| �S )N)rO r� )rj r$ r$ r% r� � r� r� rr rJ g Y@g i@)rN )r r r rQ rR )r# Zdt1Zdt2�convrS rT r$ r$ r% �#test_loc_setitem_datetime_keys_cast� s z?TestLocSetitemWithExpansion.test_loc_setitem_datetime_keys_castc C sT t ddgi�}tdg|d�|jd d �df< t dgtdg|d�d��}t�||� d S )Nr4 r3 rC )r� r_ )r4 r_ )r r r rQ rR )r# r� ro rT r$ r$ r% �1test_loc_setitem_categorical_column_retains_dtype� s zMTestLocSetitemWithExpansion.test_loc_setitem_categorical_column_retains_dtypec C s� t ddd�}t ddd�}t ddd�}t||ddd�}||ks@t�t|d gd �}d |j|d f< ttjgt|� d g |�t |g��d gt d �}t � ||� d S ) Nz2017-10-29 00:00:00+0200z Europe/Madrid)rD z2017-10-29 03:00:00+0100z2016-10-10 03:00:00�leftrC )re r r� )rN rM r� )rN rM rv ) r r r} r r rO r� r� �appendr r� rQ rR )r# r2 �endr. r0 ro rT r$ r$ r% �0test_loc_setitem_with_expansion_and_existing_dst� s zLTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_and_existing_dstc C s� t tddg�dgd�}ddg|d< |�d�j�d�|_||jdk �d�j�d �}|�� }||j |jdkdf< t |d |j d gdd �}t � |j|� |j |jdkdf t d � }||j |jdkdf< t � |j |jdkdf |� d S )Nz2015-03-30 20:12:32z2015-03-12 00:11:11r )r� rM �new�old�new_colr� z US/Pacificr )r3 r )rr Z1s)r r rb rN r� r r� Z tz_convertr| r r rQ rn r )r# rS �vr� rT r$ r$ r% �test_setitem_with_expansion� s z7TestLocSetitemWithExpansion.test_setitem_with_expansionc C sR t � }d|jd< d|jd< d|jdtjf< |j}t�ddtjg�}t�||� d S )Nr3 )r r r )r3 r3 r7 r ) r r rO �infrM r� Z Float64IndexrQ r~ )r# rS ro rT r$ r$ r% �0test_loc_setitem_with_expansion_inf_upcast_empty� s zLTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_inf_upcast_emptyz"ignore:indexing past lexsort depthc C s` t |�s d S |�d�}t |�}t�|��tj�}t||dgd�}d}||ksPt�|�t |�|�}t |t �r�|d d |ks�t�n|d |ks�t�t�|d ��tj �}t||dgd�} |� � } || j |df< t�| | � |� � d } || j |< | d �tj�} t�| | � |� � } || j |df< tt|�tjg tjg| t|�g d�|d�} t�| | � d S ) Nr r )rN rM Zkapowr� r3 )r r3 )rN )r� �repeatrO rP r� r� r r} �insert� isinstancer ru r| r rQ rR rn r{ r� r� ) r# rN �request�N�arr�origrf r� Zexp_datarT rS r� r$ r$ r% �/test_loc_setitem_with_expansion_nonunique_index� s6 &zKTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_nonunique_indexrv ZInt32ZInt64ZUInt32ZUInt64ZFloat32ZFloat64c C sv t ddddg|d�}td|i�}t|jd�}||j|jdf< t�||� t|jd�}|j|j|jdf< t�||� d S )Nr r3 r r7 )rv r� )rN )r r rN r rQ rR Z_values)r# rv r� rS ro r$ r$ r% �6test_loc_setitem_with_expansion_preserves_nullable_int) s zRTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_preserves_nullable_intN)rV rW rX rc rt Zslowr� r� r� r� r� r� r� r� �filterwarningsr� ru r� r$ r$ r$ r% r� � s .r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestLocCallablec C s� t ddddgtd�ddddgd��}|jdd� }t�||j|jdk � |jd d� d d �f }t�||j|jd kd d �f � |jd d� d d� f }t�||j|jdkd dd gf � |jdd� dd� f }t�||j|jdkdf � |jdd� dd� f }t�||j|jdkddgf � |jdd� dd� f }t�||j|jdkddgf � |jdd� dd� f }||jd k�s�t�d S )Nr3 r r7 r: r� )r4 r_ r` c S s | j dkS )Nr )r4 )rj r$ r$ r% r� B r� zATestLocCallable.test_frame_loc_getitem_callable.<locals>.<lambda>c S s | j dkS )NrC )r_ )rj r$ r$ r% r� E r� rC c S s | j dkS )Nr )r4 )rj r$ r$ r% r� H r� c S s | j dkS )Nr_ )rM )rj r$ r$ r% r� H r� FTc S s | j dkS )Nr )r4 )rj r$ r$ r% r� K r� c S s dS )Nr_ r$ )rj r$ r$ r% r� K r� r_ c S s | j dkS )Nr )r4 )rj r$ r$ r% r� N r� c S s ddgS )Nr4 r_ r$ )rj r$ r$ r% r� N r� r4 c S s | j dkS )Nr )r4 )rj r$ r$ r% r� Q r� c S s ddgS )Nr4 r_ r$ )rj r$ r$ r% r� Q r� c S s dS )Nr3 r$ )rj r$ r$ r% r� U r� c S s dS )Nr4 r$ )rj r$ r$ r% r� U r� )r3 r4 ) r r{ r rQ rR r4 r_ rn r} )r# rS r� r$ r$ r% � test_frame_loc_getitem_callable< s $ " z/TestLocCallable.test_frame_loc_getitem_callablec C s� t ddddgtd�ddddgd��}|jdd� d d gf }t�||j|jdkd d gf � |jddgd d� f }t�||jddgd d gf � |jdd d� f }t�||jdd d gf � d S ) Nr3 r r7 r: r� )r4 r_ r` c S s | j dkS )Nr )r4 )rj r$ r$ r% r� \ r� zITestLocCallable.test_frame_loc_getitem_callable_mixture.<locals>.<lambda>r4 r_ c S s ddgS )Nr4 r_ r$ )rj r$ r$ r% r� _ r� c S s ddgS )Nr4 r_ r$ )rj r$ r$ r% r� b r� )r r{ r rQ rR r4 rn )r# rS r� r$ r$ r% �'test_frame_loc_getitem_callable_mixtureX s $ z7TestLocCallable.test_frame_loc_getitem_callable_mixturec C s� t ddddgtd�d�td�d�}|jd d � }t�||jd d g � |jd d � d d �f }t�||jd d gd d �f � |jdd � dd � f }t�||jd d gdf � |jdd � dd � f }t�||jd d gdgf � |jd d gdd � f }t�||jd d gdf � |jd d gdd � f }t�||jd d gdgf � |jdd � df }t�||jd d gdf � |jdd � dgf }t�||jd d gdgf � d S )Nr3 r r7 r: r� )�Xr� r� )rN c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� j r� zHTestLocCallable.test_frame_loc_getitem_callable_labels.<locals>.<lambda>r4 r` c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� m r� c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� p r� c S s dS )Nr� r$ )rj r$ r$ r% r� p r� r� c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� s r� c S s dgS )Nr� r$ )rj r$ r$ r% r� s r� c S s dS )Nr� r$ )rj r$ r$ r% r� w r� c S s dgS )Nr� r$ )rj r$ r$ r% r� z r� c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� } r� c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� � r� )r r{ r rQ rR rn )r# rS r� r$ r$ r% �&test_frame_loc_getitem_callable_labelse s" " z6TestLocCallable.test_frame_loc_getitem_callable_labelsc C sZ t ddddgtd�d�td�d�}|�� }d |jd d � < |�� }d |jd d g< t�||� |�� }d|jdd � d d �f< |�� }d|jd d gd d �f< t�||� |�� }d|jdd � dd � f< |�� }d|jd d gdf< t�||� |�� }ddg|jdd � dd � f< |�� }ddg|jd d gdgf< t�||� |�� }t�ddg�|jd d gdd � f< |�� }t�ddg�|jd d gdf< t�||� |�� }d|jd d gdd � f< |�� }d|jd d gdgf< t�||� |�� }d|jdd � df< |�� }d|jd d gdf< t�||� |�� }d |jd d � dgf< |�� }d |jd d gdgf< t�||� d S ) Nr3 r r7 r: r� )r� r� r� )rN i��c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� � r� zATestLocCallable.test_frame_loc_setitem_callable.<locals>.<lambda>r4 r` r0 c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� � r� r� c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� � r� c S s dS )Nr� r$ )rj r$ r$ r% r� � r� r� r; r6 c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� � r� c S s dgS )Nr� r$ )rj r$ r$ r% r� � r� r� c S s dS )Nr� r$ )rj r$ r$ r% r� � r� c S s dgS )Nr� r$ )rj r$ r$ r% r� � r� c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� � r� ���c S s ddgS )Nr4 r` r$ )rj r$ r$ r% r� � r� )r r{ r| r rQ rR rO r� )r# rS r� r r$ r$ r% � test_frame_loc_setitem_callable� sR " z/TestLocCallable.test_frame_loc_setitem_callableN)rV rW rX r� r� r� r� r$ r$ r$ r% r� ; s r� c @ sF e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d g�d d � �Z d S )�TestPartialStringSlicingc C sd t dddgdddgd�tdd d g�d �}t ddgddgd�tdd g�d �}|jd }t�||� d S ) NrI rC r' r3 r r7 )rK rL z 2020-08-01z 2020-07-02z 2020-08-05)rN z2020-08)r r r rQ rR )r# rS rT ro r$ r$ r% �5test_loc_getitem_partial_string_slicing_datetimeindex� s zNTestPartialStringSlicing.test_loc_getitem_partial_string_slicing_datetimeindexc C sD t jdddd�}|�� }|jd d� }|jd d� }t�||� d S )Nz 2017-01-01z 2018-01-01r� )r2 r� r z2017-12r� )r� � period_range� to_seriesr rs rQ rn )r# �pir� ro rT r$ r$ r% �8test_loc_getitem_partial_string_slicing_with_periodindex� s zQTestPartialStringSlicing.test_loc_getitem_partial_string_slicing_with_periodindexc C sB t dddd�}|�� }|jd d� }|jd d� }t�||� d S )Nz1 dayz2 days�1H)r2 r� r z1 daysr� )r r� r rs rQ rn )r# r� r� ro rT r$ r$ r% �;test_loc_getitem_partial_string_slicing_with_timedeltaindex� s zTTestPartialStringSlicing.test_loc_getitem_partial_string_slicing_with_timedeltaindexc C sD t dtd�ittd�dd�d�}|jd }|jd }t�||� d S )Nrj r7 �days)r� )rN r z0 days)r rm r rs r rQ rn )r# rS rT Zslicedr$ r$ r% �#test_loc_getitem_str_timedeltaindex� s z<TestPartialStringSlicing.test_loc_getitem_str_timedeltaindex� indexer_endNz 2020-01-02 23:59:59.999999999c C s� |dgd t td�td�td�td�td�g|d�d �}|dgd t td�td�g|d�d �}td|�}|| }t�||� |j| }t�||� d S ) Nr3 r; z 2019-12-30z 2020-01-01z 2019-12-25z 2020-01-02 23:59:59.999999999z 2019-12-19)rD )rN r )r r rF rQ r r )r# Ztz_aware_fixturer� r r rT r~ ro r$ r$ r% �/test_loc_getitem_partial_slice_non_monotonicity� s( zHTestPartialStringSlicing.test_loc_getitem_partial_slice_non_monotonicity) rV rW rX r� r� r� r� rc rt ru r� r$ r$ r$ r% r� � s r� c @ s� e Zd Zdd� Zdd� Zej�dej dddd �e d d d d �g�d d� �Z dd� Z dd� Z ej�de�dd�dedd�ge�dd�e�dd�edd�gde�d d�edd �gg�d d � �Zej�d d!d"g�d#d$� �Zej�d%dd&g�d'd(� �Zd)d*� ZdS )+�TestLabelSlicingc C s t tjddd�dd� dD �d�}t tjddd�d d� d D �d�}x�||gD ]�}|jtd d d d �d � }t�||� |jd td d dd �� }t�||� |jtd d d d �td d dd �� }t�||� |jtd d d d�d � }|jd d � }t�||� |jdd � }t�||� qJW d S )Ng @ru )rv c S s g | ]}t d d|dd��qS )i� r3 r6 r )r )ri r� r$ r$ r% rk s zMTestLabelSlicing.test_loc_getitem_slicing_datetimes_frame.<locals>.<listcomp>)r3 r r7 r: )rN g @c S s g | ]}t d d|dd��qS )i� r3 r6 r )r )ri r� r$ r$ r% rk s )r3 r r r7 r: i� r3 r6 r: r� z 20010101 11)r rO rP r r rQ rR rs )r# Z df_uniqueZdf_dupsrS ro rT r$ r$ r% �(test_loc_getitem_slicing_datetimes_frame s$ " z9TestLabelSlicing.test_loc_getitem_slicing_datetimes_framec C s� t ddddd�}tdddd d g|d �}td dd �}tddd �}|j||� }tdd g|dd � d �}t�||� || }d}||ks�t�d S )Nz2017-10-29 01:30:00z Europe/Berlinr; z30 min)rD r3 r r r3 r r7 r: )rN z2017-10-29 02:30:00+02:00)rD z2017-10-29 02:00:00+01:00)r r r r rQ rn r} )r# r0 Zseries2Zt_1Zt_2ro rT r$ r$ r% �'test_loc_getitem_label_slice_across_dst" s z8TestLabelSlicing.test_loc_getitem_label_slice_across_dstrN z 2017-01-01z 2018-01-01r� )r2 r� r z1 dayz2 daysr� c C s8 |� � }|jd |d � }|jd d� }t�||� d S )Nr� r� )r� r rs rQ rn )r# rN r� ro rT r$ r$ r% �-test_loc_getitem_label_slice_period_timedelta3 s z>TestLabelSlicing.test_loc_getitem_label_slice_period_timedeltac C s~ dddg}t tj�dd�|d�}|jdd� }t|�dks>t�|jdd � }t|�dks\t�|jdd � }t|�dkszt�d S ) Ng�x"p|�@g\Va��|�@g����|�@r7 r )rN g3333c|�@g �|�@g3333�|�@g�����|�@)r rO rl r� r r� r} )r# rN rS �s1r$ r$ r% �%test_loc_getitem_slice_floats_inexactA s z6TestLabelSlicing.test_loc_getitem_slice_floats_inexactc C s� t tj�d�tjddtd�d�}t|jdd � �dks:t�t|jdd � �dksTt�tjddtd�}d |d < ||_ t|jdd � �dks�t�t|jdd � �dks�t�d S ) Nr6 r0 )rv )rN g (@rE g )@r9 gffffff(@r ) r rO rl r� rP r� r� r r} rN )r# r� r0 r$ r$ r% �)test_loc_getitem_float_slice_float64indexN s z:TestLabelSlicing.test_loc_getitem_float_slice_float64indexzstart,stop, expected_slicer �nsNr� r3 r� r8 r9 r: r; c C s> t td�tdd��}|jt||� }|j| }t�||� d S )Nr� z0 daysz10 days)r rm r r rF rs rQ rn )r# r2 �stopZexpected_slicer� ro rT r$ r$ r% �$test_loc_getitem_slice_label_td64objZ s z5TestLabelSlicing.test_loc_getitem_slice_label_td64objr2 Z2018Z2020c C sL |dddgt d�t d�t d�gd�}t�t�� |j|d� W d Q R X d S ) Nr3 r r7 Z2016Z2019Z2017)rN Z2022)r rQ r> r? r )r# r r2 r r$ r$ r% �)test_loc_getitem_slice_unordered_dt_indexi s z:TestLabelSlicing.test_loc_getitem_slice_unordered_dt_indexr� g �?c C sN |t d�|dddgd�}|j|d� }|t d�|dddgd�}t�||� d S )Nr: r� r �third)rN )rm r rQ r )r# r r� r ro rT r$ r$ r% �1test_loc_getitem_slice_labels_int_in_object_indexr s zBTestLabelSlicing.test_loc_getitem_slice_labels_int_in_object_indexc C sZ t dddd�dgd�}t ddggdgtddgtd�d�}t�|jd d �dd �f |� d S ) Nr3 r r7 )�testr3 r r )rN )rv )r� rN rM )r r r� rQ rR r )r# rS rT r$ r$ r% �*test_loc_getitem_slice_columns_mixed_dtypez s z;TestLabelSlicing.test_loc_getitem_slice_columns_mixed_dtype)rV rW rX r� r� rc rt ru r� r� r r� r� r� rO r_ rF r� r� r� r� r$ r$ r$ r% r� s $ r� c @ sH e Zd Zej�dddg�dd� �Zej�dddg�dd� �Zdd � Zd S ) � TestLocBooleanLabelsAndSlices� bool_valueTFc C sH |� d�}|j dkrD||dd�}tjt|d�� |j| W d Q R X d S )Nz7: boolean label can not be used without a boolean indexr� r� )rN rv )ra )Z inferred_typerc rd r" r )r# rN r r� �messager r$ r$ r% �'test_loc_bool_incompatible_index_raises� s zETestLocBooleanLabelsAndSlices.test_loc_bool_incompatible_index_raisesc C s& |t ddgdd�dd�}|j| d S )NTFr� )rv r� )rN rv )r r )r# r r� r r$ r$ r% � test_loc_bool_should_not_raise� s z<TestLocBooleanLabelsAndSlices.test_loc_bool_should_not_raisec C s< d}||dd�}t jt|d�� |jdd� W d Q R X d S )NzEslice\(True, False, None\): boolean values can not be used in a slicer� )rN rv )ra TF)rc rd rG r )r# rN r r� r r$ r$ r% �test_loc_bool_slice_raises� s z8TestLocBooleanLabelsAndSlices.test_loc_bool_slice_raisesN) rV rW rX rc rt ru r� r� r� r$ r$ r$ r% r� � s r� c @ sL e Zd Zdd� Zej�dddg�dd� �Zdd � Zd d � Z e j d d � �Z dS )�TestLocBooleanMaskc C s� t dtd�i�}ttd�dd�|_|d dk|d dk|d dk g}ddddddddddg ddddd d d d d dg ddddd d d d d dg g}xVt||�D ]H\}}|�� }d|j|df< t |ttd�dd�dgdd�}t�||� q�W d S )Nrj r6 r� )r� r7 r r3 r r: r; r8 r9 rE r� r� )rN rM rv ) r rm r rN r r| r rQ rR )r# rS Z conditions� expected_dataZcondr� ro rT r$ r$ r% �)test_loc_setitem_bool_mask_timedeltaindex� s "z<TestLocBooleanMask.test_loc_setitem_bool_mask_timedeltaindexrD Nr� c C s� t �ddddg�}tdd|d�}tdt �d�i|d��d�}|�� }|j|d d �f |j|d d �f< t� ||� |�� }|j| |j|< t� ||� d S ) NTFZ20010101r: )r3 rD rI )rN ru ) rO r� r r rP r� r| r rQ rR )r# rD r� r0 rS ro r$ r$ r% �+test_loc_setitem_mask_with_datetimeindex_tz� s z>TestLocBooleanMask.test_loc_setitem_mask_with_datetimeindex_tzc C s~ t t�d��dd�td�tdddd�d�}|�� }|jd gtj tj g |d < |j d k }|j | j|j |d f< t � ||� d S ) Ng @r7 r ZABz1/1/2000r� )r3 r )rM rN r r` r3 )r rO rP r| r{ r r| rN r� r� r4 r rQ rR )r# rS rT r� r$ r$ r% �2test_loc_setitem_mask_and_label_with_datetimeindex� s zETestLocBooleanMask.test_loc_setitem_mask_and_label_with_datetimeindexc C s� t d�}t d�}tdt||g�i�}|�� }t|g�}|d jd j}||jddg< |d jd j}||kspt�t� ||� d S )Nr l ,GsO1�colr3 TF) r r r r| rs r� r r} rQ rR )r# Ztd1Ztd2rS Zdf_copyr� rT ro r$ r$ r% �'test_loc_setitem_mask_td64_series_value� s z:TestLocBooleanMask.test_loc_setitem_mask_td64_series_valuec C sB |� � }|d dk}d|j|df< d|j|jdf< t�||� d S )Nr4 r r_ r3 )r| r r� rQ rR )r# r/ rT r� r$ r$ r% �#test_loc_setitem_boolean_and_column� s z6TestLocBooleanMask.test_loc_setitem_boolean_and_column) rV rW rX r� rc rt ru r� r� r� rv � skip_array_manager_invalid_testr� r$ r$ r$ r% r� � s r� c @ sX e Zd Zej�ddd� ejeg�dd� �Z dd� Z dd � Z ej�d d d g�d d� �Z dS )�TestLocListliker� c C s | S )Nr$ )rj r$ r$ r% r� � r� zTestLocListlike.<lambda>c C s t ddtjg�}ttd�|d�}|j||� }t�||� |||� }t�||� |�� j||� }t� ||�� � |d d� }|dd � }d}t j t |d�� |j||� W d Q R X t j t |d�� |||� W d Q R X t j t |d�� |�� j||� W d Q R X d S ) Nr4 r_ r7 )rN r� r3 z not in index)ra ) r rO r� r rm r rQ rn Zto_framerR rc rd r" )r# r� �cir� ro Zser2Zci2re r$ r$ r% �8test_loc_getitem_list_of_labels_categoricalindex_with_na� s" zHTestLocListlike.test_loc_getitem_list_of_labels_categoricalindex_with_nac C sZ t jddddgdd�}tdddd gtd d d d ��}tjtdd�� |j| W d Q R X d S )Nz 2001-01-04z 2001-01-02z 2001-01-14r� )rv r r; rE r� z 2001-01-01r� r: )r r3 z not in index)ra )rO r� r r rc rd r" r )r# rf r� r$ r$ r% �1test_loc_getitem_series_label_list_missing_values s zATestLocListlike.test_loc_getitem_series_label_list_missing_valuesc C sR t t�ddg�t�ddg�d�}tjtdd�� |jt�ddg� W d Q R X d S )Nl PF AHl UB�eN l P2�$�$�n)rN r� z not in index)ra l NZ4F�G )r rO r� rc rd r" r )r# r� r$ r$ r% �9test_loc_getitem_series_label_list_missing_integer_values s zITestLocListlike.test_loc_getitem_series_label_list_missing_integer_values� to_periodTFc C sD t ddddd�}|r |�d�}tddg|dd �}td�td�g}|rRd d � |D �}|j| }tddg|dd �}|s�|j�d �|_tj||d d � td�td�td�g}|r�dd � |D �}tdddgt |d|j d�dd �}|j| }tj||d d � td�td�td�g}|�r dd � |D �}t j t dd�� |j| W d Q R X d S )Nz 2011-01-01z 2011-01-02r� r0 )r rr g�������?g�������?r� )rN rr c S s g | ]}|� d ��qS )r� )r� )ri rj r$ r$ r% rk + s zRTestLocListlike.test_loc_getitem_listlike_of_datetimelike_keys.<locals>.<listcomp>T)r� c S s g | ]}|� d ��qS )r� )r� )ri rj r$ r$ r% rk 8 s )rr rv z 2011-01-03c S s g | ]}|� d ��qS )r� )r� )ri rj r$ r$ r% rk E s z not in index)ra )r r� r r r rN r rQ rn r rv rc rd r" )r# r� r0 r� r ro r r$ r$ r% �.test_loc_getitem_listlike_of_datetimelike_keys s8 z>TestLocListlike.test_loc_getitem_listlike_of_datetimelike_keysN) rV rW rX rc rt ru rO Zasarrayr{ r� r� r� r� r$ r$ r$ r% r� � s " r� z%columns, column_key, expected_columnsi� i� i� r3 ZAllr c C s� t tj�dd�| td�d�}|jd d �|f }|jdddg|f }|j�� rlt dd� |D ��rl|j� t �|_t j ||d d � d S ) Nr7 rn )rM rN r4 r_ r` c s s | ]}t |t�V qd S )N)r� r[ )ri rj r$ r$ r% � <genexpr>Y s z=test_loc_getitem_label_list_integer_labels.<locals>.<genexpr>T)r� )r rO rl r� r{ rs r rM Z is_objectr� r� r[ rQ rR )rM Z column_keyZexpected_columnsrS rT ro r$ r$ r% �*test_loc_getitem_label_list_integer_labelsK s r� c C s� t j�dd�} t| �}t j|jd d �df< t �| t �t jgd ��dd�f�}t|dddddgd �}t � ||� t| �}t j|jd d �df< t � ||� d S ) NrE r: g �?r3 g g �?g @g @)rM ) rO rl r� r r� r Zhstackr� r| rQ rR )Z rand_dataro r� rT r$ r$ r% � test_loc_setitem_float_intindex_ s $ r� c C sv dd� dD �} t t�d�td�t�| �d�}|jdd�d d � }t t�d �td�t�d d d d g�d�}t�||� d S )Nc S s g | ]}d D ] }||f�q qS ))r9 rE r� r6 r$ )ri �yrr� r$ r$ r% rk o s z)test_loc_axis_1_slice.<locals>.<listcomp>)i� i� )r6 rE Z ABCDEFGHIJ)rN rM r3 )r� )i� r� )i� rE )r6 r: )i� r6 )i� r9 ) r rO r� rp r rZ r rQ rR )r� rS ro rT r$ r$ r% �test_loc_axis_1_slicem s r� c C sX t dtd�t�td�td�g�d�} | �� }|jddgf |jddgf< t�|| � d S )NrI r )rN rM r )r r3 )r rm r r� r| r rQ rR )rT ro r$ r$ r% �!test_loc_set_dataframe_multiindex~ s " r� c C s4 t td�tddgtd��} | jd }|dks0t�d S )Nr r3 g @)rv r )r rm r r� r r} )r� ro r$ r$ r% �test_loc_mixed_int_float� s r c C sh t td�ddddgd�} t�t�� d| jd d�< W d Q R X t ddddgddddgd�}t�| |� d S ) Nr: r4 r_ r` r� )rN r r7 )r rm rQ r> r? r rn )r� rT r$ r$ r% �*test_loc_with_positional_slice_deprecation� s r c C s� t ddd�} ttj�d�| d�}|d }d}x\||gD ]P}tjt|d�� |jd d� W d Q R X t� t �� d |jd d�< W d Q R X q6W tjt|d�� |jd d�d f W d Q R X tj t d d �� d |jd d�d f< W d Q R X d S ) Nz 2016-01-01r7 )r3 )r7 r )rN r zOcannot do slice indexing on DatetimeIndex with these indexers \[1\] of type int)ra r3 F)Zcheck_stacklevelr ) r r rO rl rc rd rG r rQ r> r? )�dtirS r� re r r$ r$ r% �#test_loc_slice_disallows_positional� s r c C s� t tj�dd�dddgtdddd�d �} | jd d d dg �� } | j}t� |j �}d }t j t |d �� | j| W d Q R X t j t |d �� | d j| W d Q R X d S )Nr; r7 rI rC r' Z2012rC )r r3 )rM rN r r z/None of \[TimedeltaIndex.* are in the \[index\])ra )r rO rl r� r rs r| rN r� ZTimedeltaIndexZasi8rc rd r" r )rS r r8 re r$ r$ r% �'test_loc_datetimelike_mismatched_dtypes� s r c C s� t jdddd�} ttj�dd�| d�}t�||j| � t�||jt | � � t�||jt | � � t�|j dd � |j| dd � � t�||jt | � � d S ) Nz2002-01z2003-12r� )r rG r6 )rN r r; ) r� r� r rO rl r� rQ rR r r{ rs )r0 rS r$ r$ r% �"test_loc_with_period_index_indexer� s $r c @ s e Zd Zej�dddg�dd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zej�dejejeeg�ej�dddd d g�d d � ��Zej�d ed!d"d"�f d!ed"�fg�d#d$� �Zd%d&� Zej�d'd(d)g�d*d+� �Zej d,d-� �Z d"S ).� TestLocSeriesz val,expected)l ���� r7 )l r: c C s$ t ddd��}|j| |ks t�d S )Nr7 r: )l ���� l )r r r} )r# r� rT r� r$ r$ r% r � s z TestLocSeries.test_loc_uint64c C s� |j dddg }t�|j| |�|�� t�|jdd d� |dd d� � |j ddg \}}|j||� }|�||�}t�||� ||�� k}t�|j| || � |j| || ks�t�|j| || ks�t�d S )Nr7 r: r9 r; r � ) rN rQ rn r r� rs �truncate�medianr} ) r# � string_series�datetime_series�inds�d1�d2ro rT r� r$ r$ r% �test_loc_getitem� s " z TestLocSeries.test_loc_getitemc C s� |j ddg \}}|d d d� dddg }d}tjt|d�� |j||� W d Q R X tjt|d�� d|j||�< W d Q R X d S )Nr; r r r3 r z"Timestamp\('2000-01-10 00:00:00'\))ra )rN rc rd r" r )r# r r r Zts2re r$ r$ r% � test_loc_getitem_not_monotonic� s z,TestLocSeries.test_loc_getitem_not_monotonicc C s> t tj�d�ttddd��d�}|�� }d|jdd�< |jdd� dk�� sPt �|�� }d|jdd�< |jdd� dkj �� s�t �|jdd � }|j dd� }|� dd d dg�}t �||� t �||� |jttd ��ttd dd �� }tjtdd�� |j dd� W d Q R X tjtdd�� d|j dd�< W d Q R X d S )Nr6 r r0 r )rN r: r7 r� r8 rE r; r� r� z^3$)ra )r rO rl r� r{ rm r| rs r� r} r� r r� rQ rn rc rd r" )r# r� �cpro r� rT �s2r$ r$ r% �0test_loc_getitem_setitem_integer_slice_keyerrors� s" "z>TestLocSeries.test_loc_getitem_setitem_integer_slice_keyerrorsc C s4 t |jd d� �}|j| }t�||d d� � d S )Nr6 )rx rN r rQ rn )r# r r0 ro r$ r$ r% �test_loc_getitem_iterator s z'TestLocSeries.test_loc_getitem_iteratorc C s: ||� � k}|�� }d|j|< |}d||< t�||� d S )Nr )r r| r rQ rn )r# r r� ro rT r$ r$ r% �test_loc_setitem_boolean s z&TestLocSeries.test_loc_setitem_booleanc C sP t |jdddg �}d|j|< d}tjt|d�� d|j|dg < W d Q R X d S )Nr; rE r� z\['foo'\] not in index)ra rZ )r{ rN r rc rd r" )r# r r re r$ r$ r% �test_loc_setitem_corner& s z%TestLocSeries.test_loc_setitem_cornerc C s� |j dddg }|�� }|�� }d||< d|j|< t�||� |�� }|�� }d||d |d �< d|j|d |d �< t�||� d S )Nr; r6 r r r )rN r| r rQ rn )r# r �indicesr r r$ r$ r% � test_basic_setitem_with_labels- s z,TestLocSeries.test_basic_setitem_with_labelsc C s t tj�d�ttddd��d�}dddg}t�dddg�}|�� }|�� }d||< d|j|< t � ||� |�� }|�� }d||< d|j|< t � ||� ddddg}t�ddddg�}d }t j t |d �� d||< W d Q R X t j t|d �� d||< W d Q R X d S ) Nr6 r r0 r )rN r: r8 r; z\[5\] not in index)ra )r rO rl r� r{ rm r� r| r rQ rn rc rd r" � Exception) r# r� r Zarr_indsr r Z inds_notfoundZarr_inds_notfoundre r$ r$ r% �!test_loc_setitem_listlike_of_ints< s( z/TestLocSeries.test_loc_setitem_listlike_of_intsc C s� t tdddd�dddgd�}|�� }td dd �}||jd< |jd }||ksPt�|�� }||jd < |jd }||ksxt�|�� }||d< |d }||ks�t�d S ) Nz 2011-01-01r7 z US/Eastern)r3 rD rI rC r' )rN z 2011-01-03)rD r )r r r| r r r} rs )r# r� r rT ro r$ r$ r% � test_loc_setitem_dt64tz_valuesW s z,TestLocSeries.test_loc_setitem_dt64tz_values�array_fnr� r r: r; r8 c C s~ |dg| �}t |ddddgtd�td�}t dtd�td�}||jd< t�||� t dtd�td�}||jd< t�||� d S )Nr Zabcde)rN rv rI )r r{ r� r rQ rn rs )r# r� r r� rT r� r$ r$ r% �#test_loc_iloc_setitem_with_listlikem s z1TestLocSeries.test_loc_iloc_setitem_with_listliker~ r4 Nc C s~ t t�dddg�dddgd�}d}tjt|d �� |j|d d �f W d Q R X tjt|d �� d |j|d d �f< W d Q R X d S ) N)r4 �0)r4 �1)r_ r � r� r� )rN r� zToo many indices)ra r3 )r r rZ rc rd r r )r# r~ r� re r$ r$ r% �+test_loc_series_getitem_too_many_dimensions s z9TestLocSeries.test_loc_series_getitem_too_many_dimensionsc C s� |j dddg }|�� }d|j|< |�� }d|dddg< t�||� d|jdd�< d|dd�< t�||� |j ddg \}}d|j||�< d|dd�< t�||� d|j|< d|j|< || dks�t�|| dks�t�d S ) Nr7 r: r9 r; r6 r r8 r� )rN r| r rQ rn rs r} )r# r r ro rT r r r$ r$ r% �test_loc_setitem� s" z TestLocSeries.test_loc_setitemrv r� r� c C s\ t ddgddgd�|d�}ddd�|jd d d �f< t ddgddgd�|d�}t�||� d S ) N�abc�defZghiZjkl)r4 r_ )rv ZnewAZnewBr )r r rQ rR )r# rv rS rT r$ r$ r% �test_loc_assign_dict_to_row� s z)TestLocSeries.test_loc_assign_dict_to_rowc C sp t ddgd�}tddd�dd�|jd < tddd�dd�|jd < t tddd�dggddgd gd �}t�||� d S ) Nr r� )rM r8 r� )r� rZ )r r� r3 )rM rN )r r r rQ rR )r# ro rT r$ r$ r% �,test_loc_setitem_dict_timedelta_multiple_set� s z:TestLocSeries.test_loc_setitem_dict_timedelta_multiple_set) rV rW rX rc rt ru r r r r r r r r r r rO r� r� r{ rp r r rF r! r" r% rv r� r& r$ r$ r$ r% r � s . r )B�__doc__� collectionsr r r r r �ior r� Z dateutil.tzr �numpyrO rc Z pandas.util._test_decorators�utilZ_test_decoratorsrv Zpandasr� r r r r r r r r r r r r r r r r Zpandas._testingZ_testingrQ Zpandas.api.typesr Z pandas.tests.indexing.commonr r rY ry r� r� r� r� r� r� r� rt ru r� r� r� r� r r r r r r r$ r$ r$ r% �<module> sb H i L / 9J QZ$
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка