Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/multiindex/__pycache__/test_partial.cpython-37.pyc
Назад
B �5�g>$ � @ sh d dl Zd dlZd dlm mZ d dlmZm Z m Z m Z m Z m Z d dlmZ G dd� d�Zdd� ZdS )� N)� DataFrame� Float64Index� Int64Index� MultiIndex� date_range� to_datetimec @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze j d d� �Z e j �deeg�dd� �Zdd� Zdd� Ze j �dedd�ed�gddgfedd�ed dd d �d d ddgfg�d!d"� �ZdS )#�TestMultiIndexPartialc C s� ddg}ddg}t td�t�||g�d�}t td�|d�}|d }t�||� t td�t�|dd � |g�d�}|dg }t�||� tjtdd �� |d W d Q R X tjtd d �� |dg W d Q R X d S ) N� � �a�b� )�index�columns� �1)�matchz'\[1\] not in index') r �ranger � from_product�tm�assert_frame_equal�pytest�raises�KeyError)�self�l1�l2�df�expected�result� r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/multiindex/test_partial.py�test_getitem_partial_int s z.TestMultiIndexPartial.test_getitem_partial_intc C s d S )Nr )r r r r! �test_series_slice_partial) s z/TestMultiIndexPartial.test_series_slice_partialc C s |}|}|� d�}|jd }|jd j}t�||� t�||� |� d�}|jd }t�||� tddgddgddggddddddddgddddddddgddddddddggd �}ttj� d d �|t d �d �} t� t �� | � ddg�}W d Q R X | jd }t�||� d S )N�foo)i� � �bar�one�two���r r )�levels�codes� r% �abcd)r r )r$ r' ) Zxs�loc�Tr r r r �np�random�randn�list�assert_produces_warning� FutureWarning) r � multiindex_dataframe_random_data�/multiindex_year_month_day_dataframe_random_data�frame�ymdr Zresult2r r r r r r! �test_xs_partial, s( z%TestMultiIndexPartial.test_xs_partialc C sT |}|j }|d }|j|j|jjd dk d�}|j�d��d�|_t�||� d S )N)i� r r )r r )r/ Zreindexr r+ � droplevelr r )r r7 r9 r r r r r! �test_getitem_partialM s z*TestMultiIndexPartial.test_getitem_partialc C sl |}|j dd� }|dd� }t�||� |}|j dd� }|jjd }||dk|dk@ }t�||� d S )Nr& Zbaz� � )i� r )i� r% r )r. r r r r+ )r r6 r7 r8 r r r9 Zlevr r r! �test_fancy_slice_partialV s z.TestMultiIndexPartial.test_fancy_slice_partialc C s� t dddgdddgdddggddgddgddggd �}ttj�d d �|d �}|jd d d �f }|jd }t�||� |jd ddgf }|jd ddg }t�||� tj t dd�� |jdd d �f W d Q R X d S )Nr r r r �x�y�p�q)r+ r* r= r )r )r rA z\('a', 'foo'\))r )r r$ ) r r r0 r1 Zrandr. r r r r r )r �idxr r r r r r! �"test_getitem_partial_column_selectf s z8TestMultiIndexPartial.test_getitem_partial_column_selectc C s� |}|� � }|� � }d|jd< d|jd jd d �< t�||� d|d jd< d|d jd jd d �< t�||� d|jd< d|jd jd d �< t�||� d|d jd< |d jd dks�t�d S )Nr )i� r% r �A� i� � )�copyr. �valuesr r �iloc�AssertionError)r r7 r9 r �expr r r! �test_partial_setz s z&TestMultiIndexPartial.test_partial_set�dtypec C s |}|j j}|j �|d �|�g|dd � �|_ |d }|j }t|t�sNt�|tkrlt|jd t�s�t�nt|jd t �s�t�d|jd ks�t�|jd � � r�t�|� � r�t�t j t dd�� |d W d Q R X t j t dd��( t�t�� |�|d� W d Q R X W d Q R X d S )Nr r rF rH Z14)r )r r* Z set_levelsZastype� isinstancer rL �intr r Z _should_fallback_to_positionalr r r r r4 r5 � get_value)r r7 rO r9 r* Zser�mir r r! �!test_getitem_intkey_leading_level� s" & z7TestMultiIndexPartial.test_getitem_intkey_leading_levelc C s~ t �d� |}|jd }|jd d }t�||� t jtdd�� |jd W d Q R X t jtdd�� |jd W d Q R X d S ) Nzskipping for now)i� r i� rF �6)r )i� � z (2000, 6)))i� rV r )r �skipr. r �assert_series_equalr r )r r7 r9 r r r r r! �test_partial_loc_missing� s z.TestMultiIndexPartial.test_partial_loc_missingc C s |}|� � }|� � }d|jddg< d|jd< d|jd< t�||� |� � }|� � }d|jdd�< d|jd< d|jd< t�||� |d � � }|d � � }d|jddg< d|jd< d|jd< t�||� |d � � }|d � � }d|jdd�< d|jd< d|jd< t�||� d S )Nr r$ r& rF )rI r. r r rX )r r6 r8 r r r r r! � test_setitem_multiple_partial� s2 z3TestMultiIndexPartial.test_setitem_multiple_partialz indexer, exp_idx, exp_valuesz2019-2Nz 2019-02-01r r= �2019�MS)�periods�freqr r c C s t dddd�}tttd��tj|ddggdd gd �d �}t|tj|ddggdd gd �d �}|| }t�||� |j| }t�||� |jdd �| }t�||� |j|d d �f }t�||� |� dd�� � }|� dd�� � }|jd d �|d d �f }t�||� d S ) Nr[ r r\ )r] r^ r% r r r@ rA )�names)r )�axis) r r r3 r r r r r r. Z swaplevelZ sort_index) r ZindexerZexp_idxZ exp_valuesZdate_idxr r r Zdf2r r r! �!test_partial_getitem_loc_datetime� s& z7TestMultiIndexPartial.test_partial_getitem_loc_datetime)�__name__� __module__� __qualname__r" r# r: r< r? rE �tdZ skip_array_manager_invalid_testrN r �markZ parametrizerQ �floatrT rY rZ �slicer r ra r r r r! r s ! r c C s� ddgddgg} t j| ddgd�}t j| dd gd�}ttj�d d �||d �}|jd d�dd �f �d�jddd �}|jd }t � ||� d S )Nr r r r Zcol1Zcol2)r_ Zrow1Zrow2r% )r r )r` )r r ) r r r r0 r1 r2 rK r; r. r r )� iterablesr �rowsr r r r r r! �"test_loc_getitem_partial_both_axis s & rk )�numpyr0 r Z pandas.util._test_decorators�utilZ_test_decoratorsre Zpandasr r r r r r Zpandas._testingZ_testingr r rk r r r r! �<module> s x
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка