Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/__pycache__/test_sample.cpython-37.pyc
Назад
B �5�g � @ s� d dl Z d dlmZmZmZ d dlmZ e j� dddg�dd� �Z dd � Zd d� Zdd � Z dd� Ze j� dddg�dd� �Zdd� Zdd� Ze j� dddddgddddgfddd d!gddd d gfg�d"d#� �Zd$d%� ZdS )&� N)� DataFrame�Index�Serieszn, frac)� N)Ng�������?c C s� dgd dgd }t ||d��}|�d�j| |d�}dgd dgd }t ||d�|jd�}t�||� |�d�d j| |d�}t|d|jd �}t�||� d S ) N� � r )�a�br )�n�frac)�indexr )�namer )r �groupby�sampler �tm�assert_frame_equalr �assert_series_equal)r r �values�df�result�expected� r �V/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/test_sample.py�)test_groupby_sample_balanced_groups_shape s r c C s� dgd dgd } t | | d��}|�d�jdd�}dgd dgd } t | | d�|jd �}t�||� |�d�d jdd�}t| d |jd�}t�||� d S )Nr 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 �+test_groupby_sample_unbalanced_groups_shape s r c C s� dgd dgd } t | | d�ddddddgd�}|�d�jdd�}dgd dgd } t | | d�|jd�}t�||� |�d�d jdd�}t| d|jd �}t�||� d S ) Nr � r )r r )r r )r r )r r )r r r r r r r r )r r r r r r r �,test_groupby_sample_index_value_spans_groups( s r c C s~ t ddgddgd��} d}tjt|d�� | �d�jddd� W d Q R X tjt|d�� | �d�d jddd� W d Q R X d S ) Nr r )r r z0Please enter a value for `frac` OR `n`, not both)�matchr g �?)r r r )r �pytest�raises� ValueErrorr r )r �msgr r r �%test_groupby_sample_n_and_frac_raises6 s r$ c C s~ t ddgddgd��} d}tjt|d�� | �d�jddd � W d Q R X tjt|d�� | �d�d jddd � W d Q R X d S )Nr r )r r zJReplace has to be set to `True` when upsampling the population `frac` > 1.)r r g �?F)r �replacer )r r r! r"