Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/__pycache__/test_reductions.cpython-37.pyc
Назад
B �5�g� � @ s� d dl mZ d dlmZ d dlZd dlmZ d dlZd dl Z d dl m Z d dl m mZ d dlmZ d dlZd dlmZmZmZmZmZmZmZmZm Z m Z m Z d dl m Z! d dl"m# m$Z$ d dl%m# m&Z& d0d d �Z'd1dd�Z(d2dd�Z)d3dd�Z*G dd� d�Z+G dd� d�Z,G dd� d�Z-dd� Z.d d � Z/e j0�1d dd dg�e j0�1d d d!g�d"d#� ��Z2e j0�1d$d!d d%d&d'g�d(d)� �Z3e j0�1d*ed+�ej4g�d,d-� �Z5d.d/� Z6dS )4� )� timedelta)�DecimalN)�tzlocal)�is_platform_windows)�is_categorical_dtype) � Categorical� DataFrame�Index� MultiIndex�Series� Timestamp� date_range�isna�notna� to_datetime� to_timedeltaTF��h㈵��>�:�0�yE>c sZ t || �} |r�| dkrtnd} tdtddd�i�} t�| �� t | | �� } W dQ R X t| t�sbt�t t | ��| d< t�| �� t | | �� } W dQ R X t| t�s�t�t | �s�t�|�r � fdd �} t� � |�}| d d d �}| d d d �}tj ||� | �|||d� tj ||j | d d�d ||d� n� }| d d�}| d d�}tj ||� |�|||d� | dk�r||j |d d�}tj ||d ||d� |�r�|jj}||jk�s�t�||jk�s�t�tjtdd�� | dd� W dQ R X |�rV|tj }t || �d d�}t || �d d�}| dk�rV| dk�rd nd }t||j|jd�}t� ||� t||j|jd�}t� ||� dS )a� Check that operator opname works as advertised on frame Parameters ---------- opname : str Name of the operator to test on frame alternative : function Function that opname is tested against; i.e. "frame.opname()" should equal "alternative(frame)". frame : DataFrame The object that the tests are executed on has_skipna : bool, default True Whether the method "opname" has the kwarg "skip_na" check_dtype : bool, default True Whether the dtypes of the result of "frame.opname()" and "alternative(frame)" should be checked. check_dates : bool, default false Whether opname should be tested on a Datetime Series rtol : float, default 1e-5 Relative tolerance. atol : float, default 1e-8 Absolute tolerance. skipna_alternative : function, default None NaN-safe version of alternative )�mean�medianN�bz1/1/2001� )�periods�ac s � | j �S )N)�values)�x)� alternative� �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/test_reductions.py�wrapperW s z$assert_stat_op_calc.<locals>.wrapperr F)�axis�skipna� )� check_dtype�rtol�atol)r )�sum�prodzNo axis named 2)�matchr' )�index�dtype)�getattr� FutureWarningr r �tm�assert_produces_warning� isinstancer �AssertionError�range�lenZ_make_skipna_wrapper�assert_series_equal�applyr r* �pytest�raises� ValueError�np�NaNr) )�opnamer �frame� has_skipnar# � check_datesr$ r% �skipna_alternative�fZexpected_warning�df�resultr �skipna_wrapper�result0�result1�expectedZ lcd_dtype�all_na�r0�r1�unitr )r r �assert_stat_op_calc! sl % rJ c C sp t || �dd� t || �dd� |rlt || �ddd� t || �ddd� t || �ddd� t || �ddd� dS )a� Check that API for operator opname works as advertised on frame Parameters ---------- opname : str Name of the operator to test on frame float_frame : DataFrame DataFrame with columns of type float float_string_frame : DataFrame DataFrame with both float and string columns has_numeric_only : bool, default False Whether the method "opname" has the kwarg "numeric_only" r )r r" T)r � numeric_onlyFN)r+ )r: � float_frame�float_string_frame�has_numeric_onlyr r r �assert_stat_op_api� s rO c sP t || �}|rl� fdd�}� fdd�}|ddd�}|ddd�}t�||�|�� tj||j|dd �dd � n� }� }|dd �}|dd �}t�||�|�� tj||j|dd �dd � tjtd d �� |d d � W dQ R X |�rL|tj } t | | �dd �} t | | �dd �} | dk�r0| � � �r t �| � � �rLt �n | � � �s>t �| � � �sLt �dS )a� Check that bool operator opname works as advertised on frame Parameters ---------- opname : str Name of the operator to test on frame alternative : function Function that opname is tested against; i.e. "frame.opname()" should equal "alternative(frame)". frame : DataFrame The object that the tests are executed on has_skipna : bool, default True Whether the method "opname" has the kwarg "skip_na" c s | � � j}� |�S )N)�dropnar )r Znona)r r r rB � s z+assert_bool_op_calc.<locals>.skipna_wrapperc s � | j �S )N)r )r )r r r r � s z$assert_bool_op_calc.<locals>.wrapperr F)r r! r" )r )r# zNo axis named 2)r( r N�any) r+ r- r3 r4 r5 r6 r7 r8 r9 rQ r0 �all) r: r r; r< r? rB r rC rD rF rG rH r )r r �assert_bool_op_calc� s6 rS c C s� |}t j�t|��dk|d< t|| �dd� t|| �dd� |r�t|| �ddd� t|| �ddd� t|| �ddd� t|| �ddd� d S ) a� Check that API for boolean operator opname works as advertised on frame Parameters ---------- opname : str Name of the operator to test on frame float_frame : DataFrame DataFrame with columns of type float float_string_frame : DataFrame DataFrame with both float and string columns has_bool_only : bool, default False Whether the method "opname" has the kwarg "bool_only" g �?Z_bool_r )r r" T)r � bool_onlyFN)r8 �random�randnr2 r+ )r: �bool_frame_with_narM � has_bool_only�mixedr r r �assert_bool_op_api� s rZ c - @ s� e Zd Zej�d�dd� �Zdd� Zej�d�dd� �Zej� d d d d d ddddg�dd� �Z ej� dd dd dddg�dd� �Z dd� Z dd � Z ej� d d d g�d d!� �Zej� d d d g�d"d#� �Zd$d%� Zd&d'� Zej� d(dd dg�d)d*� �Zd+d,� Zejd-d.� �Zej� d/d0d1gd2gd3gd4ged4gd4gd5�ed6g�ed7g�d8�fd9d1gd2gejgej ejge d:�eejgd4gd5�e j gee j g�d8�fd0d;d<ejejgd;d<ejejgd=ejejejged4ejejejgd4gd5�ed6d>d>d>g�ed7d?d?d?g�d@d=dAdBgdC�fd9d;d<ejejgd;d<ejejgd=ejejejgeejd4ejejgd4gd5�ed>d6d>d>g�ed?d7d?d?g�d@d=dAdBgdC�fg�dDdE� �Z dFdG� Z!dHdI� Z"dJdK� Z#dLdM� Z$dNdO� Z%ej� dPdQdRg�ej� dSd d0d9g�dTdU� ��Z&dVdW� Z'dXdY� Z(dZd[� Z)d\d]� Z*d^d_� Z+d`da� Z,dbdc� Z-ddde� Z.dfdg� Z/dhdi� Z0djdk� Z1dldm� Z2dndo� Z3dpdq� Z4ej� drdsd@dtgfdud@dvgfg�dwdx� �Z5dydz� Z6ej� d{d|d}g�d~d� �Z7ej� d{d|d}g�d�d�� �Z8d�d�� Z9ej� d�d@d=g�ej� d�d|d}g�ej� d�d0d9g�d�d�� ���Z:d�d�� Z;d�d�� Z<ej� d�ej=i d9fej>i d0fej=d�g id9fej>d�g id0fej=d�d9d9gid9fej>d�d9d9gid9fej=d�d0d9gid0fej>d�d0d9gid9fej=d�d0d0gid0fej>d�d0d0gid0fej=d9gd9gd��d9fej>d9gd9gd��d9fej=d9d9gd9d0gd��d0fej>d9d9gd9d0gd��d9fej>d�e?d�d3gd�d:�id9fej=d�e?d�d3gd�d:�id0fej>d�e?d@d=ge@d:�id9fej=d�e?d@d=ge@d:�id0fe�Aej>d�e?d@d=gd�d:�id9�e�Aej>d�e?d@d=gd�d:�id9�e�Aej=d�e?d@d=gd�d:�id0�e�Aej=d�e?d@d=gd�d:�id0�e�Aej>d�e?d=dAgd�d:�id0�e�Aej>d�e?d=dAgd�d:�id0�e�Aej=d�e?d=dAgd�d:�id0�e�Aej=d�e?d=dAgd�d:�id0�e�Aej>d�e?d@d=gd�d:�id9�e�Aej=d�e?d@d=gd�d:�id0�e�Aej>d�e?d=dAgd�d:�id0�e�Aej=d�e?d=dAgd�d:�id0�ej>d�e?d@d=gd�d:�id0fej=d�e?d@d=gd�d:�id9fej>d�e?d=dAgd�d:�id0fej=d�e?d=dAgd�d:�id9fe�Aej>e?d�d�gd�d:�e?d�d�gd�d:�d��d0�g#�d�d�� �ZBd�d�� ZCd�d�� ZDej� d d|d}g�d�d�� �ZEd�d�� ZFd S )��TestDataFrameAnalyticsz)ignore:Dropping of nuisance:FutureWarningc C s� t d||� t d||� t d||dd� t d||� t d||� t d||� t d ||� t d ||� t d ||dd� t d ||� t d ||� t d||� t d||� y,ddlm}m} t d||� t d||� W n tk r� Y nX d S )N�countr& �nuniqueF)rN r �productr �min�max�mad�var�std�semr )�kurtosis�skewrf �kurt)rO � scipy.statsre rf � ImportError)�selfrL rM re rf r r r �test_stat_op_api s* z'TestDataFrameAnalytics.test_stat_op_apic C s4 dd� }dd� }dd� }dd� }d d � }d d � }d d� } dd� } t d||dddd� t dtj|�d�ddd� t dtj|tjd� t dtj|dd� t d tj|tjd� t d ||� t d ||� t d ||� t d ||� t d!||dddd� y,d"d#lm } m } t d$| |� t d%| |� W n t k �r. Y nX d S )&Nc S s t | ��� S )N)r r& )�sr r r r\ 0 s z7TestDataFrameAnalytics.test_stat_op_calc.<locals>.countc S s t t�| �� ��S )N)r2 � algorithmsZunique1drP )rl r r r r] 3 s z9TestDataFrameAnalytics.test_stat_op_calc.<locals>.nuniquec S s t �| | �� ��� S )N)r8 �absr )r r r r ra 6 s z5TestDataFrameAnalytics.test_stat_op_calc.<locals>.madc S s t j| dd�S )Nr" )�ddof)r8 rb )r r r r rb 9 s z5TestDataFrameAnalytics.test_stat_op_calc.<locals>.varc S s t j| dd�S )Nr" )ro )r8 rc )r r r r rc < s z5TestDataFrameAnalytics.test_stat_op_calc.<locals>.stdc S s t j| dd�t �t| �� S )Nr" )ro )r8 rc �sqrtr2 )r r r r rd ? s z5TestDataFrameAnalytics.test_stat_op_calc.<locals>.semc S s* ddl m} t| �dk r tjS || dd�S )Nr )rf � F)�bias)rh rf r2 r8 �nan)r rf r r r �skewnessB s z:TestDataFrameAnalytics.test_stat_op_calc.<locals>.skewnessc S s* ddl m} t| �dk r tjS || dd�S )Nr )re � F)rr )rh re r2 r8 rs )r re r r r rg I s z6TestDataFrameAnalytics.test_stat_op_calc.<locals>.kurtr] FT)r< r# r= r& Zfloat32g����MbP?)r# r$ )r> r )r= r^ ra rb rc rd r\ r )re rf rf rg ) rJ r8 r& �astypeZnansumr r' ZnanprodZscipyre rf ri ) rj �float_frame_with_naZmixed_float_framer\ r] ra rb rc rd rt rg re rf r r r �test_stat_op_calc/ sV z(TestDataFrameAnalytics.test_stat_op_calczignore::RuntimeWarningc C s. dd� }t d||dd� t d||ddd� d S )Nc S s t | ��� rtjS t�| �S )N)r rQ r8 rs r )r r r r r � s z3TestDataFrameAnalytics.test_median.<locals>.wrapperr T)r= F)r# r= )rJ )rj rw � int_framer r r r � test_median� s z"TestDataFrameAnalytics.test_median�methodr& r r' rb rc rf r_ r` c C s� dddgdddgdddgd �}t |d d d gd d�}t tjdgtjdgtjdgd�td�}xX||gD ]L}|jjtjksxt�t||�d�}t|� d�|�d�}|dkrbt � ||� qbW d S )Ngȃw@Ba@�gC�V�Z�g ?Z��.F?r g g�94��c4?g�@�}�jX?g���N�)r r �c�foo�bar�baz�O)r) r* r rq ru )r r" r )r* r" Zf8)r& r' ) r r8 rs �objectr r* Zobject_r0 r+ rv r- r3 )rj r{ �data�df1�df2r@ rA rE r r r �%test_stat_operators_attempt_obj_array� s &z<TestDataFrameAnalytics.test_stat_operators_attempt_obj_array�oprg rd c C s� t ddddgddddgd d d d gd ��}tjtdd�� t||�� }W d Q R X t|�dks^t�t�dd��< tjtdd�� t||�� }W d Q R X t|�dks�t�W d Q R X d S )Nr" r rq ru g �?g @g @g @r r r| �d)�int�float�strzSelect only valid columns)r( �use_bottleneckF) r r- r. r, r+ r2 r0 �pd�option_context)rj r� r@ rA r r r �test_mixed_ops� s z%TestDataFrameAnalytics.test_mixed_opsc C s� t dddddgdddddgdd d d d gd ��}|jdddgd� |jdd�}t�|jtjdddgtd�� |j jdd�}t� ||� d S )NTF� � � �( �2 r r r| r� �e)� bool_data�int_data� string_datar� r� r� )�columnsr )r r � Zabcde)r* r" ) r �reindexr& r- Zassert_numpy_array_equalr r8 �arrayr� �Tr3 )rj r@ �testZaltr r r �test_reduce_mixed_frame� s z.TestDataFrameAnalytics.test_reduce_mixed_framec C s� t dddgdddgdtjdgd��}t�|�� tdddd��� t�|jdd�tdddd��� t�|jdd�tdddd��� t�|jddd �tdddd��� d S ) Nr" r rq )�A�B�CF)rP )r )r r" r )r rP )r r8 rs r- r3 r] r )rj r@ r r r � test_nunique� s $ z#TestDataFrameAnalytics.test_nunique�tzN�UTCc C s^ t ddgtd|d�gd d��}t�t�� |�� }W d Q R X tdgdgd�}t�||� d S ) Nr" �2000)r� r )r� r� g �?r� )r) )r r r- r. r, r r r3 )rj r� r@ rA rE r r r � test_mean_mixed_datetime_numeric� s z7TestDataFrameAnalytics.test_mean_mixed_datetime_numericc C sT t dtd|d�gd i�}t�t�� |�� }W d Q R X ttjd�}t� ||� d S )Nr� r� )r� r )r* ) r r r- r. r, r r r8 �float64r3 )rj r� r@ rA rE r r r � test_mean_excludes_datetimes� s z3TestDataFrameAnalytics.test_mean_excludes_datetimesc C s� dd t d�d�dd t d�d�dd t d�d�dd t d�d�d d t d �d�dd t d �d�dd t d �d�d d t d�d�dd t d�d�dd t d�d�g }t|�}tjtdd�� |�� }W d Q R X tddgddgd�}t�||� d S )Nr z628.00)r� r� r� r" z383.00rq z651.00z575.00ru z1114.00ZTESTz241.00z572.00z609.00z820.00� z1223.00zSelect only valid columns)r( g������@g�����L�@r� r� )r) )r r r- r. r, r r r3 )rj r� r@ rA rE r r r � test_mean_mixed_string_decimal� s z5TestDataFrameAnalytics.test_mean_mixed_string_decimalc C s� |j dd�}|�dd� �}t�||� |jdd�}|�dd� �}t�||� t�tj�d�dd�}tj |dd �}|dk � � r�t �t � d d ��$ tj |dd �}|dk � � r�t �W d Q R X d S ) Nru )ro c S s | j dd�S )Nru )ro )rc )r r r r �<lambda> � z5TestDataFrameAnalytics.test_var_std.<locals>.<lambda>c S s | j dd�S )Nru )ro )rb )r r r r r� r� )r" i� i� r )r r� F)rc r4 r- �assert_almost_equalrb r8 �repeatrU �nanopsZnanvarrQ r0 r� r� )rj �datetime_framerA rE �arrr r r � test_var_std s z#TestDataFrameAnalytics.test_var_std�methc C s t tj�dd�dddgd�}d|jd< t tj�dd�dddgd�}d |jd< t||�d d d �}t|ddg |�d d �}t�||� t||�d d d �}t|ddg |�d d �}t�||� d}tj t |d�� t||�d dd � W d Q R X d}tj t |d�� t||�d dd � W d Q R X d S )Nr� rq r} r~ r )r� �100)r r} r r" T)r rK )r z6unsupported operand type\(s\) for -: 'float' and 'str')r( Fz&could not convert string to float: 'a') r r8 rU rV �locr+ r- r3 r5 r6 � TypeError)rj r� r� r� rA rE �msgr r r �test_numeric_only_flag$ s z-TestDataFrameAnalytics.test_numeric_only_flagc C s� |j dd�}|�dd� �}t�||� t�tj�d�dd�}tj|dd�}|dk � � rZt �t � d d ��$ tj|dd�}|dk � � r�t �W d Q R X d S ) Nru )ro c S s | j dd�t�t| �� S )Nru )ro )rc r8 rp r2 )r r r r r� A r� z1TestDataFrameAnalytics.test_sem.<locals>.<lambda>)r" i� i� r )r r� F) rd r4 r- r� r8 r� rU r� ZnansemrQ r0 r� r� )rj r� rA rE r� r r r �test_sem? s z TestDataFrameAnalytics.test_semc C s� t dgdddgddggddddddgddddddgddddddggd�}ttj�d d �|d �}|�� }t�t�� |jdd �� d�}W d Q R X tj ||d d� |j d ks�t �|j dks�t �d S )Nr~ �one�two�threer r" r )Zlevels�codes� rq )r) )�levelF)Z check_names) r r r8 rU rV rg r- r. r, Zxsr3 �namer0 )rj r) r@ rg Zkurt2r r r � test_kurtL s 2 z TestDataFrameAnalytics.test_kurtzdropna, expectedT� g $@g �?r )� categoriesz2000-1-2z1 days)r� r� r� �D�E�F�GF)r* � � r" �NaTrs r r rq )�H�I�J�K�L�M�Nc C s� t ddddgddtjdgdtjtjtjgtjtjdtjgttjtjdtjg�tdd ddg�td d d d g�d d d d gd d d d gddtjtjgtdtjdtjg�td d ddg�td d d d g�tjddd�d��}|t|�� � j |d�}t |�}t � ||� d S )Nr� � � r� rq r" r r� z2000-1-2z1 daysrs r� r� ru �int64)r* )r� r� r� r� r� r� r� r� r� r� r� r� r� r� )rP ) r r8 rs r r r �arange�sorted�keys�moder- �assert_frame_equal)rj rP rE r@ rA r r r �test_mode_dropna[ s$ 7 z'TestDataFrameAnalytics.test_mode_dropnac C sp t dtjtjddgi�}t ddtjgi�}t�t��& |jdd�}|jdd�jdd�}W d Q R X t� ||� d S )Nr� r F)rP )ZbyT)Zdrop) r r8 rs r- r. � UserWarningr� Z sort_valuesZ reset_indexr� )rj r@ rE rA r r r �test_mode_sortwarning� s z,TestDataFrameAnalytics.test_mode_sortwarningc C sB t g ddgd�}|�� }t g ddgtg td�d�}t�||� d S )Nr r )r� )r* )r� r) )r r� r r� r- r� )rj r@ rA rE r r r �test_mode_empty_df� s z)TestDataFrameAnalytics.test_mode_empty_dfc C s t tdddd�tdddd�td�tddd� d ��}t |d |d |d |d d ��}|�� }|d |jd ksxt�|d |jd ks�t�|jdd�}||jd k�� s�t�|�� }|d |jd ks�t�|d |jd ks�t�|jdd�}||d k�� �st�|� � }t |�}t |d |d |d |d d ��}t � ||� t � ||� |� � }d|d < d|d< d|d< td�|d< |�� }t t�tdd��t�td d ��dddtd�g|jd �}t �||� t jtd d �� |jdd�}W d Q R X t dddgddd!gd �}t �||� |d d g �d�}t td d �gd �}t �||� |d d g �� }t tdd�td d �gd d gd �}t �||� t td"dd#�td$dd#�d%��}|d& |d' |d(< |d( jd)k�s�t�|d' |d& |d*< |�� |d( jd)k�s�t�|d* jd)k�s t�d S )+Nz2012-1-1rq r� )r �freqz2012-1-2Z20120101r� )�minutes�seconds)r� r� r� r� r� r� )r� r� r )r r� r" )r r� )r )r r� )r r� r} g �?r� Z20130101r� i1 )r� ���)�days)r) zSelect only valid)r( r Z20130102)r Z20130105)�time�time2r� r� Zoff1ztimedelta64[ns]Zoff2)r r r r r_ r� r0 rR r` rn r- r� �copyr r� � Timedeltar� r3 r. r, r* �_consolidate_inplace)rj r@ ZdiffsrA Zresult2rE rY r r r �test_operators_timedelta64� sn & & z1TestDataFrameAnalytics.test_operators_timedelta64c C s� t jddd�}t||d��}t j|jd< |jdd�}t|d �� t jgdd gd d �}t�||� |jd dd �}tt � d�gd t jt � d�g �}t�||� d S )Nz1 Dayr� )r )r� r� )���r� F)r! r� r� ztimedelta64[ns])r) r* r" )r r! r r� ) r� �timedelta_ranger r� �ilocrc r r- r3 r� )rj Ztdir@ rA rE r r r �!test_std_timedelta64_skipna_false s $z8TestDataFrameAnalytics.test_std_timedelta64_skipna_falsec C sZ t � }|�d�}|�d�}t|t�s(t�t|t�s6t�t|�dksFt�t|�dksVt�d S )Nr r" )r r& r/ r r0 r2 )rj Z empty_frameZaxis0Zaxis1r r r �test_sum_corner$ s z&TestDataFrameAnalytics.test_sum_cornerz method, unit)r& r )r' r" rK c C s� dddg}t ||g|tjgtjtjgd��}t||�|d�}t|||g|dd�}t�||� t||�|dd �}t||tjg|d �}t�||� t||�|d d �}t|||g|dd�}t�||� t|jdd � |�|dd �}t|tjtjg|d �}t�||� t |gd |gd tjgd d��}t||�|d d �}t|ddgd �}t�||� t||�|dd �}t|ddgd �}t�||� d S )Nr r r| )r r r| )rK r� )r) r* r" )rK � min_count)r) r r� r� )r� r� r� r� r� )r r8 rs r+ r r- r3 r� )rj r{ rI rK �idxr@ rA rE r r r �test_sum_prod_nanops. s* " & z+TestDataFrameAnalytics.test_sum_prod_nanopsc C s� dddg}t ddgdtjgtjtjgd��}|�t�}|�� }tdddgd|d�}t�||� |jdd�}t�||� |jd d�}tddtjgd|d�}t�||� d S ) Nr r r| r )r r r| zm8[ns])r* r) )r� r" ) r r8 rs r4 r r& r r- r3 )rj r� r@ r� rA rE r r r �test_sum_nanops_timedeltaP s " z0TestDataFrameAnalytics.test_sum_nanops_timedeltac C sN t dddgdddgd��}|jdd �}ttjtjgd d gd �}t�||� d S ) Nr" r rq ru r� r� )r �yr� )r� r r� )r) )r r& r r8 rs r- r3 )rj r@ rA rE r r r �test_sum_nanops_min_counte s z0TestDataFrameAnalytics.test_sum_nanops_min_countc C s6 |j �t�}t||j|jd�}|td� }|�� d S )N)r) r� r" )r rv r� r r) r� r r& )rj rL r r; Zdeltasr r r �test_sum_objectl s z&TestDataFrameAnalytics.test_sum_objectc C s" t �|�}|�d� |�d� d S )Nr" r )r8 �isnanr& )rj rL �boolsr r r � test_sum_boolr s z$TestDataFrameAnalytics.test_sum_boolc C sh t tddd�ddddgd���dddg�}tjtdd �� |�� }W d Q R X td d i�}t�||� d S ) Nr� ru )r r" r rq )r� r� zSelect only valid)r( r� g @) r r r� r- r. r, r& r r3 )rj r@ rA rE r r r �test_sum_mixed_datetimex s z.TestDataFrameAnalytics.test_sum_mixed_datetimec C s� t jtdd�� |jdd�}W d Q R X |jddd�}t �|j|j� t|j�t|j�k s\t �t jtdd�� |jdd�}W d Q R X |jddd�}t �|j|j� |d dk|d < |�d�}|d |d j �� ks�t �d S ) NzSelect only valid)r( r )r T)r rK r" r� �bool) r- r. r, r r& Zassert_index_equalr) r2 r� r0 r )rj rL rM Zthe_meanZthe_sumZmeansr r r �test_mean_corner� s z'TestDataFrameAnalytics.test_mean_cornerc C s� t t�d�tddd�tjddd�tjdddd�d��}|jd d �}tdd i�}t � ||� t � t �� |�� }W d Q R X td |j d d ��}t � ||� d S )Nrq z 2016-01-01)r �1D�2016r� )r r� )r� r� r� r� T)rK g �?)r" r� )r� r� )r r8 r� r r� r� � period_ranger r r- r3 r. r, r� )rj r@ rA rE r r r �test_mean_datetimelike� s z-TestDataFrameAnalytics.test_mean_datetimelikec C s� t t�d�tddd�tjddd�d��}|jdd�}td|jd |jd d��}t � ||� tj d dd d �|d< t j tdd�� |jdd� W d Q R X d S )Nrq z 2016-01-01)r r� )r� r� r� F)rK r" )r" r� )r" r� r� r� )r r� r� z"mean is not implemented for Period)r( )r r8 r� r r� r� r r r� r- r3 r� r5 r6 r� )rj r@ rA rE r r r �)test_mean_datetimelike_numeric_only_false� s z@TestDataFrameAnalytics.test_mean_datetimelike_numeric_only_falsec C sD t jjddd�}t|dd�}|jdd�}t|��� }t�||� d S )Ni� )r� r� )�size�Int64)r* T)rK )r8 rU �randintr r r- r3 )rj r� r@ rA rE r r r �*test_mean_extensionarray_numeric_only_true� s zATestDataFrameAnalytics.test_mean_extensionarray_numeric_only_truec C sF t jtdd��. |�d� |�d� |�d� |�d� W d Q R X d S )NzSelect only valid columns)r( r" )r- r. r, rc rb r rf )rj rM r r r �test_stats_mixed_type� s z,TestDataFrameAnalytics.test_stats_mixed_typec C s8 t td�td�d�}t|�}|jdd�d dks4t�d S )Nr" r� )r) r� )r r )r r1 r r& r0 )rj r@ r� r r r �test_sum_bools� s z%TestDataFrameAnalytics.test_sum_boolsc C s� |}t j|jdd�< t j|jdd�dd �f< xXdD ]P}xJdD ]B}x<||gD ]0}|j||d�}|jtj||d�}t�||� qJW q<W q2W d } tj t | d �� |jd d � W d Q R X d S ) Nr� r� � r� r� )TF)r r" )r r! z)No axis named 2 for object type DataFrame)r( r )r ) r8 rs r� �idxminr4 r r- r3 r5 r6 r7 ) rj rL ry r; r! r r@ rA rE r� r r r � test_idxmin� s z"TestDataFrameAnalytics.test_idxminc C s� |}t j|jdd�< t j|jdd�dd �f< xXdD ]P}xJdD ]B}x<||gD ]0}|j||d�}|jtj||d�}t�||� qJW q<W q2W d } tj t | d �� |jd d � W d Q R X d S ) Nr� r� r r� r� )TF)r r" )r r! z)No axis named 2 for object type DataFrame)r( r )r ) r8 rs r� �idxmaxr4 r r- r3 r5 r6 r7 ) rj rL ry r; r! r r@ rA rE r� r r r � test_idxmax� s z"TestDataFrameAnalytics.test_idxmaxc C sj t ddd�}tdddgtd�d d d� |d��}|�� }tdddgdddgd �}t�||� |�� }tdddgdddgd �}t�||� tj |j d < |�� }tdddgdddgd �}t�||� |�� }tdddgdddgd �}t�||� |d d d� |d < |� � |�� }tddddgdddd gd �}t�||� |�� }tddddgdddd gd �}t�||� d S ) Nz 2016-01-01rq )r r r r" r� )r" r rq )r) )r rq ru ) r r r1 r r r- r3 r r� r� r� r� )rj �dtir@ rA rE r r r �test_idxmax_mixed_dtype� s. $ z.TestDataFrameAnalytics.test_idxmax_mixed_dtypezop, expected_valuer ru r r� c C sn t ddddddgddddddgd�dd�}|�d �}t||�� }t d |itddgd d d �d �}t�||� d S )N�d �� r r" r )�ID�valuer )r* r r r� )r* r� )r) )r �groupbyr+ r r- r� )rj r� Zexpected_valuer@ rA rE r r r �!test_idxmax_idxmin_convert_dtypes s z8TestDataFrameAnalytics.test_idxmax_idxmin_convert_dtypesc C s� t ddd�}t||d d d� d��}tj|jd< |�� |jdd�}td ddg�}t� ||� |j dd�}td dd g�}t� ||� d S ) Nz 2016-01-01rq )r r� )rq ru )r r r" )r ru ) r r r� r� r� r� r r r- r3 r )rj r r@ rA rE r r r �"test_idxmax_dt64_multicolumn_axis12 s z9TestDataFrameAnalytics.test_idxmax_dt64_multicolumn_axis1r: rQ rR c C s t |||dd� d S )NT)rX )rZ )rj r: rW rM r r r � test_any_allD s z#TestDataFrameAnalytics.test_any_allc C s$ |� d�}t|tt|�|dd� d S )NT)r< )ZfillnarS r+ r8 )rj r: rW r@ r r r �test_any_all_bool_frameJ s z.TestDataFrameAnalytics.test_any_all_bool_framec C s. t dddgdddgdddgd�dddgd�}|dd g �d �}tdddgdddgd�}t�||� |dd g jd dd �}t�||� |�d �}tdddgdddgd�}t�||� |jd dd �}t�||� |jd d ��� }|dks�t�|jd d ��� }|dk�st�|d g jd d ��� }|dk�s*t�d S )NTF)r� r� r� r r r| )r) r� r� r" )rT )r r� )r rQ r r- r3 rR �itemr0 )rj r@ rA rE r r r �test_any_all_extraQ s* z)TestDataFrameAnalytics.test_any_all_extrar � bool_agg_funcr! c C sx t dtjtjdgtjdtjdgtjtjtjdgtjtjdtjggd�}t||�||d�}tddddg�}t�||� d S )Nr" Tr �5)r� )r r! )r r8 rs r+ r r- r3 )rj r r r! r@ rA rE r r r �test_any_all_object_dtyper s z0TestDataFrameAnalytics.test_any_all_object_dtypec C s` dt jdt jg}td�td�tjtjg}t||d��}|�d�}tddddg�}t� ||� d S )Nr" rq z 1960-02-15z 1960-02-16)r� r� TF) r8 rs r r� r� r rQ r r- r3 )rj Z float_dataZ datetime_datar@ rA rE r r r �test_any_datetime� s z(TestDataFrameAnalytics.test_any_datetimec C s� t dddgdddgd d d gd��}|jdd �}ttjd �}t�||� t dddgdddgd d d gd d dgd ��}|jdd �}td d i�}t�||� d S )Nr" r rq ru r� r� )�col1�col2�col3T)rT )r* F)r r r �col4r )r rR r r8 �bool_r- r3 )rj r@ rA rE r r r �test_any_all_bool_only� s z-TestDataFrameAnalytics.test_any_all_bool_onlyzfunc, data, expectedr� )r� r� g r� zM8[ns]z M8[ns, UTC]zm8[ns]�categoryr� r� c C s� t |�}d }tdd� |jD ��r$t}tj|ddd�� ||�}W d Q R X t|tj�sXt �|� � |ksht �tj|ddd�� t t |�|j �d d�}W d Q R X t|tj�s�t �|� � |ks�t �d S )Nc s s | ]}t |�V qd S )N)r )�.0r r r r � <genexpr>� s z>TestDataFrameAnalytics.test_any_all_np_func.<locals>.<genexpr>zSelect only valid columnsF)r( �check_stacklevel)r ) r rQ Zdtypesr, r- r. r/ r8 r r0 r r+ �__name__)rj �funcr� rE �warnrA r r r �test_any_all_np_func� s 5 z+TestDataFrameAnalytics.test_any_all_np_funcc C sL t �tddgd���� }|dks$t�t �tddgd���� }|dksHt�d S )Nr r )r� TF)r8 rR r r r0 rQ )rj rA r r r �test_any_all_object� s z*TestDataFrameAnalytics.test_any_all_objectc C s0 t ddgddgd���t�}|�� tddg�|d< |d �d�|d< |�� }|d dg }t�||� |jdd d �}tddgd dgd �}t� ||� |d dg jdd d �}t� ||� |jdd d �r�t �|j dd d �}tddgd dgd �}t� ||� |d dg j dd d �}t� ||� |j dd d ��s,t �d S ) Nr} r TF)r� r� r� r" r� r� r )rT r )r) ) r rv r� r� r Z_get_bool_datar- r� rR r3 r0 rQ )rj r@ �resrE r r r � test_any_all_object_bool_only� s& z4TestDataFrameAnalytics.test_any_all_object_bool_onlyc C st t dditjddgddggddgd�d �}d }tjt|d ��. t�t�� t ||�d dd � W d Q R X W d Q R X d S ) Nr� r" r� r r �out�in)�names)r) z.Must specify 'axis' when aggregating by level.)r( )r r� ) r r Z from_productr5 r6 r7 r- r. r, r+ )rj r{ r@ Zxprr r r �#test_any_all_level_axis_none_raises s z:TestDataFrameAnalytics.test_any_all_level_axis_none_raisesc C s� t dddg�}t dtjdtjgi�}tdddg�}ttjtjdg�}t�d ��4 |j|dd� x dD ]}t||�|dd� qdW W d Q R X d S ) Ng �?r� g @r" r )�lowerr )�lt�le�gt�ge�eq�ne)r )r r8 rs r r- r. Zclipr+ )rj r@ Zdf_nanrl Zs_nanr� r r r �test_series_broadcasting/ s z/TestDataFrameAnalytics.test_series_broadcasting)Gr& � __module__� __qualname__r5 �mark�filterwarningsrk rx rz � parametrizer� r� r� r� r� r� r� r� r� r� �tdZskip_if_no_scipyr� r r r r8 rs r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r r r r r! rQ rR r r� �paramr) r* r, r0 r8 r r r r r[ s� X " X ! $"! "r[ c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d g�ej� d d dg�e d�dd���Z dd� Z dd� Z ej� dddddddd d d d d d d!d"d#g�d$d%� �Zd&S )'�TestDataFrameReductionsc C s� t dtjtjtd�gi�}|�� }ttd�gdgd�}t�||� |�� }ttd�gdgd�}t�||� t dtjtjgi�}|�� }ttjgdgd�}t�||� |�� }ttjgdgd�}t�||� d S )Nr} z 2012-05-01)r) ) r r� r� r r_ r r- r3 r` )rj r@ r+ �expr r r �test_min_max_dt64_with_NaT> s z2TestDataFrameReductions.test_min_max_dt64_with_NaTc C s� |}t |t�r t� r t�d� ttd|d�td|d�gtd|d�tjgd��}|j ddd �}t |j d tjg�}|j |d j ks�t �t�||� |jddd �}t |j d tjg�}|j |d j ks�t �t�||� d S ) NzSGH#37659 OSError raised within tzlocal bc Windows chokes in times before 1970-01-01z2020-01-01 08:00:00)r� z1920-02-01 09:00:00z2020-02-01 08:00:00)r r r" F)r r! )r r r )r r )r/ r r r5 �skipr r r� r� r_ r r� r* r0 r- r3 r` )rj �requestZtz_naive_fixturer� r@ r+ rE r r r �'test_min_max_dt64_with_NaT_skipna_falseU s z?TestDataFrameReductions.test_min_max_dt64_with_NaT_skipna_falsec C s� t dtg �i�}ttg ��}|jdd�jtjk|�� tjkksBt�|jdd�jtjk|�� tjkksht�t � |jdd�|� t � |jdd�|� d S )Nr r )r r" ) r r r r_ r r� r� r0 r` r- r3 )rj r@ Zexpected_dt_seriesr r r �*test_min_max_dt64_api_consistency_with_NaTs s &&zBTestDataFrameReductions.test_min_max_dt64_api_consistency_with_NaTc C s� t dg i�}tg td�}t�|jdd�j�t�|�� �ks>t�t�|jdd�j�t�|�� �ksdt�t � |jdd�|� t � |jdd�|� d S )Nr )r* r )r r" ) r r r� r8 r� r_ r r0 r` r- r3 )rj r@ Zexpected_float_seriesr r r �*test_min_max_dt64_api_consistency_empty_df� s &&zBTestDataFrameReductions.test_min_max_dt64_api_consistency_empty_df�initialz2018-10-08 13:36:45+00:00z2018-10-08 13:36:45+03:00r{ r_ r` )rH c C s< t |�}t|g�}t|g�}t||�dd�}t�||� d S )Nr" )r )r r r r+ r- r3 )rj rH r{ Z initial_dtrE r@ rA r r r �test_preserve_timezone� s z.TestDataFrameReductions.test_preserve_timezonec C s� t ddddddddgidddddddgddddddd ggd �}tjtd d �� |jdd �}W d Q R X t dddgiddgd �}t�||� tjtd d �� |jdd �}W d Q R X t dddgiddgd �}t�||� d S )Nr� FTr� r� r r" r rq )r) zUsing the level)r( )r� )r r- r. r, rQ r� rR )rj r@ rA �exr r r � test_frame_any_all_with_level� s z5TestDataFrameReductions.test_frame_any_all_with_levelc C s� t tddg�ttdd�tdd�g�d��}|jdd�}tddgd d gd �}t�||� |jdd�}tddgd �}t�||� d S ) Nr rl r" �ms)r �t)r FTr rM )r� r) )r� )r r r rQ r- r3 )rj r@ rA rE r r r � test_frame_any_with_timedelta� s z5TestDataFrameReductions.test_frame_any_with_timedeltar' rQ rR r\ r& r' r r rf rg rd rb rc ra c C sV |dddgt �dddgdddgg�d�}tjtdd �� t||�d d � W d Q R X d S ) Nr" r rq ru r� r� )r) r� )r( r )r� )r Z from_arraysr- r. r, r+ )rj Zframe_or_seriesr' �objr r r �*test_reductions_deprecation_level_argument� s &zBTestDataFrameReductions.test_reductions_deprecation_level_argumentN)r& r9 r: rB rE rF rG r5 r; r= r� rI rK rN rP r r r r r@ = s8 r@ c @ sP e Zd Zej�dddg�dd� �Zdd� Zej�ddd g�d d � �Zd d � Z dS )�TestNuisanceColumnsr{ rQ rR c C s t ddgddd�}|�� }tjtdd�� t||�� W d Q R X tjtdd�� tt|�|� W d Q R X tjtdd�� t||�dd � W d Q R X tjt d d�� t||�d d �}W d Q R X t g t g �t d �}t� ||� tjt d dd �� tt|�|dd �}W d Q R X t� ||� d S )Nr r" r" r� )r* r� z does not implement reduction)r( F)rT zSelect only valid columns)r) r* )r( r% )r ) r �to_framer5 r6 r� r+ r8 r- r. r, r r� r3 )rj r{ �serr@ rA rE r r r �.test_any_all_categorical_dtype_nuisance_column� s" zBTestNuisanceColumns.test_any_all_categorical_dtype_nuisance_columnc C s& t dtdddddg�i�}|d }tjtdd�� |�� W d Q R X tjtdd�� |jdd� W d Q R X tjtd d�� |�� }W d Q R X t g t g �t j d �}t� ||� |d �t�|d < tjtdd�� |jdd� W d Q R X tjtd d�� |�� }W d Q R X t d gd gd �}t� ||� d S )Nr� r" r rq z does not implement reduction)r( F)rK zSelect only valid columns)r) r* r� g @)r) )r r r5 r6 r� r r- r. r, r r r8 r� r3 rv r� )rj r@ rS rA rE r r r �-test_median_categorical_dtype_nuisance_column� s& zATestNuisanceColumns.test_median_categorical_dtype_nuisance_columnr_ r` c C s� t ddddgdd�}t|�}|�d�}tjtdd�� t||�� W d Q R X tjtdd�� tt|�|� W d Q R X tjtdd�� t||�dd � W d Q R X tj t d d�� t||�� }W d Q R X tg t g �tj d �}t� ||� tj t d dd �� tt|�|�}W d Q R X t� ||� |d �t�|d < tj t d d�� t||�� }W d Q R X |dk�rntdgd gd�}ntdgd gd�}t� ||� tj t d dd �� tt|�|�}W d Q R X t� ||� d S )Nr r r| F)Zorderedr� z is not ordered for operation)r( )rK zSelect only valid columns)r) r* )r( r% r� r_ )r) )r r rR r5 r6 r� r+ r8 r- r. r, r r� r3 rv r� )rj r{ �catrS r@ rA rE r r r �:test_min_max_categorical_dtype_non_ordered_nuisance_column s< zNTestNuisanceColumns.test_min_max_categorical_dtype_non_ordered_nuisance_columnc C s� t dddgdddgd�td�}tjtd d �� |�� }W d Q R X td gd gd �}t�||� |d �t �d |d< tjtd d �� |�� }W d Q R X td dgd dgd �}t�||� d S )Nr r" r r r r| )r� r� )r* zSelect only valid columns)r( g �?r� )r) ru r� g @) r r� r- r. r, r r r3 rv r� )rj r@ rA rE r r r �3test_reduction_object_block_splits_nuisance_columnsO s zGTestNuisanceColumns.test_reduction_object_block_splits_nuisance_columnsN) r& r9 r: r5 r; r= rT rU rW rX r r r r rQ � s #0rQ c C s� t �d��t j��d��dd�} d| d< t| �}|jdd�}tt j d d �t j g�}t � ||� |jd dd �}t � ||� |jd dd �}tt j d d �t j dd �t j dd �t j g�}t � ||� d S )Nr� zm8[s]ru r ZNat)r� r� F)r! r� )r� r )r r! r" r� r� )r8 r� rv r� �viewZreshaper r& r r� r� r� r- r3 )r� r@ rA rE r r r �!test_sum_timedelta64_skipna_falsef s rZ c C sP t ddggtd�d�} | �ddi�} | �� }tddgddgd�}t�||� d S )Nr r" �ab)r� r r )r) )r �listrv r& r r- r3 )r@ rA rE r r r �!test_mixed_frame_with_integer_sum� s r] rK r{ r_ r` c C sn t �d�}t|jd |jgt�� d�}td|i�}t|| �|d�}tt|| �gt dgdd�d�}t � ||� d S )Nr� )r* r )rK r� )r) ) r8 Ziinfor r` r_ r� Z Int64Dtyper r+ r r- r3 )r{ rK Z int64_inforS r@ rA rE r r r �test_minmax_extensionarray� s r^ r� r& r r c C s� t tjdd�tjdd�gddgd��}|�� }t�t�� t|| �dd�}W d Q R X t�||� t|j dd�| �d d �}t�||� d S ) Nr� )�hours� g333333(@g������*@)r r r )r� F)rK ) r r� r� r� r- r. r, r+ r� r )r� r@ rE rA r r r �*test_groupby_regular_arithmetic_equivalent� s $ ra �ts_valuez 2000-01-01c C sb t dgdgdg| gd��}tjtdd�� |�� }W d Q R X tdddgtd�d�}t�||� d S ) Nr" g�������?r} )r r r| r� zDropping of nuisance)r( �abc)r) )r r- r. r, r& r r\ r3 )rb r@ rA rE r r r �.test_frame_mixed_numeric_object_with_timestamp� s rd c C sl t dddg�} | jdddd�}tdg�}t�||� t�d�}tjt |d�� | j dddd� W d Q R X d S ) Nr" r Tr F)r r� rK z2unsupported operand type(s) for +: 'int' and 'str')r( ) r r' r r- r3 �re�escaper5 r6 r� r& )r@ rA rE r� r r r �$test_prod_sum_min_count_mixed_object� s rg )TTFr r N)T)T)F)7�datetimer �decimalr re Z dateutil.tzr �numpyr8 r5 Z pandas.compatr Z pandas.util._test_decorators�utilZ_test_decoratorsr> Zpandas.core.dtypes.commonr Zpandasr� r r r r r r r r r r r Zpandas._testingZ_testingr- Zpandas.core.algorithms�corerm Zpandas.core.nanopsr� rJ rO rS rZ r[ r@ rQ rZ r] r; r= r^ ra r� rd rg r r r r �<module> sT 4 h @ ;
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка