Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/__pycache__/test_indexing.cpython-37.pyc
Назад
B �5�g̿ � @ s� d dl m Z mZ d dlZd dlZd dlZd dlmZ d dlm m Z d dl m Z d dlZd dlmZmZmZmZmZmZmZmZmZmZ d dlmZ d dl m m Z dZ!G dd� d�Z"G d d � d �Z#d d � Z$d Z%dZ&G dd� d�Z'dS )� )�datetime� timedeltaN)�iNaT)� is_integer) � Categorical� DataFrame� DatetimeIndex�Index� MultiIndex�Series� Timestamp� date_range�isna�notnazBslice indices must be integers or None or have an __index__ methodc @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zej d1d2� �Z d3d4� Z d5d6� Z d7d8� Z ej d9d:� �Z!d;d<� Z"ej d=d>� �Z#d?d@� Z$dAdB� Z%dCdD� Z&dEdF� Z'dGdH� Z(dIdJ� Z)dKdL� Z*dMdN� Z+dOdP� Z,dQdR� Z-dSdT� Z.dUdV� Z/dWdX� Z0dYdZ� Z1d[d\� Z2e3j4�5d]d^d_g�d`da� �Z6dbdc� Z7ddde� Z8dfdg� Z9dhdi� Z:djdk� Z;dldm� Z<dndo� Z=dpdq� Z>drds� Z?dtdu� Z@dvdw� ZAdxdy� ZBdzS ){�TestDataFrameIndexingc C s� |d d� }t |j�dks t�x:|�� D ].\}}t |j�dksBt�t�|j|j�s(t�q(W x&|j�� D ]\}}|| d k sft�qfW d|ks�t�tjt dd�� |d W d Q R X d S )N� �random)�match) �len�index�AssertionError�items�tmZ equalContentsZ_series�pytest�raises�KeyError)�self� float_frame�sl�_�series�key� r" �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/test_indexing.py� test_getitem$ s z"TestDataFrameIndexing.test_getitemc C sx |� � }tj�t|��|d< tj�t|��}||d< tjtt� d�d�� |� d� W d Q R X |d }t � ||j � d S )Nz$10z@awesome_domainz 'df["$10"]')r z df["$10"])�copy�npr �randnr r r r �re�escape� __getitem__r Zassert_numpy_array_equal�values)r r �df�ad�resr" r" r# � test_getitem25 s z#TestDataFrameIndexing.test_getitem2c C s� d|d< |ddg }||ddg< t j|d |d dd� t j|d |d dd� d}tjt|d�� |ddg |dg< W d Q R X tt|j�d �}d t|�� d t|�� d �}tjt|d�� ||d< W d Q R X d S ) N�foo�E�A�BF)� check_namesz"Columns must be same length as key)r � zLength of values \(z$\) does not match length of index \(z\))r �assert_series_equalr r � ValueError�ranger r )r r �data�msgZ newcolumndatar" r" r# �test_setitem_listC s z'TestDataFrameIndexing.test_setitem_listc C s� t dtd�ddgtjd�}ddg|jdddgf< |j|jd ddgf }tddg|jtjdd�}t� ||� d |d< |d< d d g|j|jd ddgf< |j|jd ddgf }td d g|jdd �}t� ||� d S ) Nr � Ztt1Ztt2)r �columns�dtyper5 � )r> �name�0�1�2)r@ ) r r8 r& �int_�locr r r= r r6 )r r, �result�expectedr" r" r# �test_setitem_list2W s z(TestDataFrameIndexing.test_setitem_list2c s� |j d }|j |k}|�t�}|j | }|| }t�||j � tjtdd�� ||d d� W d Q R X || } t�| |� tjtdd�� || W d Q R X t ||j �}|| } t�| |� t� t ��0 |� |j d d d� �}|| } t�| |� W d Q R X x�|||gD ]�} | � � � | | dk } t� fdd�� jD �� j � jd �} x&| jD ] } | | k�rL| | | | < �qLW | j | jd �} tj| | d d � x@| jD ]6} | | j| | jk�r�| | j| | jk�s�t��q�W �q W d S ) N� zItem wrong length)r ���zBoolean array expectedr c s* i | ]"}t �� | d k� | t j�|�qS )r )r& �where�nan)�.0�c)r9 r" r# � <dictcomp>� s z>TestDataFrameIndexing.test_getitem_boolean.<locals>.<dictcomp>)r r= )r= F)Z check_dtype)r �astype�objectr Zassert_index_equalr r r7 �assert_frame_equalr Zassert_produces_warning� UserWarning�reindexZ_get_numeric_datar r= r> r )r Zmixed_float_frameZmixed_int_frame�datetime_frame�d�indexerZ indexer_objZsubindexZsubframeZ subframe_objr, ZbifZbifwrN r" )r9 r# �test_getitem_booleanf sJ z*TestDataFrameIndexing.test_getitem_booleanc C sX |� � }d|d< |d �d�|d< |d � � |d< d|d< |d �d�|d< |d � � |d< ||dk }|j}tt�d �gd t�d�gd t�d�gd d d ddddddgd�}t�||� d|j|j dd� ddgf< ||dk }|j}tt�d �gd t�d�g t�d �g t�d�g t�d �g d d ddddddgd�}t�||� d S )Nr5 r1 Zint32ZE1�F�int64ZF1r �float64� r? r2 r3 �C�D)r r< ) r% rP �dtypesr r& r> r r6 rE r )r rU r, ZcastedrF rG r" r" r# � test_getitem_boolean_casting� s( . >z2TestDataFrameIndexing.test_getitem_boolean_castingc sP t t�d��dd��� � fdd�}|dddg� |dddg� |dddg� d S )N� r< r\ c s( � | }� j � j| }t�||� d S )N)rE r r rR )�lstrF rG )r, r" r# �_checkit� s zATestDataFrameIndexing.test_getitem_boolean_list.<locals>._checkitTF)r r& �arangeZreshape)r rc r" )r, r# �test_getitem_boolean_list� s z/TestDataFrameIndexing.test_getitem_boolean_listc C sb t j�dd�}t|�� dddddgd�}||dk d 7 < ||dk d 7 < t�|j|� d S ) N� r2 r3 r] r^ r1 )r= r r5 )r& r r' r r% r �assert_almost_equalr+ )r �arrr, r" r" r# �test_getitem_boolean_iadd� s z/TestDataFrameIndexing.test_getitem_boolean_iaddc C s> t t�ddg�dgtg �d�}t�g t�}|| d||< d S )Nr r5 r2 )r= r )r r& �emptyr �array�bool)r Zblah�kr" r" r# � test_boolean_index_empty_corner� s z5TestDataFrameIndexing.test_boolean_index_empty_cornerc C s� t tj�dd�ddddgdddgd�}|jd d � }|j|jd d � }t�||� |jddg }|jt ddg� }t�||� d S ) Nr\ r< r5 rI r] r1 r? )r r= rJ ) r r& r r' �ilocrE r r rR r )r r, rF rG r" r" r# � test_getitem_ix_mixed_integer� s $ z3TestDataFrameIndexing.test_getitem_ix_mixed_integerc C s� t dddddgdddd gd dd d gd �ddd dgd�}|dg }|jd d �dgf }t�||� |dg }|jd d �dgf }t�||� d S )N)g �?g������@g������ @g @� � �$ �( rI � �+ �" r r � )�rnai���r i� ry i���i� )r= r< r5 )r ro r rR )r r, rF rG r" r" r# � test_getitem_ix_mixed_integer2� s z4TestDataFrameIndexing.test_getitem_ix_mixed_integer2c C s: t �|j|d � d}tjt|d�� |j W d Q R X d S )Nr2 z6'DataFrame' object has no attribute 'NONEXISTENT_NAME')r )r r6 r2 r r �AttributeErrorZNONEXISTENT_NAME)r r r: r" r" r# � test_getattr s z"TestDataFrameIndexing.test_getattrc C s0 t dditd�d�}d|_|jdk�� s,t�d S )N�foobarr5 rI )r rf )r r8 r} �allr )r r, r" r" r# �test_setattr_column s z)TestDataFrameIndexing.test_setattr_columnc C s� |d d d d� }||d< d|ks&t �t|�dks6t �t|�dksFt �t�t�|jtjgd f��}t||jdd�}t � |d |� |d }||d< t j ||d dd � tj � t|��}||d < |d |k� � s�t �d |d < |d d k� � s�t �d |d< |d d k� � �st �d|d< |d dk� � �s,t �|d d� }d}tjtj|d�� ddg|d< W d Q R X |d jtjk�szt �|d ddgk� � �s�t �d S )Nr2 r? Zcol5� rx )r r@ Zcol6F)r4 Zcol9rf Zcol7g ��Q� @Zcol0r0 Zcol8zC\nA value is trying to be set on a copy of a slice from a DataFrame)r rB rC Zcol10)r r r& ZravelZ column_stackr+ rL r r r r6 r r' r~ r r �com�SettingWithCopyErrorr> �object_)r r r �exprh Zsmallerr: r" r" r# � test_setitem s4 z"TestDataFrameIndexing.test_setitemc C sb t ddgg�}tj|jd< t tjtjgg�}t�||� t ddgg�}tj|jd< t�||� d S )Nr )r r& rL ro r rR rE )r r, rG r" r" r# � test_setitem2: s z#TestDataFrameIndexing.test_setitem2c C s |� � }|j}d||d dk< d||d d �df dk< t�|j|� |d dk}|�|jd d d� �}d||< d||d d �df dk< t�|j|� d||dk< d||dk< t�|j|� d||dk< d||dk< t�|j|� d||d d� dk < t�|d d� |d d� dk d� t�|j|� d||d d d� dk< d||dk< t�|j|� d }tj t |d �� d||d < W d Q R X |t� |�k}|� � }tj ||t� |�k< tj |j|j< t� ||� |� � }|d ||t� |�k< t�|j|j|jd � t� ||� d S ) Nr\ r2 r rJ r5 rf r? r< z;Must pass DataFrame or 2-d ndarray with boolean values only)r )r% r+ r rg rT r r& Zputmaskr r � TypeError�absrL rR )r r r, r+ r r: �maskrG r" r" r# �test_setitem_booleanE sD " z*TestDataFrameIndexing.test_setitem_booleanc C s |d � d�|d< |d jtjks&t�d|d< |d jtjksBt�t�t|��|d< t|d jjtj �slt�d|d< d|d< |d jtjks�t�d|d< d|d< |d jtj ks�t�d|d< |d jtjks�t�d |d< |d jtjks�t�d|d< |d jtj k�s t�d S ) Nr^ �i8r r3 �barr0 g @Z somethingr? ) rP r> r& rZ r rd r � issubclass�type�integerr[ )r r r" r" r# �test_setitem_castw s$ z'TestDataFrameIndexing.test_setitem_castc C s� t dddgdddgd�t�d�d �}|d = dddg|d < d |ksDt�t|j�d ksVt�d |d < d|d< d|d< d|t�� < d|t�� < t |jd �}d|d < d|d < t|j�d ks�t�|j j tj ks�t�d|d< |d j tj ks�t�d|d< |d j tj ks�t�d|d < d|d d k�s t�t t�d�d �}d|d < d|d< |d= d|d< |d j tj k�s`t�dddg|d< |d j tj k�s�t�d S )Ng �?g @g @�a�brN )r3 r] r< )r r3 r? Z beginningr2 r0 r1 r� r^ �dateg @r5 r] r rB rC �3Z coercible)r r& rd r r r= r �nowr r+ r> r� rZ r[ )r r r, �dmr" r" r# �test_setitem_corner� s: $ z)TestDataFrameIndexing.test_setitem_cornerc C s� dddgdgd t j�d�d�}t|�}||d dk j}d|j|dgf< d|j|dgf< |jd dksnt�|jd dks�t�d S ) Nr} r� � r )�title�cruftr� r r� )r5 r� )r5 r� )r& r r r rE r )r r9 r, �ixr" r" r# �test_setitem_corner2� s z*TestDataFrameIndexing.test_setitem_corner2c s� ddl m� ttd�td�d�}t� fdd�td�D �td�d�}tdd d gtd�d�}t�d�|d< t|j�dksxt �||d < t|j�dks�t �||d < t|j�dks�t �|d j tj ks�t �d S ) Nr )�Decimalr< )r r= c s g | ] }� d ��qS )r5 r" )rM r )r� r" r# � <listcomp>� s z<TestDataFrameIndexing.test_setitem_ambig.<locals>.<listcomp>)r r0 �bzr�bazr5 r? ) �decimalr� r r8 r r& �onesr r= r r> r� )r r� Zcoercable_seriesZuncoercable_seriesr" )r� r# �test_setitem_ambig� s "z(TestDataFrameIndexing.test_setitem_ambigc C st |d |d < t j|jd d �df |d dd� t j|jd d �d f |d dd� t j|d |d dd� t|� d S )Nr2 rJ F)r4 )r r6 ro rE �repr)r r r" r" r# �test_setitem_None� s z'TestDataFrameIndexing.test_setitem_Nonec C sP t dddgdddgddd gd ��}|�� }|j|j|j�� d f< t�||� d S ) NrB rC r� Z11Z22Z33Z111Z222Z333)r� r� rN r� )r r% r� rE r� r r rR )r r, rF r" r" r# �&test_loc_setitem_boolean_mask_allfalse� s z<TestDataFrameIndexing.test_loc_setitem_boolean_mask_allfalsec C sV t tj�dd��}|jd dd� }tj|jd dd�< t|jd dd� �j�� sRt �d S )NrI rf � r? ) r r& r r' ro rL r r+ r~ r )r r, rF r" r" r# �&test_getitem_fancy_slice_integers_step� s z<TestDataFrameIndexing.test_getitem_fancy_slice_integers_stepc C sF t tj�dd�tddd�d�}|�� }d|jdd�< |jdd� dkj�� sPt �|�� }d|jdd �< |jdd � dkj�� s�t �|jdd � }|j dd � }|� dd d dg�}t � ||� t � ||� |jttd��ttdd��d d d � }tjtd d�� |j dd � W d Q R X tjtd d�� d|j dd �< W d Q R X d S )NrI rf r r r? )r r\ r< rq � r� rJ z^3$)r )r r& r r' r8 r% ro r+ r~ r rE rT r rR �listr r r )r r, �cprF �result2rG �df2r" r" r# �,test_getitem_setitem_integer_slice_keyerrors� s" *zBTestDataFrameIndexing.test_getitem_setitem_integer_slice_keyerrorsc C s� |j d d �dd �f }|d jtjks*t�|j d d �dd �f }d}tjtj|d�� d|d< W d Q R X |d dk� � s|t�d S )N���r^ zC\nA value is trying to be set on a copy of a slice from a DataFrame)r g @r] r\ ) ro r> r& r[ r r r r� r� r~ )r r �float_string_frameZslicedr: r" r" r# � test_fancy_getitem_slice_mixed s z4TestDataFrameIndexing.test_fancy_getitem_slice_mixedc C s� t �� }|jddg \}}|j||� }|||� }|dd� }t �||� t �||� |�� }d|j||�< |�� }d|||�< |�� }d|dd�< t �||� t �||� d S )Nrf rI rq r )r ZmakeTimeDataFramer rE rR r% )r r, �start�endrF r� rG r" r" r# �"test_getitem_setitem_non_ix_labels) s z8TestDataFrameIndexing.test_getitem_setitem_non_ix_labelsc C sF t tj�dd��}|j|jdkd d �f }|�dg�}t�||� d S )Nr< r? r ) r r& r r' rE r rT r rR )r r, �rsZxpr" r" r# �test_ix_multi_take= s z(TestDataFrameIndexing.test_ix_multi_takec C sX |}|j }xH|jD ]>}|| }x0|jd d d� D ] }|||f || ks0t�q0W qW d S )Nrf )rE r= r r )r r �fr� �col�ts�idxr" r" r# �test_getitem_fancy_scalarK s z/TestDataFrameIndexing.test_getitem_fancy_scalarc C s� |}|� � }|j}xtt|j�D ]f\}}|| }xT|jd d d� D ]@}|j�|�} tj�� } | |j | |f< | |||f< t � ||� q@W q W d S )Nrf ) r% rE � enumerater= r Zget_locr& r r' r+ r rR ) r r r� rG r� �jr� r� r� �i�valr" r" r# �test_setitem_fancy_scalarU s z/TestDataFrameIndexing.test_setitem_fancy_scalarc C s |}|j }|jddgd�}|d d �ddddgf }t�||� |j|jdd� ddgd�}||jdd� ddddgf }t�||� |j|jd k}|j|j| d �}|| }t�||� ||d d �f }t�||� |||jd d � f }|j|j| d dgd�}t�||� d S ) Nr3 r^ )r= FTrf rI )r r= � )r r? r] )rE rT r rR r r= )r r r� r� rG rF Zboolvecr" r" r# �test_getitem_fancy_booleanf s" z0TestDataFrameIndexing.test_getitem_fancy_booleanc C s� |� � }|� � }|d dk}d|j|< d|j|j< t�||� |� � }|� � }d|j|ddgf< d|j|jd d�f< t�||� d S )Nr2 r g r3 r? )r% rE r+ r rR )r r �framerG r� r" r" r# �test_setitem_fancy_boolean} s z0TestDataFrameIndexing.test_setitem_fancy_booleanc C sx |j dddg }|j|jdddg }t�||� |j d d �dddgf }|jd d �|jdddg f }t�||� d S )Nr5 r\ r� r? r )ro rE r r rR r= )r r rF rG r" r" r# �test_getitem_fancy_ints� s z-TestDataFrameIndexing.test_getitem_fancy_intsc C st |d d d d� dk}|j | }|j |d d d� }t�||� |�� }|�� }d|j |< d|j |< t�||� d S )Nr2 rJ r5 r )rE r rR r% )r r r� rF rG r� r" r" r# �'test_getitem_setitem_boolean_misaligned� s z=TestDataFrameIndexing.test_getitem_setitem_boolean_misalignedc C s� t tj�dd��}t�dddg�}t�ddg�}|j||f }|jddgdgf }t�||� |�� }d|jt�dddg�t�ddg�f< d|jddgdgf< t�||� d S )Nr< r? TFr r5 rf ) r r& r r' rk rE r rR r% )r r, Zk1Zk2rF rG r" r" r# �"test_getitem_setitem_boolean_multi� s $z8TestDataFrameIndexing.test_getitem_setitem_boolean_multic C s� t dddddg�}ttj�dd�|d�}|jdd� }|�ddddg�}t�||� t |�dksdt �|jdd� }|�ddg�}tj||dd� t |�dks�t �|jdd� }|�d d g�}t�||� t |�dks�t �|jd d� }|j d d� }t�||� d |jd d�< |d d� }|d k� � � � �s0t �t d dddd g�}ttj�dd�|d�}d}t jt|d�� |j d d� W d Q R X |j dd� }|�d g�}t�||� t |�d k�s�t �|�� }t jttd�� d |j d d�< W d Q R X t jt|d�� |j d d� d k}W d Q R X |j� � �s(t �|j d d � |j d d � kj� � �sPt �|�� }d |j dd�< |j dd� d kj� � �s�t �|j d d� |j d d� kj� � �s�t �|jd d� }|}t�||� t |�dk�s�t �|jdd� }|�dddd g�}t�||� t |�dk�st �|jdd� }|�d g�}t�||� t |�d k�sRt �|jd d � }|�d dddd g�}t�||� t |�dk�s�t �|�� }d |jd d �< |jd d � }|d kj� � �s�t �d S )Ng �?r? r< r\ rf )r F)Zcheck_index_typeg @g @r5 r g �?g @g @g @zWcannot do positional indexing on Float64Index with these indexers \[1.0\] of type float)r g�������?g ףp= @)r r r& r r' rE rT r rR r r ro r~ r r r� r% � _slice_msgr+ )r r r, rF rG r: r� r" r" r# �!test_getitem_setitem_float_labels� sx ( ( z7TestDataFrameIndexing.test_getitem_setitem_float_labelsc C s t tj�dd�dddddgdd d gd �}td �|d < |j}tt�d�gd t�d�g dd d d gd�}t� ||� t |j d< t |j d �r�t �|d jtjks�t �|j d t ks�t �tj|j d< t |j d �s�t �tj|j dd d �f< t |j dd d �f ��� �rt �d S )Nrf r< r� r� rN rV �er0 r� r� )r r= Z20010102� timestampr[ zdatetime64[ns])r )r� r� )rN r� )r r& r r' r r_ r r> r r6 r rE r r r� rL r~ )r r, rF rG r" r" r# �)test_setitem_single_column_mixed_datetime s$ z?TestDataFrameIndexing.test_setitem_single_column_mixed_datetimec C s� t ddddddgtddd�dddtddd�td dd�gd ��}t dtd �td �d �}tj|d< tddd�|jd< d|jd< d|jddgdf< t�dt� d�gdt� d�gg�}||jddgddgf< t � ||� d S )Nr � � i� r5 �x�yi� i� )r� r� �abr� )r= r r� )r r� )r5 r� )r� r� r? r< z2013-01-01T00:00:00z2014-01-01T00:00:00r\ rf r� ) r r r� r8 �pd�NaTrE r& rk � datetime64r rR )r rG r, r2 r" r" r# �test_setitem_mixed_datetime0 s$ z1TestDataFrameIndexing.test_setitem_mixed_datetimec C sn |j |jd d� ddgf }|j|j |jd d �ddgf< |j |jdd � ddgf j}|j}t�||� d S )Nr? r2 r3 ���)rE r r+ r rg )r r �piecerF rG r" r" r# �test_setitem_frame_floatM s z.TestDataFrameIndexing.test_setitem_frame_floatc C s~ |� � }tddgddgg|jdd� ddgd �}|jtd d� ddgf}||j|< t�|j|jdd� ddgf j|j� d S ) Ng �?g @g @g @r r? r2 r3 )r r= )r% r r �slicerE r rg r+ )r r� r� r� r! r" r" r# �test_setitem_frame_mixedT s & z.TestDataFrameIndexing.test_setitem_frame_mixedc C s� |� � }tddgddgddgddggt|jd d � �d d g d dgd�}|jtd d � d dgf}||j|< t�|j|jd d � d dgf j|jd d � � d S )Ng �?g @g @g @g @g @g @g @r r? r0 r� r2 r3 )r r= ) r% r r� r r� rE r rg r+ )r r� r� r� r! r" r" r# �'test_setitem_frame_mixed_rows_unaligned` s z=TestDataFrameIndexing.test_setitem_frame_mixed_rows_unalignedc C s� |� � }|j|jd d� dgf }|jdd � |_|jtdd � ddgf}||j|< tj|d< t�|j|jdd � ddgf j|j� d S )Nr? r2 r� r3 ) r% rE r r� r& rL r rg r+ )r r� r� r� r! r" r" r# �&test_setitem_frame_mixed_key_unalignedn s z<TestDataFrameIndexing.test_setitem_frame_mixed_key_unalignedc C st |� � }|j|jd d� ddgf }|jtdd � ddgf}|j|j|< t�|j|jdd � ddgf j|j� d S )Nr? r2 r3 r� )r% rE r r� r+ r rg )r r� r� r� r! r" r" r# � test_setitem_frame_mixed_ndarrayx s z6TestDataFrameIndexing.test_setitem_frame_mixed_ndarrayc C s� t dddgdddggddd gd �}|�� }|jd d �ddgf d |jd d �ddgf< |jddgd �}|d 7 }|d |d < t�||� d S ) Nr5 r? r0 r< r\ r� r2 r3 r] )r= g �?)r r% rE rT r rR )r r, r� rG r" r" r# �test_setitem_frame_upcast� s ", z/TestDataFrameIndexing.test_setitem_frame_upcastc C s� |j |jd d� ddgf }|jdd � |_ddg|_||j |jdd � ddgf< |j |jdd � ddgf j}|j}t�||� d S )Nr? r2 r3 r� )rE r r= r+ r rg )r r r� rF rG r" r" r# �test_setitem_frame_align� s z.TestDataFrameIndexing.test_setitem_frame_alignc C s� t tj�dd�dddddgd�}|jd }|d d� }t�||� |jd }|jddg }t�||� |jd }|jd }t�||� d S ) Nrf r< r0 r� r� )r r? r\ ) r r& r r' rE r rR ro r6 )r r, rF rG r" r" r# �"test_getitem_setitem_ix_duplicates� s z8TestDataFrameIndexing.test_getitem_setitem_ix_duplicatesc C s� t tj�dd�dddddgd�}|jdg }|jddg }t�||� |j|d d k }||d d k }t�||� |j|d d k }||d d k }t�||� d S ) Nrf r< r0 r� r� )r r? r\ r5 r )r r& r r' rE ro r rR )r r, rF r� r" r" r# �+test_getitem_ix_boolean_duplicates_multiple� s zATestDataFrameIndexing.test_getitem_ix_boolean_duplicates_multiple� bool_valueTFc C sl t ddddgi�}|� d�}tjt|d�� |j| W d Q R X d}tjt|d�� d|j|< W d Q R X d S ) Nr� r5 r? r< z7: boolean label can not be used without a boolean index)r z.cannot use a single bool to index into setitemr )r r r r rE )r r� r, �messager: r" r" r# �%test_getitem_setitem_ix_bool_keyerror� s z;TestDataFrameIndexing.test_getitem_setitem_ix_bool_keyerrorc C s� d|d< t |d jjttjf�s$t�|j|jd df }t |�sDt�t ddgi�}d|d< |jd }t |�spt�t dgd gdd �}|jd gdf }t � ||� d S ) Nr5 r1 rf r� g�G�z��?i� r� )r r� r )r@ )r� r> r� �intr& r� r rE r r r r r r6 )r r rF r, rG r" r" r# �"test_single_element_ix_dont_upcast� s z8TestDataFrameIndexing.test_single_element_ix_dont_upcastc C s� t tj�dd�tddd�d�}|jd }|jd }t�||� |jd }|jd }t�||� |jt dd� }|jdd � }t� ||� |jdddd g }|� |j dddd g �}t� ||� d S ) NrI r\ r r r? )r r5 r� r� r� ) r r& r r' r8 ro rE r r6 r� rR rT r )r r, rF r� rG r" r" r# � test_iloc_row� s z#TestDataFrameIndexing.test_iloc_rowc C s� t tj�dd�tddd�d�}|�� }|jtdd� }d}tj t j |d �� d |d< W d Q R X |d �� }|szd |dd�< t � |d |� d S ) NrI r\ r r r? )r r� zC\nA value is trying to be set on a copy of a slice from a DataFrame)r g )r r& r r' r8 r% ro r� r r r� r� r r6 )r �using_array_managerr, �original�subsetr: Zexp_colr" r" r# �test_iloc_row_slice_view� s z.TestDataFrameIndexing.test_iloc_row_slice_viewc C s� t tj�dd�tddd�d�}|jd d �df }|jd d �df }t�||� |jd d �df }|jd d �df }t�||� |jd d �t dd�f }|jd d �dd �f }t� ||� |jd d �dddd gf }|j |j dddd g d�}t� ||� d S ) Nr\ rI r r r? )r= r5 r� r� r� ) r r& r r' r8 ro rE r r6 r� rR rT r= )r r, rF r� rG r" r" r# � test_iloc_col� s z#TestDataFrameIndexing.test_iloc_colc C s� t tj�dd�tddd�d�}|�� }|jd d �tdd�f }|s�d}tj t j |d �� d |d< W d Q R X |d dk� � s�t �n(d |d< |d dk� � s�t �t�||� d S ) Nr\ rI r r r? )r= r� zC\nA value is trying to be set on a copy of a slice from a DataFrame)r g )r r& r r' r8 r% ro r� r r r� r� r~ r r rR )r r� r, r� r� r: r" r" r# �test_iloc_col_slice_view s z.TestDataFrameIndexing.test_iloc_col_slice_viewc C s� t tdddd�tdddd�d�}|jdtdddd�d�}td|dd gd �}t�d d d d d d g�}d |j|| df< tddddd d gddddddgd�|d�}t�||� td|dd gd �}|j|| df d 7 < t�||� d S )Ni� r5 )�year�month�dayrf )r� r� )rE �itemr r2 r3 )r r= FTr� )r2 r3 )r ) r r �insertr r& rk rE r rR )r Ztranger, Zbool_idxrG r" r" r# �test_loc_duplicates$ s ( z)TestDataFrameIndexing.test_loc_duplicatesc C s� t tdddd�dd�}td|i�}|ddd g |d< t�|d |� td|i�}|ddd g |jddd gdf< t�|d |� d S ) Nz 2015-01-01r< �utc)�periods�tz�dates)r@ r5 r r? )r r r r r6 rE )r �columnr, r" r" r# �4test_setitem_with_unaligned_tz_aware_datetime_column? s zJTestDataFrameIndexing.test_setitem_with_unaligned_tz_aware_datetime_columnc C s� t dd�}ttddd�d�}tjd| gd dd�|d < tjd | gd dd�|jd d �d f< tjd | gd dd�|j|jd d � d f< tjd| gd dd�|jd d �df< tjd| gd dd�|j|jd d � df< t�d�|d< tjd| gd dd�|j|jd d� df< tdd d�|j|jd d �df< t�d�|d< |j }t t� d�gd t� d�gd t d�d�}t �||� d S )Nr5 )�hoursZ20130101r\ )r� )r zm8[ns])r> r2 r? r3 r< r] r^ rf r1 r� rY r� rJ r� �G�Hztimedelta64[ns]zdatetime64[ns]ZABCDEFGH)r r r r& rk rE r Z timedelta64r� r_ r r> r� r r6 )r Zone_hourr, rF rG r" r" r# �,test_loc_setitem_datetimelike_with_inferenceL s &,&,, zBTestDataFrameIndexing.test_loc_setitem_datetimelike_with_inferencec C s t tjtjtjtjtjd�tjdtjddd�tjtjdtjtjd�tjtjtjtjtjd�dtjtjtjtjd�dddddd�d��}|�� }d||d k< |�� }d|jd < d|jd < d|jd < d|jd < t�||� d|d< d}tjt |d�� d||d k< W d Q R X d S )N)�# rt rv �1 �2 gj�B�z��?g% t��?g���k��?g�������?r r5 )r r5 r? r< r\ r� g333333�?)rt r5 )r� r5 )r r5 )r� r\ �testr0 z1not supported between instances|unorderable types)r ) r r& rL r% rE r rR r r r� )r r, r� rG r: r" r" r# �#test_getitem_boolean_indexing_mixedb s> z9TestDataFrameIndexing.test_getitem_boolean_indexing_mixedc C s2 t ddddgddddgdddd gddddgdddd ggd �}|jddddgd �}tjtd d �� |d d �df W d Q R X ttd�dd�}tddgddggddgddggd dgd�}t ddgdd gg||d�}|jd d �t d �dff }t � ||� d}ttd�dd�}t ddg||d�}|d }t � ||� d S )Nr� rN r� r� r r5 r? r< r\ )r= r9 )r r= r+ zunhashable type)r )r@ )Zlevels�codes�names)r= r )r� r )r r@ )r Z pivot_tabler r r� r r8 r rE r� r rR r r6 )r r, �dgr r= rG rF r@ r" r" r# �test_type_error_multiindex� s" 0( z0TestDataFrameIndexing.test_type_error_multiindexc C sh t t�d�tj�t�d��d�}|jd d �df }|d }t� ||� |j d d �df }t� ||� d S )N)r< r\ rf )r= r g �?) r r& r� r� Z IntervalIndexZ from_breaksrd ro r r6 rE )r r, rG r. r" r" r# �,test_getitem_interval_index_partial_indexing� s zBTestDataFrameIndexing.test_getitem_interval_index_partial_indexingc C sZ t ddgtd�}t�d�t�d�d�|jd< t t�d�gt�d�gd��}t�||� d S )Nr� r� )r= r> )r? )r? r? )r� r� r )r rQ r& ZzerosrE r rR )r r, rG r" r" r# � test_setitem_array_as_cell_value� s z6TestDataFrameIndexing.test_setitem_array_as_cell_valueN)C�__name__� __module__� __qualname__r$ r/ r; rH rX r` re ri rn rp rz r| r r� r� r� r� r� r� r� r� r� r� r� �tdZ skip_array_manager_invalid_testr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r �mark� parametrizer� r� r� r� r� r� r� r� r� r r r r r" r" r" r# r # sv <# * 2 * W% .r c @ s e Zd Zdd� ZdS )�TestDataFrameIndexingUInt64c C s� |}|d � d�}d|jks t�||d< t�|d t|dd�� d|jksNt�d|d< ||d< t�|d t|dd�� |d= |�� }tj|j d< tj|j d< |d }t�t |�td d d gdd�� t�|j tt � d �t � d �t � d �gdddgd �� d S )Nr2 r0 r] )r@ r^ )r5 r5 )r5 r? r3 TFZuint64�O)r )�renamer= r r r6 r r% r� r� ro r r_ r& r> )r Z uint64_framer, r� r� rF r" r" r# r� � s( z(TestDataFrameIndexingUInt64.test_setitemN)r r r r� r" r" r" r# r � s r c C s� t ddgtddd�tjddd�d��}|jd }t|jd t�sDt�t|jd tj �sZt�|j d }t|jd t�sxt�t|jd tj �s�t�|j ddd�}t|jd t�s�t�t|jd tj �s�t�| r�d S |j }|� � |�d�}t|d t�s�t�t|d tj ��st�|j|jd }|jd k�s.t�|�d �}t|t��sHt�|j|jd }|jd k�sht�|�d �}t|tj ��s�t�d S ) Nr5 r? �2000)r� z1 Day)r2 r3 r] r )ZaxiszM8[ns])r r zm8[ns])r r r� Ztimedelta_rangerE � isinstancer+ r r Z Timedeltaro ZxsZ_mgrZ_rebuild_blknos_and_blklocsZfast_xs�blocksZblknosr> �iget)r� r, ZserZmgrrh Zblkr� r" r" r# �/test_object_casting_indexing_wraps_datetimelike� s8 r zMCannot setitem on a Categorical with a new category, set the categories firstzCCannot set a Categorical with another, without identical categoriesc @ s e Zd Zejdd� �Zejdd� �Zejdd� �Zejdd� �Zejd d � �Z ej � d e j e jg�d d � �Zej � d e j e je je jg�dd� �Zej � d e j e jg�dd� �Zej � d e j e jg�dd� �Zej � d e j e jg�dd� �Zej � d e j e jg�dd� �ZdS )� TestLocILocDataFrameCategoricalc C s\ t dddddddgddgd�}tdddddd d g�}d d d d d d d g}t||d �|d �}|S )Nr� r� )� categories�hr� r� rm �l�m�nr5 )�catsr+ )r )r r r )r r r� r+ �origr" r" r# r s z$TestLocILocDataFrameCategorical.origc C s\ t dddddddgddgd�}tdddddd d g�}d d d d d d d g}t||d �|d�}|S )Nr� r� )r r r� r� rm r r r r5 r? )r r+ )r )r r r )r Zcats1Zidx1Zvalues1�exp_single_rowr" r" r# r s z.TestLocILocDataFrameCategorical.exp_single_rowc C s\ t dddddddgddgd�}tdddddd d g�}d d d d d d d g}t||d �|d�}|S )Nr� r� )r r r� r� rm r r r r5 r? )r r+ )r )r r r )r Zcats2Zidx2Zvalues2� exp_multi_rowr" r" r# r # s z-TestLocILocDataFrameCategorical.exp_multi_rowc C s\ t dddddddgddgd�}tdddddd d g�}d d d d d d d g}t||d �|d �}|S )Nr� r� )r r r� r� rm r r r r5 )r r+ )r )r r r )r Zcats3Zidx3Zvalues3�exp_parts_cats_colr" r" r# r! - s z2TestLocILocDataFrameCategorical.exp_parts_cats_colc C s\ t dddddddgddgd�}tdddddd d g�}d d d d d d d g}t||d �|d �}|S )Nr� r� )r r r� r� rm r r r r5 )r r+ )r )r r r )r Zcats4Zidx4Zvalues4�exp_single_cats_valuer" r" r# r" 6 s z5TestLocILocDataFrameCategorical.exp_single_cats_valuerW c C s� |� � }tdd�}|tjkr&tdd�}ddgddgg||�|d d �f< t�||� |� � }tjttd��& ddgddgg||�|d d �f< W d Q R X d S )Nr? r\ r� rm r� )r rN ) r% r� r rE rR r r r7 �msg1)r r r rW r, r! r" r" r# �#test_loc_iloc_setitem_list_of_listsA s zCTestLocILocDataFrameCategorical.test_loc_iloc_setitem_list_of_listsc C sr |� � }d}|tjtjgkr0|jd |jd f}d||�|< t�||� tjt t d�� d||�|< W d Q R X d S )N)r? r r? r r� )r rN ) r% r rE �atr r= rR r r r7 r# )r r r" rW r, r! r" r" r# �7test_loc_iloc_at_iat_setitem_single_value_in_categoriesQ s zWTestLocILocDataFrameCategorical.test_loc_iloc_at_iat_setitem_single_value_in_categoriesc C sJ |� � }|jdk}d}|tjkr*|j| }d||�||f< t�||� d S )Nr� r r� )r% r r rE r= rR )r r r" rW r, r� r! r" r" r# �5test_loc_iloc_setitem_mask_single_value_in_categoriesd s zUTestLocILocDataFrameCategorical.test_loc_iloc_setitem_mask_single_value_in_categoriesc C sz |� � }d}|tjkr |jd }ddg||�|d d �f< t�||� tjttd�� ddg||�|d d �f< W d Q R X d S )Nr? r� )r rN ) r% r rE r rR r r r7 r# )r r r rW r, r! r" r" r# �2test_loc_iloc_setitem_full_row_non_categorical_rhss s zRTestLocILocDataFrameCategorical.test_loc_iloc_setitem_full_row_non_categorical_rhsc C s� |� � }tdd�df}|tjkr4tdd�|jd f}tddgddgd�}|||�|< t�||� ttd �td �d�}tj t t d �� |||�|< W d Q R X ttd �td �d�}tj t t d �� |||�|< W d Q R X d S ) Nr? r\ r r� rm r� r� )r Zbb�abc)r �cc) r% r� r rE r= r rR r� r r r7 �msg2) r r r! rW r, r! �compatZ semi_compatZincompatr" r" r# �1test_loc_iloc_setitem_partial_col_categorical_rhs� s zQTestLocILocDataFrameCategorical.test_loc_iloc_setitem_partial_col_categorical_rhsc C s~ |� � }tdd�df}|tjkr4tdd�|jd f}ddg||�|< t�||� tjtt d�� ddg||�|< W d Q R X d S ) Nr? r\ r r� rm r� )r rN ) r% r� r rE r= rR r r r7 r# )r r r! rW r, r! r" r" r# �)test_loc_iloc_setitem_non_categorical_rhs� s zITestLocILocDataFrameCategorical.test_loc_iloc_setitem_non_categorical_rhsN)r r r r Zfixturer r r r! r" r r r rE ro r$ r% Ziatr& r' r( r- r. r" r" r" r# r s & r )(r r r( �numpyr& r Z pandas._libsr Z pandas.util._test_decorators�utilZ_test_decoratorsr Zpandas.core.dtypes.commonr Zpandasr� r r r r r r r r r r Zpandas._testingZ_testingr Zpandas.core.common�core�commonr� r� r r r r# r+ r r" r" r" r# �<module> s4 0 '!+
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка