Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/multiindex/__pycache__/test_setitem.cpython-37.pyc
Назад
B �5�ga? � @ s� d dl Zd dlZd dlm mZ d dlZd dlm Z m Z m Z m Z m Z mZmZ d dlmZ d dlm mZ dd� ZG dd� d�ZG dd� d�Zejd d � �Zd d � Z d d� Z dS )� N)� DataFrame� MultiIndex�Series� Timestamp� date_range�isna�notnac C s | |ks t �d S )N)�AssertionError)�a�b� r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/multiindex/test_setitem.py� assert_equal s r c @ s� e Zd Zedfdd�Zdd� Zdd� Zdd � Zej d d � �Z d d � Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� ZdS )*�TestMultiIndexSetItemNc C s. ||j |< |j | }|d kr |}|||� d S )N)�loc)�self�target�indexers�value� compare_fn�expected�resultr r r �check s zTestMultiIndexSetItem.checkc C s� ddddddddg}t jt�d d �t�d d �gd d gd�}d\}}ttj||d�}| j|||fdfd d� td||d�}| j|||fdfdd� t||d�}| j|||fdfdd� td||d�}| j|||fdft�d�dd� d S )N�A�w�lr �x�X�dZprofitr �d �P �timeZfirm)�names)r � )�columns�index)r r r i���� r# � )r r r r )r � from_product�np�aranger �nanr �array)r �colsr% �t�n�dfr r r �test_setitem_multiindex s& " z-TestMultiIndexSetItem.test_setitem_multiindexc C s� t t�d��dd�d�d�td�}d|d< |d d d k}d d g}|d |j||f< t d d d dgd d gd�}t�|j||f |� | j |||f|d tj|d� d S )N� � z A,B,C,D,E�,)r$ �dtype�c �Fr r# r �B�Cg �X@� )r% r$ )r r r r r ) r r) r* �reshape�split�floatr �tm�assert_frame_equalr )r r0 Z row_selectionZ col_selection�outputr r r �test_setitem_multiindex2> s z.TestMultiIndexSetItem.test_setitem_multiindex2c C s t �dddgtdddd�g�}t �dd gtd d dd�g�}ttj�d �||d �}t �dtd�fdtd�fg�}t �dtd �fdtd �fg�}ttj�d�||d �}| j|||f|t j d� ttj�d�||d �}| j||t d d d �f|t j d� |� � }| j||j |jf|t j |d� d S )Nr r8 r9 z 2015-01-01z 2015-04-01ZMS)�freq�foo�barz 2016-01-01z 2016-02-01)� r: )r% r$ z 2015-02-01)r# r# )r r r r )r# r: )r r r r r )r r( r r r) �random� from_tuplesr r r> r? �slice�copyr% r$ )r �idxr- r0 ZsubidxZsubcols�valsrI r r r �test_setitem_multiindex3Q s8 z.TestMultiIndexSetItem.test_setitem_multiindex3c C s� t �ddddddg�t �ddddddg�t �ddd�g}tt j�dd �|d d d gd ��� }|jdg d }|�� }|jdg d9 < t � |jdg |� d}t j t |d�� |jd d9 < W d Q R X d S )NrD ZbazZqux�one�twor � r& r' r r8 r9 )r% r$ r# zAcannot align on a multi-index with out specifying the join levels)�match)r) r, r* r rF �randn� sort_indexr rI r> r? �pytest�raises� TypeError)r �arrays�df_origr r0 �msgr r r �test_multiindex_setitem~ s z-TestMultiIndexSetItem.test_multiindex_setitemc C s� t �dddddddd�i�}tj|jdd d gd �|_|�� }|jddd g d9 < tj}|�� }|j |d d �d d �d f d d �f d9 < t � ||� |�� }|j |d d �d d �d f df d9 < t � ||� d S )NZpricer# r: r& r3 r ))�DEZCoal�Stock)rZ �Gasr[ )rZ ZElecZDemand)�FRr\ r[ )r] ZSolar�SupIm)r] ZWindr^ ZSitZCom�Type)r" r' r[ ) r � from_dictr rG r% rI �iloc�pdZ IndexSlicer r> r? )r rW r rJ r0 r r r �test_multiindex_setitem2� s& , (z.TestMultiIndexSetItem.test_multiindex_setitem2c C s� t tjjdddd��dd�td�dddgddd ggd �}tj|d < t�d d g�}||jd< t � |jd t |ddgd d�� t tjjdddd��dd�td�dddgddd ggd �}||jd< t |ddgddd�}t � |jd |� d|jd< t dddgddd�}t � |jd |� d}t j t|d�� ddddg|jd< W d Q R X t j t|d�� dg|jd< W d Q R X dg|jddgf< |jd dk�� �s�t�d S )Nr3 � � )�sizer' �abcr: � rE )r$ r% r g g �?)r: r )r% �name)r: �crj �float64)r% ri r5 z@Must have equal len keys and value when setting with an iterable)rP r r& r# )r r) rF �randintr; �listr+ r, r r> �assert_series_equalr rS rT � ValueError�allr )r r0 Zarr�exprX r r r �test_multiindex_assignment� s2 z0TestMultiIndexSetItem.test_multiindex_assignmentc C s� d}d}dd� t tt�|��� �D �}|d d� }ttjjd||fd�tj|d�}|� |�� � }|j |d d� d �}tj |d < x.|D ]&\}}t�|j d �} | |j|d f< q�W d S ) Nr rd c S s g | ] }d | �qS )r r )�.0�numr r r � <listcomp>� s z>TestMultiIndexSetItem.test_groupby_example.<locals>.<listcomp>r3 )rf )r5 r$ r: )�levelZnew_colr )�map�strr) r* �tolistr rF rl �int64� set_indexrR �groupbyr+ �shaper ) r ZNUM_ROWSZNUM_COLSZ col_namesZ index_colsr0 �grpri Zdf2�new_valsr r r �test_groupby_example� s z*TestMultiIndexSetItem.test_groupby_examplec C s� |}|d }t j|d< t|jdd� ��� s0t�t|jd d� ��� sJt�t|jdd � ��� sdt�t j|d< t|jd �s�t�tj t dd�� |d W d Q R X d S ) Nr )i� r' �* �A )i� r' rd �1 �49)rP ) r) r+ r �valuesrp r r ra rS rT �KeyError)r �/multiindex_year_month_day_dataframe_random_data�ymd�sr r r �test_series_setitem� s z)TestMultiIndexSetItem.test_series_setitemc C s� |}|j �� }|j}||dk }|�|dk�}t�||� 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|� t j t dd�� d||d < W d Q R X d S )Nr r3 r# ���zboolean values only)rP ) �TrI r� �wherer> r? Zassert_almost_equalr) ZputmaskrS rT rU )r � multiindex_dataframe_random_data�framer0 r� r r r r r �"test_frame_getitem_setitem_boolean s" "z8TestMultiIndexSetItem.test_frame_getitem_setitem_booleanc C s ddgdddgg}dddddgdddddgg}t ||d d gd �}td ddd d dgi|d�}|jd d �d f }t�|d |� |j|jdd � d f }t�|d dd � |� |jd d �d d �f }t�||� |}d|jd d �d f< d|d < t�||� d|jd d �d d �f< t�||� d S )N�t1�t2r r rj r r& r# �id)�codes�levelsr" r r' � rh )r% rd )r r r r> rn r% r? )r r� r� Zmidxr0 r r r r �%test_frame_getitem_setitem_multislice# s z;TestMultiIndexSetItem.test_frame_getitem_setitem_multislicec C s� t tj�dd�ddddgddddggd�}|�� }|d |d< t�|d |d � |�� }|d j|d< t�|d |d � d S )Nrd r: r r r r& )r$ )r r) rF rQ rI r> r? r� )r r0 �cpr r r � test_frame_setitem_multi_column: s ( z5TestMultiIndexSetItem.test_frame_setitem_multi_columnc C s� t �dddg�}tdddg|d�}d|d < |d jd k�� s@t�dd dg|d< |d |d < |d }|d }|d }tj||d d � tj||d d � |jdks�t�|jdks�t�|jdks�t�d S )N)r �1)r �2)r8 r� r& r' r3 )r% r$ g r r r# F)Z check_names) r rG r r� rp r r> rn ri )r r$ r0 Z sliced_a1Z sliced_a2Z sliced_b1r r r � test_frame_setitem_multi_column2H s z6TestMultiIndexSetItem.test_frame_setitem_multi_column2c C sH |}|d d� }|j ddddgf }|j d dddg }t�||� d S )Nr3 )i� r& rO r r8 r9 )r r> rn )r r� r� r0 r r r r r �'test_getitem_setitem_tuple_plus_columnsa s z=TestMultiIndexSetItem.test_getitem_setitem_tuple_plus_columnsc C s� t dddgddggddddddgddddddggd�}ttj�t|�d�|dddd gd �}|tk rj|d }|jdd� }|�|jdd � �}t � ||� d |jdd�< |jdd� d kj � � s�t �d S ) Nr r& r# )r� r� r: r r rj r )r% r$ r� )r r r) rF rQ �lenr �reindexr% r> r r� rp r )r Zframe_or_seriesr% �obj�resrq r r r �'test_loc_getitem_setitem_slice_integersl s 4" z=TestMultiIndexSetItem.test_loc_getitem_setitem_slice_integersc C sd |}|j }|d }||�� k|d< t�|d ||�� k� |jdgd�}t�|d ||�� k� d S )N)rC rN )r$ )r� Zmedianr> rn r� )r r� r� Zdftr� Z reindexedr r r �test_setitem_change_dtype~ s z/TestMultiIndexSetItem.test_setitem_change_dtypec C sd |}|j dddg }d|j|< |j| jdk�� s6t�|d }d||< |j|df dk�� s`t�d S )Nr& r: r3 r6 r8 �a )r% r r� rp r )r r� r� Zsubset�colr r r � test_set_column_scalar_with_loc� s z5TestMultiIndexSetItem.test_set_column_scalar_with_locc C st t ddddgddddgddddggtd �d �}|�d d g�}t�d g�}d|j|df< |�d �d dk�� spt�d S )Nr& r r �y�Yr# �z�ZZABCD)r$ r r8 )r& r& �_r9 ) r rm r{ r rG r Zxsrp r )r r0 �mir r r � test_nonunique_assignment_1750� s , z4TestMultiIndexSetItem.test_nonunique_assignment_1750c C sZ t �dddg�}tt�d��d�|td�}|j�� }|d � tj �|d< t � |j|� d S )N)r r� )r8 r� )r r� r' )r& r' )r$ r5 r ) r rG r r) r* r; �objectr% rI Zastyperk r> Zassert_index_equal)r r- r0 r% r r r � test_astype_assignment_with_dups� s z6TestMultiIndexSetItem.test_astype_assignment_with_dupsc C sj t jdddgddgd�}tddd g|d gd �}tjd d dd�|jd< td d ddgi|d�}t�||� d S )N)r rj )r r )r r �l1�l2)r" r r& r# �e)�datar% r$ r6 �e rz )r5 )r r� r )r% )r rG r r) r* r r> r? )r r% r0 r r r r �test_setitem_nonmonotonic� s z/TestMultiIndexSetItem.test_setitem_nonmonotonic)�__name__� __module__� __qualname__r r r1 rA rL �tdZ&skip_array_manager_not_yet_implementedrY rc rr r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r s( - - r c @ s e Zd Zdd� Zdd� ZdS )�"TestSetitemWithExpansionMultiIndexc C s� ddddddgddddddgdd d dddgg}t t|� �}t�|�}ttj�d d �|d �}|�� }|�� }dddd g|d< dddd g|d< t � ||� d S )Nr �topZroutine1Zroutine2� ZODZresult1Zresult2ZwxZwyr: rO )r$ r& r# r' r )r r� r� ) �sorted�zipr rG r r) rF rQ rI r> r? )r rV Ztuplesr% r0 r r r r r �#test_setitem_new_column_mixed_depth� s zFTestSetitemWithExpansionMultiIndex.test_setitem_new_column_mixed_depthc C s\ t �dddg�}tddgddgdd gg|d �}tddd ��}||d < |d �� �� sXt�d S ) N)�1aZ2a)r� Z2b)r� Z2cr& r# r' r: r3 rO )r% ))r& r& )r& r# �new)r rG r r r rp r )r Zmixr0 r� r r r � test_setitem_new_column_all_na� s zATestSetitemWithExpansionMultiIndex.test_setitem_new_column_all_naN)r� r� r� r� r� r r r r r� � s r� c C s2 | j }d|d jd d �< |d jdk�� s.t�d S )Nr rC )r� r� rp r )r� r0 r r r � test_frame_setitem_view_direct� s r� c C s6 | j }d}tjtj|d�� d|d d< W d Q R X d S )NzAA value is trying to be set on a copy of a slice from a DataFrame)rP r# rC rM )r� rS rT �com�SettingWithCopyError)r� r0 rX r r r � test_frame_setitem_copy_raises� s r� c C sR | j }|}|�� }d}tjtj|d�� d|d d< W d Q R X |}t�||� d S )NzAA value is trying to be set on a copy of a slice from a DataFrame)rP r# rC rM )r� rI rS rT r� r� r> r? )r� r� r r0 rX r r r r � test_frame_setitem_copy_no_write� s r� ) �numpyr) rS Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasrb r r r r r r r Zpandas._testingZ_testingr> Zpandas.core.common�core�commonr� r r r� Z skip_array_manager_invalid_testr� r� r� r r r r �<module> s $ !
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка