Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/__pycache__/test_xs.cpython-37.pyc
Назад
B �5�gd5 � @ s� d dl Z d dlZd dlZd dlmZmZmZmZm Z m Z d dl m Z d dlm mZ d dlmZ ejdd� �ZG dd� d�ZG dd � d �ZdS ) � N)� DataFrame�Index� IndexSlice� MultiIndex�Series�concat)�BDayc C s� t �dddddgdddd d gd d d ddgg�} tddgddgdddgdddggdddgdddgddd gd ddggd d d d gd!�}t| |td"�d#�S )$NgE���JY�gr�鷯�g�|?5^�ݿg���6��?g����K�?g �-���?g��T����?g���9#�?g�9]�?g46<�R�?g� ���Q�g�^)���g���x�ֿgI��&�?g=�U��@�a�x�b�qgı.n�$@g 4@g >@� � � r � � �one�two�three�four)�levels�codes�names�ABCDE)�index�columns)�np�arrayr r �list)�arrr � r �Y/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/test_xs.py�four_level_index_dataframe s "r" c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestXSc C sB |j d }|�|�}xJ|�� D ]>\}}t�|�rHt�|| | �s\t�q ||| | ks t�q W ddd�dddd�d �}t|�}|�d�}|jtjks�t�|d dks�t�|d dks�t�t j t t � d �d �� |�|j d t� � W d Q R X |jd dd�} |d } t�| | � |jd dd�} d| d d �< | dk�� �s>t�d S )Nr r r )�1�2r$ r% �3)r$ r% r& )�A�Br' r( z*Timestamp('1999-12-31 00:00:00', freq='B'))�matchr )�axis)r �xs�itemsr �isnan�AssertionErrorr �dtypeZobject_�pytest�raises�KeyError�re�escaper �tm�assert_series_equal�all) �selfZ float_frameZdatetime_frame�idxr+ �item�valueZ test_data�frameZseries�expectedr r r! �test_xs& s* zTestXS.test_xsc C s� t dgd�}d|d< d|d< d|d< d |d < d |d < |�d�}tdddd d gtd �dd�}t�||� t dddgd�}|�d�}tg dtg �tjd�}t�||� d S )Nr )r g �?r' �foor( g @�C�bar�Dg @�Er )r �namer r �c)rD r r/ ) r r+ r r r5 r6 r r Zfloat64)r8 �dfr+ �exp�resultr= r r r! �test_xs_cornerF s zTestXS.test_xs_cornerc C sD t tj�dd�dddddgd�}|�d�}|jd }t�||� d S )Nr r r rE r )r )r r �random�randnr+ �ilocr5 r6 )r8 rF �crossrG r r r! �test_xs_duplicatesY s zTestXS.test_xs_duplicatesc C s� t ddd�ddd�ddd�ddd�d���d d d g�}|jdd d d �}|d d� }t�||� t�t�� |jddgd d gd d �}W d Q R X t�||� d S )N�sat�sun)r r � strawberry� � i� )�day�flavourZsales�yearrV rU rT F)�level� drop_levelr )r � set_indexr+ r5 �assert_frame_equal�assert_produces_warning� FutureWarning)r8 rF rH r= r r r! �test_xs_keep_level` s "zTestXS.test_xs_keep_levelc C s� t t�d��dd�td�td�d�}|rrd}tjtj|d�� d|� d�d d �< W d Q R X |� d�dk� � r�t �n(d|� d�d d �< |� d�dk� � s�t �d S ) Ng 4@r r )r r zC\nA value is trying to be set on a copy of a slice from a DataFrame)r) � r ) r r ZarangeZreshape�ranger0 r1 �com�SettingWithCopyErrorr+ �anyr. r7 )r8 �using_array_manager�dm�msgr r r! � test_xs_viewq s $ zTestXS.test_xs_viewN)�__name__� __module__� __qualname__r> rI rN r] rf r r r r! r# % s r# c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d d d gdgfg�dd� �Z dd� Z ej � ddddd� e ddgdd�fddd d� e dgdd�fg�d d � �Zej � d d d� d!d� g�d"d#� �Zej � d d$d� d%d� g�d&d'� �Zd(d)� Zd*d+� Zej � d,eeg�d-d.� �Zej � d,eeg�d/d0� �Zd1d2� Zd3d4� Zd5d6� Zd7d8� Zd9S ):�TestXSWithMultiIndexc C sx t dd�}td�}tj||gddgd�}ttj�t|�d�|dd d g�}|j ddd �}|j dd d �f }t � ||� d S ) Nia�2ie�2�abcde�dateZsecid)r r �X�Y�Z)rW ) r_ r r Z from_productr r rJ rK �lenr+ �locr5 rZ )r8 �datesZidsr rF rH r= r r r! �test_xs_integer_key� s z(TestXSWithMultiIndex.test_xs_integer_keyc C sL |}|j ddd�}||j�d�dk }tddddgd d �|_t�||� d S ) Nr �second)rW r r? rA �baz�qux�first)rD )r+ r Zget_level_valuesr r5 rZ )r8 � multiindex_dataframe_random_datarF rH r= r r r! � test_xs_level� s z"TestXSWithMultiIndex.test_xs_levelc C s� t j�dd�}tdddgdddgd d d ggd d dgd d dgd d dggd�}t||d�}t|dd � dgdggd�}|jd d d�}t�||� d S )Nr r r �pr r r �yrE �r�zr r r )r r )r )rW )r rJ rK r r r+ r5 rZ )r8 r r rF r= rH r r r! �test_xs_level_eq_2� s z'TestXSWithMultiIndex.test_xs_level_eq_2c C sB |}|j ddd�}d}tjtj|d�� d|d d �< W d Q R X d S )Nr rt )rW zAA value is trying to be set on a copy of a slice from a DataFrame)r) rR )r+ r0 r1 r` ra )r8 rx rF rH re r r r! � test_xs_setting_with_copy_error� s z4TestXSWithMultiIndex.test_xs_setting_with_copy_errorc C sF |}|j dddgd�}d}tjtj|d�� d|d d �< W d Q R X d S )N)r r r r )rW zAA value is trying to be set on a copy of a slice from a DataFrame)r) rR )r+ r0 r1 r` ra )r8 r" rF rH re r r r! �(test_xs_setting_with_copy_error_multiple� s z=TestXSWithMultiIndex.test_xs_setting_with_copy_error_multiplez key, level)r rt r rt c C s� |}t |gd �}|jjdks"t�t |jddd�gd �}t|t�rjt�t �� |j||d�}W d Q R X n|j||d�}t� ||� d S )Nr Fr rt )rW ) r r Z is_uniquer. r+ � isinstancer r5 r[ r\ rZ )r8 �keyrW rx r< rF r= rH r r r! �test_xs_with_duplicates� s z,TestXSWithMultiIndex.test_xs_with_duplicatesc C sv dddddddg}t |dd d gd ��dd g�}t d d d dd gitdtjddgd d�d�}|jddd�}t�||� d S )N)r rk r )r Zbbcder )r{ Zyzcde� )r} �xbcde� )r} N� )r} �zbcder� )r} �ybcder� Za1Za2Zcnt)r r� r� r� r� r� r� )rD )r r} )rW )r rY r r �nanr+ r5 rZ )r8 �accrF r= rH r r r! � test_xs_missing_values_in_index� s z4TestXSWithMultiIndex.test_xs_missing_values_in_indexz key, level, exp_arr, exp_indexr �lvl0c C s | d d �dd�f S )Nr r r )r r r r! �<lambda>� � z TestXSWithMultiIndex.<lambda>rA r? �lvl1)rD c C s | d d �dd�f S )Nr r r )r r r r! r� � r� c C s� t j�dd�}tddgddddggddd d gdd d d ggd d gd�}t||d�}|j||d d�}t||�|d�} t�|| � d S )Nr r r rA r? ZhelloZworldr r r r r� r� )r r r )r )rW r* )r rJ rK r r r+ r5 rZ ) r8 r� rW Zexp_arrZ exp_indexr r rF rH r= r r r! � test_xs_named_levels_axis_eq_1� s z3TestXSWithMultiIndex.test_xs_named_levels_axis_eq_1�indexerc C s | j dddgd�S )N)r r r r )rW )r+ )rF r r r! r� � r� c C s | � d�j ddd�S )Nr r r )rW )r+ )rF r r r! r� � r� c C s` |}dddddgg}t dgdggdgdggd d gd �}t||td �d �}||�}t�||� d S )Ng �-���?g��T����?g���9#�?g�9]�?g46<�R�?r g 4@r r r )r r r r )r r )r r r r5 rZ )r8 r� r" rF �expected_values�expected_indexr= rH r r r! �test_xs_level_multiple� s z+TestXSWithMultiIndex.test_xs_level_multiplec C s | j ddd�S )Nr r )rW )r+ )rF r r r! r� r� c C s | � d�S )Nr )r+ )rF r r r! r� r� c C s� |}dddddgdddd d gg}t d d gd dgddggddgddgddggdddgd�}t||td�d�}||�}t�||� d S )NgE���JY�gr�鷯�g�|?5^�ݿg���6��?g����K�?g �-���?g��T����?g���9#�?g�9]�?g46<�R�?r r gı.n�$@g 4@r r r r r r r )r r r r )r r )r r r r5 rZ )r8 r� r" rF r� r� r= rH r r r! �test_xs_level0 s z#TestXSWithMultiIndex.test_xs_level0c C s* |}|� d�j}|jd }t�||� d S )N)rA r r )r+ �valuesr5 Zassert_almost_equal)r8 rx rF rH r= r r r! �test_xs_values s z#TestXSWithMultiIndex.test_xs_valuesc C s( |}|� d�}|jd }t�||� d S )N)rA r )r+ rq r5 r6 )r8 rx rF rH r= r r r! �test_xs_loc_equality s z)TestXSWithMultiIndex.test_xs_loc_equality�klassc C s� t dddgddggddddddgddddddggd�}ttj�dd �|d �}|tkr\|d }d }tjt|d �� |� t dd d �f � W d Q R X d S ) N)r? rA r )r? ru r )r? rv r r r r )r r � r )r zYExpected label or tuple of labels, got \(\('foo', 'qux', 0\), slice\(None, None, None\)\))r) ) r r r rJ rK r r0 r1 � TypeErrorr+ r )r8 r� r �objre r r r! �+test_xs_IndexSlice_argument_not_implemented# s $z@TestXSWithMultiIndex.test_xs_IndexSlice_argument_not_implementedc C sR t ddddgi�}|tkr"|d }d}tjt|d�� |jddd � W d Q R X d S ) Nr' r r r zIndex must be a MultiIndex)r) r �as)rW )r r r0 r1 r� r+ )r8 r� r� re r r r! �test_xs_levels_raises7 s z*TestXSWithMultiIndex.test_xs_levels_raisesc C sp t jdddgddgd�}tddd gg|d �}|jd dd d �}tddggt jddgddgd�d �}t�||� d S )N)r r )r r{ )r r Zlevel1Zlevel2)r r r r )r r F)r* rX )r � from_tuplesr r+ r5 rZ )r8 �mirF rH r= r r r! �"test_xs_multiindex_droplevel_falseA s z7TestXSWithMultiIndex.test_xs_multiindex_droplevel_falsec C sL t dddggtdddg�d�}|jdddd �}t ddgi�}t�||� d S ) Nr r r r r rE )r F)r* rX )r r r+ r5 rZ )r8 rF rH r= r r r! �test_xs_droplevel_falseP s z,TestXSWithMultiIndex.test_xs_droplevel_falsec C s� t dddggtdddg�d�}|jdddd �}t�|jd d �d f j|jd d �d f j�s^t�d|jd < |s|t ddgi�}nt ddgi�}t� ||� t dd dggtdddg�d�}|jdddd �}d|jd < t ddgi�}t� ||� d S ) Nr r r r r rE )r F)r* rX r )r r g @) r r r+ r Z shares_memoryrL Z_valuesr. r5 rZ )r8 rc rF rH r= r r r! � test_xs_droplevel_false_viewW s 0 z1TestXSWithMultiIndex.test_xs_droplevel_false_viewc C st t �dddg�}tdddgddd gg|d �}t�t��4 tjtd d �� |j d d gddd� W d Q R X W d Q R X d S )N)r �mr )r �nr )r{ �orE r r r r r r� )r r{ )r) r F)rX r* ) r r� r r5 r[ r\ r0 r1 r2 r+ )r8 r� rF r r r! �$test_xs_list_indexer_droplevel_falseo s z9TestXSWithMultiIndex.test_xs_list_indexer_droplevel_falseN)rg rh ri rs ry r~ r r� r0 �markZ parametrizer� r� r r� r� r� r� r� r r r� r� r� r� r� r� r r r r! rj � s2 "& rj )r3 �numpyr r0 Zpandasr r r r r r Zpandas._testingZ_testingr5 Zpandas.core.common�core�commonr` Zpandas.tseries.offsetsr Zfixturer" r# rj r r r r! �<module> s ^
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка