Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_quantile.cpython-37.pyc
Назад
B �5�g�` � @ sh d dl Zd dlZd dlmZ d dlZd dlmZmZm Z m Z d dl m Z G dd� d�ZG dd� d�ZdS )� N)�np_percentile_argname)� DataFrame�Index�Series� Timestampc @ s e Zd Zej�deeej � ddg��eej � ddg��d��eddgd d �geed d d dgdd��ed gd d �gg�dd� �Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd S )3�TestDataFrameQuantilez df,expected� � � � )r r g �?g @g �?)�nameg Ng �?g @z Sparse[float])�dtypec C s |� � }t�||� d S )N)�quantile�tm�assert_series_equal)�self�df�expected�result� r �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_quantile.py�test_quantile_sparse s z*TestDataFrameQuantile.test_quantile_sparsec C s ddl m} |}|jddd�}|d ||d d�ks8t�t�|j|j� |jddd�}|d ||jd d �ksrt�t�|j|j� t g g d ��jddd�}t � |d �r�t � |d �s�t�t ddddgddddgd��}|�d�}tj t dd�� |�� �d�}W d Q R X t�||� t dddgdddgd�dddgd�}|jddd�}tdddgdddgdd�}t�||� |jdd gdd�}t dd gdd gdd gd �dd gd�}tj||d!d"� t dddgd#d$dgg�}|jddd�}td%d&gddgdd�}t�||� d S )'Nr )� percentileg�������?)�axis�A� g�������?r z 2000-01-17�Z )�x�yr r �Br r r )Zcol1Zcol2g �?zSelect only valid)�match)r r )�indexg �?g @g @)r! r g �?g �?g @g @)r r r T)Zcheck_index_type�a�bg @g @)�numpyr r �AssertionErrorr Zassert_index_equalr! �columns�locr �np�isnanZassert_produces_warning� FutureWarningZmedian�renamer r �assert_frame_equal) r �datetime_framer r �q�rsZxpr r r r r � test_quantile* s6 $ "z#TestDataFrameQuantile.test_quantilec C sL t jdddd�}t|�}t|�}|jdd�}tdgdd d �}t�||� d S ) Nz 2016-01-01r z US/Pacific)�periods�tzF)� numeric_onlyz2016-01-02 00:00:00g �?zdatetime64[ns, US/Pacific])r r )�pd� date_ranger r r r r )r Zdti�serr r r r r r �test_quantile_date_rangeU s z.TestDataFrameQuantile.test_quantile_date_rangec C s� t dddgdddgtjddd�d d d gd ��}|jd dd�}tdddgd d�}t�||� d}tjt |d�� |jd ddd� W d Q R X d S )Nr r r g @g @g @Z20130101)r1 Zfoo�barZbaz)r r �C�Dg �?)r g �?g @g @)r z>'<' not supported between instances of 'Timestamp' and 'float')r F)r r3 ) r r4 r5 r r r r �pytest�raises� TypeError)r r r r �msgr r r �test_quantile_axis_mixedc s z.TestDataFrameQuantile.test_quantile_axis_mixedc C s t dddgdddgd�dddgd�}|jddd �}td d gd d gdd�}t�||� |jddd �}t�||� |jddd �}tdddgdddgdd�}t�||� |jddd �}t�||� d}tjt|d�� |jddd � W d Q R X d}tjt|d�� |jddd � W d Q R X d S )Nr r r r )r r )r! g �?r )r g @g @r r )r! r r! g �?g @g @r& z*No axis named -1 for object type DataFrame)r g�������?���z.No axis named column for object type DataFrame�column)r r r r r r; r< � ValueError)r r r r r> r r r � test_quantile_axis_parameterw s" $ z2TestDataFrameQuantile.test_quantile_axis_parameterc C sV t dddgdddgd�dddgd�}|jdddd �}tdddgdddgdd �}t�||� tjt�dddgdddgg�dfd d itdi��}t|dddgdd d�}t�||� t dddgdddgd�dddgd�}|jdddd �}tdddgdddgdd �}t�||� tjt�dddgdddgg�dfd d itdi��}t|dddgddd�}t�||� |jddgddd �}t ddgddgddgd�ddgd�}t� ||� t g g d��}|jdd dd �}t� |d ��r�t� |d ��s�t �t dddgdddgdddggd d d gd �}|jd dgd!d"�}t d#d#d#gdddggd dgd d d gd$�}t� ||� d S )%Nr r r r )r r )r! g �?Znearest)r � interpolation)r! r r r �int64)r! r r g �?g @g @g @�float64g �?�lower)r r r )r r g�������?Zhigherr r r" r# �c)r& g �?Zmidpoint)rD g �?)r! r& ) r r r r r r( r �arrayr r, r) r% )r r r r �expr. r r r �test_quantile_interpolation� sH $ $ " $*z1TestDataFrameQuantile.test_quantile_interpolationc C s4 |}|j dddd�}|d t�|d d�ks0t�d S )Ng�������?r �linear)r rD r r )r r( r r% )r r- r r. r r r �$test_quantile_interpolation_datetime� s z:TestDataFrameQuantile.test_quantile_interpolation_datetimec C sf |}|� d�}|d t�|d d�ks*t�|j dddd�}|d t�|d d�ksVt�t�||� d S )Ng�������?r r r rL )r rD )r r( r r% r r )r Z int_framer r. �q1r r r � test_quantile_interpolation_int� s z5TestDataFrameQuantile.test_quantile_interpolation_intc C s� t dddgdddgdddggdddgd�}|�dd g�}t d d d gd d d ggdd gdddgd �}t�||� |jdd gdd �}t d d d gd d d ggdd gdddgd �}t g g d��jddgdd �}t tjtjgtjtjgd�ddgd�}t�||� d S )Nr r r r" r# rH )r& g �?g �?g �?g @)r! r& )r r )r r g�������?g�������?)r! )r r r r, r( �nan)r r r r r r r �test_quantile_multi� s * & $z)TestDataFrameQuantile.test_quantile_multic C sD t t�ddg�ddgd��}|�d�}tdgdgd �}|jdd d �}ttd �dgd dgdd�}t�||� |jdgd d �}t td �dggdgd dgd�}t�||� t�ddg�|d< |d dg jddd d�}ttd �td�gddgdd�}t�||� |d dg jdgdd d�}t td �td�ggdgddgd�}t�||� d S )N�2010�2011r � )r" r# g �?g @r# )r! F)r3 z2010-07-02 12:00:00r" )r! r )r! r& �2012rH r )r r3 z2011-07-02 12:00:00) r r4 � to_datetimer r r r r r, )r r r r r r r �test_quantile_datetime� s0 z,TestDataFrameQuantile.test_quantile_datetimec C sJ d}x@ddddgddggD ](}t jt|d�� |�|� W d Q R X qW d S )Nz2percentiles should all be in the interval \[0, 1\]r@ r g �?)r )r; r< rB r )r r- r> �invalidr r r �test_quantile_invalid$ s z+TestDataFrameQuantile.test_quantile_invalidc C sr t td�td�td�gtddd�tddd�tddd�gt�d�t�d�t�d�gd ��}|jd d d �}ttd�tddd�t�d�gd d ddgd�}t�||� |jd gd d �}t td�tddd�t�d�ggd gd ddgd�}t�||� t td�tj td�td�gtd�td�tj td�gtddd�tj tddd�tddd�gtddd�tddd�tj tddd�gt�d�t�d�t�d�tj gtj t�d�t�d�t�d�gd�t d�d�}|jd d d �}ttd�td�tddd�tddd�t�d�t�d�gd t d�d�}t�||� |jd gd d �}t td�td�tddd�tddd�t�d�t�d�ggd gt d�d�}t�||� d S )Nz 2011-01-01z 2011-01-02z 2011-01-03z US/Eastern)r2 z1 daysz2 daysz3 days)r r r9 g �?F)r3 r r r9 )r r! )r! r& )r r" r r# r9 rH ZAaBbCc)r& ) r r r4 Z Timedeltar r r r r, �NaT�list)r r �resrJ r r r �test_quantile_box* s� z'TestDataFrameQuantile.test_quantile_boxc C sp t t�dd�t�dd�d��}tj|jd< |�d�}tddgdd gdd �}t�||� |�dd g�}t dd gdd gd�dd gd�}t� ||� |jddd�}tt�dd�dd�}t�||� |jdd gdd�}t t�dd�gd dd gd�}t� ||� tj|d < |�d�}tdtjgdd gdd �}t�||� |�dd g�}t dd gtjtjgd�dd gd�}t� ||� d S )Nr g @)r" r# )r@ r g �?g @g @r" r# )r! r g �?g @g @)r! )r g �?)r r ) r r( �arangerP �ilocr r r r r, )r r r\ rJ r r r �test_quantile_nan� s* "z'TestDataFrameQuantile.test_quantile_nanc C s t dtjtjtjgi�}|jddd�}ttjgdgdd�}t�||� |jdgdd�}t dtjgidgd�}t�||� t td�td�td �gtjtjtjgd ��}|jddd�}ttd�tjgdd gdd�}t�||� |jdgdd�}t td�tjggdgdd gd �}t�||� d S ) Nr" g �?F)r3 )r! r )r! z 2012-01-01z 2012-01-02z 2012-01-03)r" r# r# )r! r& ) r r4 rZ r r r r r, r )r r r\ rJ r r r �test_quantile_nat� s& z'TestDataFrameQuantile.test_quantile_natc C s� t ddgdd�}|�d�}ttjtjgddgdd�}t�||� |�dg�}t tjtjggddgdgd�}t�||� t ddgdd�}t ddgd d�}d S ) Nr" r# rF )r& r g �?)r! r )r& r! rE zdatetime64[ns])r r r r( rP r r r, )r r r\ rJ r r r �test_quantile_empty_no_rows� s z1TestDataFrameQuantile.test_quantile_empty_no_rowsc C s~ t tjddd��}d|j_|�d�}tg g dtjd�}d|j _t � ||� |�dg�}t g dgg d�}d|j_t � ||� d S )Nz1/1/18rT )r1 zcaptain tightpantsg �?)r! r r )r! r& ) r r4 r5 r& r r r r( rF r! r r r, )r r r r r r r � test_quantile_empty_no_columns s z4TestDataFrameQuantile.test_quantile_empty_no_columnsc C s� t tj�dd�dddgd�}|d d |d< |d }|sLt|jj�dksLt�|jd d � d |j d < |j d |d d ks|t�d S )Nr r r r r9 )r& r r: F)r3 �c r )r r ) r r( �randomZrandn�lenZ_mgr�blocksr% r �valuesr_ )r Zusing_array_managerr r6 r r r �test_quantile_item_cache s z.TestDataFrameQuantile.test_quantile_item_cache) �__name__� __module__� __qualname__r; �mark� parametrizer r r4 ZarraysZ SparseArrayr r0 r7 r? rC rK rM rO rQ rW rY r] r` ra rb rc ri r r r r r s0 + 9 ,{!# r c @ s� e Zd Zejejej�e d��ej j dd�d�ej dddd�ej ddd d �eje�d�d d �eje�d�d d �gdd� d�dd� �Zejdd� �Zdd� Zdd� Zdd� Zej �d�d d � �Zd d � Zej �d d!g g d"gd#g g d"gd!ejejgd$d%gd&gd#ejejgd$d%gd&gg�d'd(� �Zej �d)ejd*g g d"d*ej j d+d�d�d*ej ej gd$d%gd&d*gg�d,d-� �Z ej �d.ejejge d/�d"gg g d&gg�d0d1� �Z d2S )3�TestQuantileExtensionDtyper z#raises when trying to add Intervals)�reason)Zmarksz 2016-01-01� r: )r1 �freqz US/Pacific)r1 r2 ZInt64)r ZFloat64c C s t | j�S )N)�strr )r r r r �<lambda>1 � z#TestQuantileExtensionDtype.<lambda>)�paramsZidsc C s |j }d|_|S )Nr )�paramr )r �request�idxr r r r! & s z TestQuantileExtensionDtype.indexc C s( ||�� � }|tkr d|_ndg|_|S )Nr )�copyr r r& )r r! �frame_or_series�objr r r r| 9 s z TestQuantileExtensionDtype.objc C s( t |t�r|�|�}n|j|dd�}|S )NF)r3 )� isinstancer r )r r| �qsr r r r �compute_quantileD s z+TestQuantileExtensionDtype.compute_quantilec C s� t jt|�t jd�}t j�|� |j| }dddg}| �||�}t|d |d |d g|j |dd�}t |�|�}t � ||� d S ) N)r g �?r r r r@ r )r r! r ) r( r^ rf �intpre �shuffler_ r r r �typer � assert_equal)r r| r! �indexerr~ r r r r r �test_quantile_eaK s " z+TestQuantileExtensionDtype.test_quantile_eac C s� |j |jd< |j |jd< tjt|�tjd�}tj�|� |j| }dddg}| �||�}t |d |d |d g|j |dd �}t |�|�}t � ||� d S ) Nr r@ )r g �?r r ���r )r r! r )� _na_valuer_ r( r^ rf r� re r� r r r r� r r� )r r| r! r� r~ r r r r r �test_quantile_ea_with_na] s " z3TestQuantileExtensionDtype.test_quantile_ea_with_naz-ignore:Using .astype to convert:FutureWarningc C s� |j |jd d �< |�|j�|d d �< t�|j|jk�s:t�tjt |�tj d�}tj � |� |j| }dddg}| � ||�}|jdddgd|j d�}t||dd �}t|�|�}t�||� d S ) N)r g �?r r r@ T)Z allow_fillZ fill_valuer )r! r )r� r_ Zastyper r( �allZdtypesr% r^ rf r� re r� r Ztaker r� r r� )r r| r! r{ r� r~ r r r r r �test_quantile_ea_all_nar s z2TestQuantileExtensionDtype.test_quantile_ea_all_nac C s� t jt|�t jd�}t j�|� |j| }d}| �||�}td|d i|j dd�}t |t�rr|d }||ks~t �n t � ||� d S )N)r g �?r r )r r )r( r^ rf r� re r� r_ r r r r} r% r r )r r| r! r� r~ r r r r r �test_quantile_ea_scalar� s z2TestQuantileExtensionDtype.test_quantile_ea_scalarz*dtype, expected_data, expected_index, axisrF r rE r" r# r c C sB t ddg|d�}|jd|d�}t|dt|�dd�}t�||� d S )Nr" r# )r& r g �?)r rF )r r! r )r r r r r r )r r � expected_data�expected_indexr r r r r r r �test_empty_numeric� s z-TestQuantileExtensionDtype.test_empty_numericz:dtype, expected_data, expected_index, axis, expected_dtypezdatetime64[ns]z #GH 41544c C sD t ddg|d�}|jd|dd�}t|dt|�|d�}t�||� d S )Nr" r# )r& r g �?F)r r3 )r r! r )r r r r r r ) r r r� r� r Zexpected_dtyper r r r r r �test_empty_datelike� s z.TestQuantileExtensionDtype.test_empty_datelikez#expected_data, expected_index, axisr c C sd t t�ddg�ddgt�ddg�d��}|ddg jd |d �}t|d t|�tjd �}t� ||� d S ) NrR rS r rT rU )r" r# rH r" rH g �?)r )r r! r ) r r4 rV r r r r( rF r r )r r� r� r r r r r r r �test_datelike_numeric_only� s z5TestQuantileExtensionDtype.test_datelike_numeric_onlyN) rj rk rl r; Zfixturerw r4 Z IntervalIndexZ from_breaks�rangerm ZxfailZ period_ranger5 rI r( r^ r! r| r r� r� �filterwarningsr� r� rn rP r� rZ r� r� r r r r ro " sF $ &ro )r$ r( r; Zpandas.compat.numpyr Zpandasr4 r r r r Zpandas._testingZ_testingr r ro r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.08 |
proxy
|
phpinfo
|
Настройка