Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/apply/__pycache__/test_frame_transform.cpython-37.pyc
Назад
B �5�g�% � @ s d dl Z d dlZd dlZd dlmZmZmZ d dlm Z d dl m Z d dl mZ dd� Zdd� Zej�d e �d d � �Zej�d ejgd gfejejgdd gfe�ejg�d gfe�ejejg�dd gfg�dd� �Zej�dg e�g �g�dd� �Zej�deeg�dd� �Zdd� Zej�di dg ig dd�dg d�g dgd�dgg d�g�d d � �Z ej�d d d g�d!d"� �Z ej�d#d$d%d&dd'g�d(d)� �Z d*d+d,d-d.d%gZ d/d0� e D �Z ej�d e d1d2� f��d3d4� �Z!ej�d e �d5d6� �Z"d7d8� Z#ej�d d d g�d9d:� �Z$d;d<� Z%dS )=� N)� DataFrame� MultiIndex�Series)�frame_transform_kernels)� zip_framesc C s, |t k r(| d } |dkr(t�d|� �� | S )zm Helper to ensure we have the right type of object for a test parametrized over frame_or_series. �Ar z%Test is only for DataFrame with axis=)r �pytest�skip)�obj�klass�axis� r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/apply/test_frame_transform.py� unpack_obj s r c C sR t ||| �}tjdd�� t�|�}W d Q R X |jtj| d�}|}t�||� d S )N�ignore)�all)r )r �np�errstate�sqrt� transform�tm� assert_equal)r � float_frame�frame_or_seriesr Zf_sqrt�result�expectedr r r �test_transform_ufunc s r �opc C sF |r*|dkr*| dkr*|j �tjjdd�� |dkr8dgng }| dksL| dkr^t�|jd �}nt�|jd �}|j|| d �j |f|�� }|j || f|�� }t � ||� d |j ks�t �|d �� |d < t|jj�d ks�t �| dks�| dkr�t�|jd �}nt�|jd �}|j|| d �j |f|�� } |j || f|�� } t � | | � d S ) N� pct_change)� �columnsz1shift axis=1 not yet implemented for ArrayManager)�reason�fillnag r �indexr )r �Er )�node� add_markerr �mark�xfailr �ones�shape�groupbyr r �assert_frame_equalr �AssertionError�copy�lenZ_mgrZarrays) r r r Zusing_array_manager�request�argsr) r r Z expected2Zresult2r r r � test_transform_groupby_kernel) s( r2 z ops, namesr �absolutec s� | dkr dnd}t jdd�� t� fdd�|D �|d�}W d Q R X | dkr^t�� j|g�|_nt�� j|g�|_� j|| d�}t� ||� d S ) N> r r# r r r )r c s g | ] }|� ��qS r r )�.0r )r r r � <listcomp>X s z+test_transform_listlike.<locals>.<listcomp>)r ) r r r r Z from_productr r# r r r, )r r �ops�namesZ other_axisr r r )r r �test_transform_listlikeK s $r8 r6 c C s4 t | |d�}tjtdd�� |�|� W d Q R X d S )Nr z$No transform functions were provided)�match)r r �raises� ValueErrorr )r r6 r r r r r � test_transform_empty_listlikea s r< �boxc C st | dks| dkr.|j d }||g �tj�}n |jd }|jdg �tj�}|j||tji�| d�}t�||� d S )Nr r# )r )r r r �absr# Zilocr r, )r r r= �er r r r r �test_transform_dictlikei s r@ c C sj t ddgddgddgd��} | �ddgdd��}t dddgd dd ggtdd gd d g�d �}t�||� d S )Nr � � )�a�b�cr r> )rD rE g �?g @)r r> )r r r )r r r )r )r r r r r, )�dfr r r r r � test_transform_dictlike_mixedv s rG r Zcumsum)r �Bc C s4 t | |d�}tjtdd�� |�|� W d Q R X d S )Nr z$No transform functions were provided)r9 )r r r: r; r )r r6 r r r r r � test_transform_empty_dictlike� s rI � use_applyTFc s@ t |� | �}� �fdd�}|j|| d�}|d }t�||� d S )Nc s �t | � �krt�| d S )Nr )� isinstancer; )�x)r rJ r r �func� s z test_transform_udf.<locals>.func)r r )r r r r )r r rJ r r rM r r r )r rJ r �test_transform_udf� s rN �methodr> �shiftr �rankc C s8 t dddgi�}|�| �}t�| �|�}t�||� d S )Nr ���rA )r r �operator� methodcallerr r, )rO rF r r r r r �test_transform_method_name� s rU ZffillZbfillr" �padZbackfillc C s g | ]}|t kr|�qS r )� wont_fail)r4 rL r r r r5 � s r5 c C s | d S )Nr r )rL r r r �<lambda>� � rX c C s | dkr |j �tjjtdd�� tddtg i�}|tk rB|d }| dkrNd nt}t � |��� tj t dd�� |� | � W d Q R X tj t d d�� |� | g� W d Q R X tj t d d�� |� d| i� W d Q R X tj t d d�� |� d| gi� W d Q R X W d Q R X d S ) NrQ z)GH 40418: rank does not raise a TypeError)r: r! r � Ztshiftz!unsupported operand|not supported)r9 zTransform function failed)r% r& r r' r( r; r �object� FutureWarningr �assert_produces_warningr: � TypeErrorr )r r r0 r �warnr r r �test_transform_bad_dtype� s" r` c C s� | dkrt �d� tdtg dddgd��}|dg �| g�}|�| g�}t�||� |dg �d| i�}|�| | d��}t�||� |dg �d| gi�}|�| g| gd��}t�||� |�dg| gd��}|�| dg| gd��}t�||� d S ) NrQ z)GH 40418: rank does not raise a TypeErrorrZ r rA )r rH rH rP )r r r r[ r r r )r rF r r r r r �(test_transform_partial_failure_typeerror� s ra c C sZ d} dd� }t dddgddd gd ��}|d g �|g�}tjt| d �� |�|g�}W d Q R X t�||� |d g �d |i�}tjt| d �� |�||d ��}W d Q R X t�||� |d g �d |gi�}tjt| d �� |�|g|gd ��}W d Q R X t�||� |�d g|gd ��}tjt| dd�� |�|d g|gd ��}W d Q R X t�||� d S )Nz>.*did not transform successfully and did not raise a TypeErrorc S s t �t �| ��dk rt�| S )N� )r �sumr; )rL r r r r � s z5test_transform_partial_failure_valueerror.<locals>.opr rA rZ i� i� iX )r rH rH )r9 rP F)r9 Zcheck_stacklevel)r r r r] r\ r )r9 r rF r r r r r �)test_transform_partial_failure_valueerror� s& rd c s@ ddg� ddi�� ���fdd�}�dg�j |df� ��� d S )Nr rA rE rZ c s6 �t | ��krt�||g� ks"t�|�d ks2t�| S )NrE )rK r; r- )rL rC rD rE )� expected_args�expected_kwargsr rJ r r �f s z%test_transform_passes_args.<locals>.fr )r )rJ r rg r )re rf r rJ r �test_transform_passes_args s rh c C sP t g ddgd�} | �dd� �}t�|| � | d �dd� �}t�|| d � d S )NZcol1Zcol2)r c S s | d S )Nrb r )rL r r r rX rY z0test_transform_empty_dataframe.<locals>.<lambda>c S s | d S )Nrb r )rL r r r rX ! rY )r r r r, Zassert_series_equal)rF r r r r � test_transform_empty_dataframe s ri )&rS �numpyr r Zpandasr r r Zpandas._testingZ_testingr Zpandas.tests.apply.commonr Zpandas.tests.frame.commonr r r r' Z parametrizer2 r r> �arrayr8 r< �dictr@ rG rI rN rU rW Zframe_kernels_raiser` ra rd rh ri r r r r �<module> sF " &
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка