Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/__pycache__/test_qcut.cpython-37.pyc
Назад
B �5�g* � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZ d dlmZ d dlmZ d dlmZ d dlm Z m Z dd� Z dd � Z d d � Z d d � Z!dd� Z"dd� Z#dd� Z$dd� Z%dd� Z&ej'�(ddddg�d d � �Z)ej'�(dd d d ge*e+d!��g�d"d#� �Z,ej'�(d$d d d ged d d gdd%�fe*e+d!��ed dd&gdd%�fg�d'd(� �Z-ej'�(d)d*d+idfi d,fd*d-id,fd*did.fg�d/d0� �Z.ej'�(d1d2d3d4g�ej'�(d5dd&g�ej'�(ddd6g�d7d8� ���Z/ej'j(d9e ed:e d;g��e e d<e d=g��gd>d?� d@�dAdB� �Z0ej'�(dCd!e�1d ddD�g�dEdF� �Z2ej'�(dGedHd!dI�e dJd=dKg�ged:d!dI�edLdMdNg�gg�dOdP� �Z3ej'�(dCdQdRg�ej'�(dSe ej4fej5ej6fe*ej7fg�dTdU� ��Z8ej'�(dVd&dWdXg�dYdZ� �Z9dS )[� N) � Categorical� DatetimeIndex�Interval� IntervalIndex�NaT�Series�TimedeltaIndex� Timestamp�cut� date_range�isna�qcut�timedelta_range)�CategoricalDtype)�quantile)�Day�Nanoc C s� t j�d�} t| ddd�\}}t| ddddd g�}|jjj}t j||d d � d d �sXt �|jj j}t j||d d � d d �s~t �t | |dd�}t � ||� d S )Ni� � T)�retbinsr g �?g �?g �?g �?���g{�G�z�?)Zatol� )Zinclude_lowest)�np�random�randnr r � categories�left�valuesZallclose�AssertionError�rightr �tm�assert_categorical_equal)�arr�labels�binsZex_bins�result� ex_levels� r& �T/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/test_qcut.py� test_qcut s r( c C s4 t j�d�} t| ddd�}tt �|��dks0t�d S )Ni� � F)r" )r r r r �len�uniquer )r! �factorr& r& r' �test_qcut_bounds2 s r- c C s: t j�d�} t| dddddg�}t| d�}t�||� d S )N�d r g �?g �?g �?g �?r )r r r r r r )r! r, �expectedr& r& r' �test_qcut_specify_quantiles9 s r0 c C s< t jtdd��$ tddddddddddg d� W d Q R X d S )Nz edges.*unique)�matchr � )�pytest�raises� ValueErrorr r& r& r& r' �test_qcut_all_bins_sameA s r6 c C sN t �d�} t| d�}ttdd�tdd�tdd�tdd�g�}t�|j|� d S )Nr) r g����MbP�g @g @g @� )r �aranger r r r �assert_index_equalr )r �iir% r& r& r' �test_qcut_include_lowestF s r; c C s@ t j�d�} t j| d d�< t| d�}t|d d� ��� s<t�d S )Nr. � r )r r r �nanr r �allr )r! r$ r& r& r' � test_qcut_nasU s r? c C s> t ddgd�} tdd�tdd�g}t|dd�}t�| |� d S )Nr � g����MbP�r T)�ordered)r r r r r )r$ � intervalsr/ r& r& r' �test_qcut_index] s rC c C s� | t j�ddd��}t�|�}t|d�}g }g }xFt�|�D ]8}|j}|j}||ksXt �|� t |�� |� t |�� q<W xnt t |d d� |dd � �t |d d� |dd � ��D ]4\\} } \} } | | k s�t �| | k s�t �| | ks�t �q�W d S )NZreshape�dataz cut_data.csvr< r r ) �os�path�joinr Zloadtxtr r+ r r r �append�float�zip) ZdatapathZcut_filer! r$ ZstartsZendsZlev�s�e�spZsn�ep�enr& r& r' �test_qcut_binning_issuese s D rP c C s� t ddddddddd g �} t| dd d dg�}t�td d �td d�tdd �g�}t |�dddddddddg ���tdd��}t� ||� d S )Nr r r@ r2 r � � � � g Zd;�O�?g Zd;�O�?g����MbP�g Zd;�O@g Zd;�O@T)rA ) r r r �arrayr Ztake�astype�CDTr �assert_series_equal)�ser�resZ exp_levels�expr& r& r' �test_qcut_return_intervals~ s ,r\ r" Zfoor Tc C s8 t d�}d}tjt|d�� t|d| d� W d Q R X d S )NrQ zJBin labels must either be False, None or passed in as a list-like argument)r1 r )r" )�ranger3 r4 r5 r )r" r �msgr& r& r' �test_qcut_incorrect_labels� s r_ �a�b�cr2 c C s8 t d�}d}tjt|d�� t|d| d� W d Q R X d S )Nr) z9Bin labels must be one fewer than the number of bin edges)r1 r )r" )r] r3 r4 r5 r )r" r r^ r& r& r' � test_qcut_wrong_length_labels� s rc zlabels, expected)rA r@ c C s& t d�}t|d| d�}t�||� d S )Nr2 )r" )r] r r r )r" r/ r r$ r& r& r' �test_qcut_list_like_labels� s rd z kwargs,msg� duplicates�dropzBin edges must be unique�raisez(invalid value for 'duplicates' parameterc C s| dddddddg}|d k rDt jt|d�� t|df| � W d Q R X n4t|df| �}ttdd�tdd�g�}t�|j|� d S )Nr r r@ r2 )r1 g����MbP�) r3 r4 r5 r r r r r9 r )�kwargsr^ r r$ r/ r& r& r' �test_qcut_duplicates_bin� s ri zdata,start,end)g "@gsh��|�!@g "@)g g����MbP�g )g "�g��n� "�g "��lengthFc C sv t | g| �}t|d|d�}|d krRtt||�g| dd�}t |��tdd��}nt dg| tjd�}t� ||� d S ) Nr )r" r )�closedT)rA r )�dtype) r r r r rV rW r Zintpr rX ) rD �start�endrj r" rY r$ rB r/ r& r& r' �test_single_quantile� s ro rY Z20180101Z20180103z0 daysz2 daysc C s t | j�S )N)�strrl )�xr& r& r' �<lambda>� � rr )Zidsc C sf t �| d t� | d t� ftj| d t� | d fg�}tt|dd��}t| d�}t � ||� d S )Nr r@ T)rA ) r � from_tuplesr r r r= r r r r rX )rY rB r/ r$ r& r& r' � test_qcut_nat� s 8 ru r# r c C s� d}t tdd|d��}t|| �}t tttd|d�td|d��ttd|d�td|d��ttd|d�td |d��g���td d ��}t� ||� d S ) Nz US/EasternZ20130101r2 )�periods�tzz 2012-12-31 23:59:59.999999999)rw z2013-01-01 16:00:00z2013-01-02 08:00:00z2013-01-03 00:00:00T)rA ) r r r r r r rV rW r rX )r# rw rY r$ r/ r& r& r' �test_datetime_tz_qcut� s rx zarg,expected_binsZ1day)rv z1 daysz3 daysz 2018-01-01z 2018-01-02z 2018-01-03c C s* t | �}t|ddd�\}}t�||� d S )Nr@ T)r )r r r r9 )�argZ expected_binsrY r$ Z result_binsr& r& r' �test_date_like_qcut_bins s rz rR rS z box, comparec C sV |dddddgd �}|dddddgd �}t || dd�}t || dd�}|||� d S )Nr r r) FTrf )re )r )r# �box�compareZ data_expectedZ data_resultr/ r$ r& r& r' � test_qcut_bool_coercion_to_int s r} �qrQ r) c C sN t jt�d�|d�}t j|d d d�< t|| �}t|�t�| �}t� ||� d S )Nr. )rl r@ ) �pdrU r r8 ZNAr rV rI r r )r~ Zany_nullable_numeric_dtyper! r$ r/ r& r& r' �test_qcut_nullable_integer' s r� ):rE �numpyr r3 Zpandasr r r r r r r r r r r r r r Zpandas._testingZ_testingr Zpandas.api.typesr rW Zpandas.core.algorithmsr Zpandas.tseries.offsetsr r r( r- r0 r6 r; r? rC rP r\ �markZ parametrizer_ �listr] rc rd ri ro ru Zlinspacerx rz rX rU r Z assert_equalr} r� r& r& r& r' �<module> sh < ( * $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка