Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/multiindex/__pycache__/test_getitem.cpython-37.pyc
Назад
B �5�g1 � @ sT d dl Zd dlZd dlmZmZmZmZ d dlm Z d dl m Z ej �ddd� dd� dd� g�ej �d d ed gd gd �fd ed d gd d gd �fg�dd� ��Zej �dddg�dd� �Zdd� Zdd� Zej �ddd� edfd d� edfd d� edfd d� e d fd d� ed!fd"d� ed!fd#d� ed$fg�d%d&� �Zd'd(� Zd)d*� Zej �d+d,d� d-fd.d� d/fg�d0d1� �Zd2d3� Zej �d4d5d� ed �fd6d� ed d7�fd8d� ed d7�fg�d9d:� �Zd;d<� Z d=d>� Z ej �d?d@gdAej gfed d gd7dBgge� dCd@ej fg�dDdE�fdFd@ged d d gdGd7dBgge� dHdCd@ej fg�dDdE�fd@ged d gd7dBgge� dCd@ej fg�dDdE�fd@gdAgfed gd7gge� dCg�dDdE�fd@gej gfed gdBgged gdIggdFd@gdAdJggdK�dDdE�fd@ej fed dBgdDd@ej fdL�fg�dMdN� �Z ej!dOdP� �Z"ej �dQdRd� dSd� g�dTdU� �Z#dVdW� Z$dXdY� Z%dZd[� Z&d\d]� Z'dS )^� N)� DataFrame�Index� MultiIndex�Series)� IndexingError� access_methodc C s | d d �|f S )N� )�s�xr r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.py�<lambda> � r c C s | j d d �|f S )N)�loc)r r r r r r r c C s | j |dd�S )N� )�level)Zxs)r r r r r r r zlevel1_value, expectedr )�index� � c C s: t dddg�}t�dddg�|_| ||�}t�||� d S )Nr r r )r r )r r )r r )r r � from_tuplesr �tm�assert_series_equal)r Z level1_value�expectedr �resultr r r � test_series_getitem_multiindex s r � level0_value�D�Ac C s t | ddgddddddd d ggdddd d d d d d d g d d dddd d d ddg gddgd�}tj�t|�d �}t||dgd�}| dkr�tjtdd�� |j d W d Q R X tjtdd�� |j d W d Q R X |j | }t |� � dd � dt dddgdd �d �}t �||� d S ) N�B�Cr � � �% �9 �C �K �R r r r � � � � �tag�day)�levels�codes�names�val)r �columnsr z^'A'$)�matchz^'X'$�X)�name)r3 r )r �np�random�randn�lenr �pytest�raises�KeyErrorr/ r Zravelr r r )r r Zarr�dfr r r r r �)test_series_getitem_duplicates_multiindex% s . &r< c C sL | d }|� |jdd� �}|j�d��d�|_||�d }t�||� d S )Nr �* �A r )i� r )�reindexr � droplevelr r )�/multiindex_year_month_day_dataframe_random_data� indexer_slr r r r r r �test_series_getitemA s rC c C s. | d }|j d }||�d }||ks*t�d S )Nr �1 )i� r � )�iloc�AssertionError)rA rB r r r r r r �"test_series_getitem_returns_scalarJ s rH z)indexer,expected_error,expected_error_msgc C s | � d�S )N)i� r r& )� __getitem__)r r r r r W r z^\(2000, 3, 4\)$c C s | d S )N)i� r r& r )r r r r r X r c C s | j d S )N)i� r r& )r )r r r r r Y r c C s | j d S )N)i� r r& r( )r )r r r r r Z r zToo many indexersc C s | � t| ��S )N)rI r7 )r r r r r [ r � c C s | t | � S )N)r7 )r r r r r \ r c C s | j t| � S )N)rF r7 )r r r r r ^ r z*single positional indexer is out-of-boundsc C s. | d }t j||d�� ||� W d Q R X d S )Nr )r1 )r8 r9 )rA �indexerZexpected_error�expected_error_msgr r r r �#test_series_getitem_indexing_errorsT s rM c C s6 | d }|dd� |D � }||dk }t �||� d S )Nr c s s | ]}|d kV qdS )r Nr )�.0r r r r � <genexpr>s s z7test_series_getitem_corner_generator.<locals>.<genexpr>r )r r )rA r r r r r r �$test_series_getitem_corner_generatoro s rP c C s2 | j }|jd d �df }|d j}t�||� d S )Nr )�foo�one)�T�valuesr Zassert_almost_equal)� multiindex_dataframe_random_datar; r r r r r �test_getitem_simple} s rV zindexer,expected_error_msgc C s | d S )N)rQ �fourr )r; r r r r � r z^\('foo', 'four'\)$c C s | d S )NZfoobarr )r; r r r r � r z ^'foobar'$c C s, | j }tjt|d�� ||� W d Q R X d S )N)r1 )rS r8 r9 r: )rU rK rL r; r r r �#test_frame_getitem_simple_key_error� s rX c C sh t dddgdddgd��} ddgdd gd d gg| _| d }t dgdgdgg| jd gd �}t�||� d S ) N�1�2�3�4)�a�bz level1 item1z level1 item2rJ z level2 item2z level3 item1z level3 item2)r r0 )r r0 r r �assert_frame_equal)r; r r r r r �*test_frame_getitem_multicolumn_empty_level� s r` zindexer,expected_slicec C s | d S )NrQ r )r; r r r r � r c C s | d S )N�barr )r; r r r r � r r( c C s | j d d �df S )Nra )r )r; r r r r � r c C s> | j }|j|j| d�}|j�d�|_||�}t�||� d S )N)r0 r )rS r? r0 r@ r r_ )rU rK Zexpected_slicer; r r r r r �test_frame_getitem_toplevel� s rb c C s� ddddddgddddddgdd d dddgg} t t| � �}t�|�}ttj�d d �|d �}|d }|d �d�}t � ||� |d }|d }|�d�}t � ||� d S )Nr] �top�routine1Zroutine2rJ ZOD�result1Zresult2ZwxZwyr& r' )r0 )r] rJ rJ )rd re )rd re rJ ) �sorted�zipr r r r4 r5 r6 �renamer r )�arraysZtuplesr r; r r r r r �test_frame_mixed_depth_get� s rj c C sr | }dddg}t d|dgd|dgd|d gd |d gg|d ��ddg�}|d �d �|d< d|f}|jd |� }t d|dgd|dgg|d ��ddg�}|d �d �|d< t�||� |j|d � }t d|dgd|d gd |d gg|d ��ddg�}|d �d �|d< t�||� d|fd|f }}|j||� }t d|dgd|d gg|d ��ddg�}|d �d �|d< t�||� d S )Nr] r^ �c� � � � � �! �) �+ )r0 �int64)r Z set_indexZastyper r r_ ) � nulls_fixture�n�colsr; �idxr r �idx1�idx2r r r �!test_frame_getitem_nan_multiindex� s. " & " &r{ zindexer,expectedr^ ra r' )r^ ra rt )r0 �dtyper] r& )r] rQ ���rQ )r- r, )r| r3 c C sP t dddgdddggt�ddd |fg�d d �}|jd d �| f }t�||� d S ) Nr r r r& r( r' )r] rQ )r^ ra r^ rt )r0 r| )r r r r r Z assert_equal)rK r ru r; r r r r �&test_frame_getitem_nan_cols_multiindex� s Cr~ c C s� ddddddgddd d d d gddd d d d gg} d ddg}t ddgddddggddddddgdd d ddd ggddgd�}t| ||d�S )z;Fixture for DataFrame used in tests for gh-4145 and gh-4146r] �d�erk �fr^ r r& r( r r' r �h1�h3�h5r r �A1�A2ZB1�B2r �main�sub)r, r- r. )r r0 )r r )�datar r0 r r r � dataframe_with_duplicate_index< s . r� rK c C s | d S )N)r r� r )r; r r r r J r c C s | j d d �df S )N)r r� )r )r; r r r r J r c C sV | }t dddg�}tjdgddgd�}tdd d gg||d �j}||�}t�||� d S ) Nr� r� r� )r r� r� r� )r. r] r )r r0 )r r r r rS r r_ )r� rK r; r r0 r r r r r �test_frame_mi_accessI s r� c C s: | }t dddgdddgdd�}|d d }t�||� d S ) Nr] r r� r� r� r� )r r3 r )r r r )r� r; r r r r r �#test_frame_mi_access_returns_seriesW s r� c C sR | }t dddgdddggtddgdd�dd d gd �j}|d d }t�||� d S ) Nr r& r� r( r� r� )r3 r� r� r� )r r0 r )r r rS r r_ )r� r; r r r r r �"test_frame_mi_access_returns_frameb s r� c C s\ t dtd�t�dgdgg�d�} | g }t ddgtdgdggg g gd�d�}t�||� d S )Nr r r )r r0 )r, r- )r �ranger Z from_productr r_ )r; r r r r r �test_frame_mi_empty_slicen s "r� c C s� ddddgddddgg} t j| dd�}tddddg|d gd �}|j|jd d �d f d kd d �f j}|j|d d �f }|jd gt|j� d d �f }t�||� d |j|j|jd d �d f d k jd f< |}tddddg|d gd �}t�||� d S )Nr] r^ )ry rz )r. r r r r& �value)r r0 r Fr( )r Z from_arraysr r r r7 r r_ )ri r r; Zempty_multiindexr r r r r �test_loc_empty_multiindexx s & (r� )(�numpyr4 r8 Zpandasr r r r Zpandas._testingZ_testingr Zpandas.core.indexingr �markZ parametrizer r< rC rH r: � IndexErrorrM rP rV rX r` �slicerb rj r{ �nanr r~ Zfixturer� r� r� r� r� r� r r r r �<module> s� 8 .
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка