Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/__pycache__/test_scalar.cpython-37.pyc
Назад
B �5�g�& � @ s� d Z ddlmZmZ ddlZddlZddlmZmZm Z m Z m Z ddl m Z ddlmZ G dd� de�ZG dd � d �Zd d � Zd d � Zdd� ZG dd� d�ZdS )z, test scalar indexing, including at and iat � )�datetime� timedeltaN)� DataFrame�Series� Timedelta� Timestamp� date_range)�Basec @ s@ e Zd Zej�dddg�dd� �Zej�dddg�dd� �ZdS ) � TestScalar�kind�series�framec s� d� fdd� }t � |�}x&|d |d gD ]}||ddd� q*W xP|d |d |d gD ]6}|d k rXd }tjt|d �� � �|d� W d Q R X qXW x4|d |d |d |d |d gD ]}||d� q�W d S )NFc sP | d k rL� � | |�}x6|D ].}t| |�| }� �|| ||�}t�||� qW d S )N)�generate_indices�getattr� get_value�tm�assert_almost_equal)�f�func�values�indices�i�result�expected)�self� �W/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/test_scalar.py�_check s z.TestScalar.test_at_and_iat_get.<locals>._check�ints�uints�iatT)r �labels�ts�floatsz1iAt based indexing can only have integer indexers)�match�at)F)r �pytest�raises� ValueError� check_values)r r r �dr �msgr )r r �test_at_and_iat_get s (z TestScalar.test_at_and_iat_getc s� d� fdd� }t � |�}x&|d |d gD ]}||ddd� q*W xN|d |d |d gD ]4}|d k rXd }tjt|d �� ||d� W d Q R X qXW x4|d |d |d |d |d gD ]}||d� q�W d S )NFc sP | d k rL� � | |�}x6|D ].}dt| |�|< � �|| ||�}t�|d� qW d S )N� )r r r r r )r r r r r r )r r r r 3 s z.TestScalar.test_at_and_iat_set.<locals>._checkr r r T)r r! r" r# z1iAt based indexing can only have integer indexers)r$ r% )F)r r& r' r( )r r r r* r r+ r )r r �test_at_and_iat_set1 s (z TestScalar.test_at_and_iat_setN)�__name__� __module__� __qualname__r& �markZ parametrizer, r. r r r r r s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� TestScalar2c s� t ddd�}ttj�dd�|ddddgd �}|d � � j|d }� jd }||ksXt�td d gd d�� t d �}x2� fdd�� fdd�gD ]}|� }||ks�t�q�W tddgdd�� t d�}x2� fdd�� fdd�gD ]}|� }||ks�t�q�W d S )Nz1/1/2000� )�periods� �A�B�C�D)�index�columns� z 2014-01-01z 2014-02-02zdatetime64[ns])�dtypec s � j d S )Nr- )r r )�sr r �<lambda>` � z2TestScalar2.test_at_iat_coercion.<locals>.<lambda>c s � j d S )Nr- )�ilocr )r? r r r@ ` rA z1 daysz2 daysztimedelta64[ns]c s � j d S )Nr- )r r )r? r r r@ g rA c s � j d S )Nr- )rB r )r? r r r@ g rA ) r r �np�random�randnr% r �AssertionErrorr r r )r �dates�dfr Zxpr �rr )r? r �test_at_iat_coercionP s z TestScalar2.test_at_iat_coercionc C s d S )Nr )r r r r �test_iat_invalid_argsk s z!TestScalar2.test_iat_invalid_argsc C s t td�dddddgdd�}|jd }|dks2t�|jd }|dksHt�d}tjt|d�� |jd W d Q R X d }tjt|d�� |jd W d Q R X |jddg }t ddgddgdd �}t� ||� |� � }|jd }t dd gdd�}t� ||� |jd }|dk�st�d S )Nr= r- � � Zint64)r; r> z0index 10 is out of bounds for axis 0 with size 5)r$ � z1index -10 is out of bounds for axis 0 with size 5i����)r> r )r; �name)rL r ) r �rangerB rF r r& r' � IndexErrorr �assert_series_equalZto_frame)r r? r r+ r rH r r r �test_imethods_with_dupsn s( z#TestScalar2.test_imethods_with_dupsc C s� t j�d��dd�}t|ddgd�}|jd }|jd }t�||� |j jd }t�||� d|jd < t d d gddgd d �}t�|jd |� d S ) N� rM rL r7 )r< )r r7 r )r7 r )r- r7 g @r- )r; rO ) rC rD rE Zreshaper r% rB r rR �Tr )r ZarrrH r r r r r �!test_frame_at_with_duplicate_axes� s z-TestScalar2.test_frame_at_with_duplicate_axesc C s� t ddgtdddddd�tddd d d d�gd ��}|d j�d�|d < tddd�}|jd }||ksht�|jd }||ks~t�d S )NZJohnZAndersoni� rM � � �8 rL � � rN )rO �dater\ z Asia/Shanghaiz2017-03-13 13:32:56+0800)�tz)r r\ )r r �dtZ tz_localize�locrF r% )r rH r r r r r � test_at_getitem_dt64tz_values� s z)TestScalar2.test_at_getitem_dt64tz_valuesc C s� t dddddgdddddgd �}x:|�� D ].\}}|j| |j| krR|ks*n t�q*W x>tt|��D ].}|j| |j| kr�|d ksjn t�qjW t j t d d �� |jd W d Q R X t j t d d �� |jd W d Q R X d S ) Nr- rL rM r6 r= �a�b�c)r; z^4$)r$ ) r �itemsr% r_ rF rP �lenr rB r& r' �KeyError)r r? �el�itemr r r r �'test_mixed_index_at_iat_loc_iloc_series� s *.z3TestScalar2.test_mixed_index_at_iat_loc_iloc_seriesc C s. t dddddgdddd d ggd d d ddgd�}xX|�� D ]L\}}xB|�� D ]6\}}|j||f |j||f krz|ksJn t�qJW q8W xXtd�D ]L}xFtd�D ]:}|j||f |j||f kr�|d | ks�n t�q�W q�W t j t dd�� |jd W d Q R X t j t dd�� |jd W d Q R X d S )Nr r- rL rM r6 r= rT � r4 � ra rb rc )r< z^3$)r$ )r rM ) r Ziterrowsrd r% r_ rF rP r rB r& r' rf )r rH ZrowIdx�rowrg rh r r r r �*test_mixed_index_at_iat_loc_iloc_dataframe� s ,6>z6TestScalar2.test_mixed_index_at_iat_loc_iloc_dataframec C sF t ddgddgd��}d |jd< t d dgddgd��}t�||� d S )Nr r- r6 r= )ra rb )r r )r r r Zassert_frame_equal)r r r r r r �'test_iat_setter_incompatible_assignment� s z3TestScalar2.test_iat_setter_incompatible_assignmentc C sh t ddgddgg�}|t�d� }tddgdd�}t�||� tddg�}|t�d� }|dksdt�d S )Nr- rL rM r6 r )rO )r rC �arrayr r rR rF )r rH r r r? r r r � test_getitem_zerodim_np_array� s z)TestScalar2.test_getitem_zerodim_np_arrayN) r/ r0 r1 rJ rK rS rV r` ri rm rn rp r r r r r3 M s r3 c C s t ddd�} | | }t| �� td�}t|�� td�}t||d��}|jtk�� sTt�xX|j d |j d |j d |j d gD ]0}||d ks�t�t |t�s�t�t |t�rzt�qzW xX|j d |j d |j d |j d gD ]0}||d ks�t�t |t�s�t�t |t�r�t�q�W d S ) Nz 2016-01-01rM )r5 )r> )r7 r8 )r r7 )r r r )r- r8 )r- r- r- )r r Z to_pydatetime�objectZto_pytimedeltar Zdtypes�allrF r% r r_ rB � isinstancer r r r )ZdtiZtdiZserZser2rH r r r r �&test_iat_dont_wrap_object_datetimelike� s **rt c C sh t dddgiddgd�} | jjdks(t�| jd dks:t�| d }|jjdksRt�|jd dksdt�d S )Nra r- rL )r- rL )rM r6 )r; ))r- rL ra )r r; �nlevelsrF r% )rH r r r r � test_at_with_tuple_index_get� s rv c C s| t dddgiddgd�} | jjdks(t�d| jd< | jd dksDt�| d }|jjdks\t�d|jd< |jd dksxt�d S ) Nra r- rL )r- rL )rM r6 )r; ))r- rL ra rM )r r; ru rF r% )rH r r r r � test_at_with_tuple_index_set s rw c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestMultiIndexScalarc C s� t dddgiddgddggd�}|jjdks0t�|jd dksBt�|jd dksTt�|d }|jjdkslt�|jd dks~t�|jd dks�t�d S ) Nra r- rL rM r6 )r; ))r- rM ra )r- rM )r r; ru rF r% r_ )r rH r r r r �test_multiindex_at_get s z+TestMultiIndexScalar.test_multiindex_at_getc C s� t dddgiddgddggd�}|jjdks0t�d|jd< |jd dksLt�d|jd< |jd dksht�|d }|jjdks�t�d|jd < |jd dks�t�d |jd < |jd d ks�t�d S ) Nra r- rL rM r6 )r; ))r- rM ra r= )r- rM rT )r r; ru rF r% r_ )r rH r r r r �test_multiindex_at_set' s z+TestMultiIndexScalar.test_multiindex_at_setc C s, t dddggd�}|jd }|dks(t�d S )N)r r- FT)r; r )r r% rF )r �s2r r r r � test_multiindex_at_get_one_level9 s z5TestMultiIndexScalar.test_multiindex_at_get_one_levelN)r/ r0 r1 ry rz r| r r r r rx s rx )�__doc__r r �numpyrC r& Zpandasr r r r r Zpandas._testingZ_testingr Z pandas.tests.indexing.commonr r r3 rt rv rw rx r r r r �<module> s 8
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка