Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_sample.cpython-37.pyc
Назад
B �5�gH/ � @ sn d dl Zd dlZd dlmZ d dlmZmZmZ d dl m Z d dl m mZ G dd� d�ZG dd� d�ZdS )� N)�np_version_under1p18)� DataFrame�Index�Seriesc @ s� e Zd Zejeegd�dd� �Zej� de e d���dd� �Z dd � Z d d � Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zej� d d d d!d"d#gfejd$d!ejjed%d&�d'�ejd(d)ejjed%d&�d'�g�d*d+� �Zd,d-� Z d.S )/� TestSample)�paramsc C s6 |j }|tkr tj�d�}ntj�dd�}||d d�S )N� )Zdtype)�paramr �np�random�randn)�self�request�klassZarr� r �\/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_sample.py�obj s zTestSample.obj�testr c C sF t j�dd�}t�|jd|d�|jd|d�� t�|jd|d�|jd|d�� t�|jdt j�|�d�|jdt j�|�d�� t�|jdt j�|�d�|jdt j�|�d�� t�|jddt j�|�d �|jddt j�|�d �� g g }}x@td�D ]4}t j�|� |� |jdd �� |� |jdd �� q�W tj|� tj|� d S ) Nr �d � )�n� random_stategffffff�?)�fracr � T)r �replacer )r )r ) r r �randint�tm� assert_equal�sampleZ RandomState�range�seed�append)r r r r Zos1Zos2�_r r r � test_sample s, zTestSample.test_samplec C sL t |jdd�dk�st�t |jdd�dk�s0t�t |jdd�dk�sHt�d S )Nr )r g�(\���?)r � g ףp= �?)�lenr �AssertionError)r r r r r �test_sample_lengthsA s z TestSample.test_sample_lengthsc C s. d}t jt|d�� |jdd� W d Q R X d S )NzYrandom_state must be an integer, array-like, a BitGenerator, a numpy RandomState, or None)�matchZa_string)r )�pytest�raises� ValueErrorr )r r �msgr r r � test_sample_invalid_random_stateG s z+TestSample.test_sample_invalid_random_statec C s0 d}t jt|d�� |jddd� W d Q R X d S )Nz0Please enter a value for `frac` OR `n`, not both)r( r$ g333333�?)r r )r) r* r+ r )r r r, r r r �"test_sample_wont_accept_n_and_fracP s z-TestSample.test_sample_wont_accept_n_and_fracc C sT d}t jt|d�� |jdd� W d Q R X t jt|d�� |jdd� W d Q R X d S )NzCA negative number of rows requested. Please provide positive value.)r( ���)r g333333ӿ)r )r) r* r+ r )r r r, r r r �$test_sample_requires_positive_n_fracV s z/TestSample.test_sample_requires_positive_n_fracc C s* t jtdd�� |jdd� W d Q R X d S )Nz$Only integers accepted as `n` values)r( g������ @)r )r) r* r+ r )r r r r r � test_sample_requires_integer_n] s z)TestSample.test_sample_requires_integer_nc C s� d}t jt|d�� |jdddgd� W d Q R X t jt|d�� dgd }|jd|d� W d Q R X t jtd d��" tddd g�}|jd |d� W d Q R X d S ) Nz5Weights and axis to be sampled must be of same length)r( r$ r � )r �weightsg �?� z%Fewer non-zero entries in p than sizeg�������?r )r) r* r+ r r )r r r, � bad_weightsZbad_weight_seriesr r r �"test_sample_invalid_weight_lengthsb s z-TestSample.test_sample_invalid_weight_lengthsc C s: dgd }d}t jt|d�� |jd|d� W d Q R X d S )Ng��������r z.weight vector many not include negative values)r( r$ )r r3 )r) r* r+ r )r r r5 r, r r r � test_sample_negative_weightsp s z'TestSample.test_sample_negative_weightsc C s� dgd }t j|d< d}tjt|d�� |jd|d� W d Q R X dgd }t j |d< tjt|d�� |jd|d� W d Q R X d S )Ng�������?r r z*weight vector may not include `inf` values)r( r$ )r r3 )r �infr) r* r+ r )r r Zweights_with_infr, Zweights_with_ninfr r r �test_sample_inf_weightsw s z"TestSample.test_sample_inf_weightsc C s6 dgd }t jtdd�� |jd|d� W d Q R X d S )Nr r z$Invalid weights: weights sum to zero)r( r$ )r r3 )r) r* r+ r )r r Z zero_weightsr r r �test_sample_zero_weights� s z#TestSample.test_sample_zero_weightsc C s8 t jgd }tjtdd�� |jd|d� W d Q R X d S )Nr z$Invalid weights: weights sum to zero)r( r$ )r r3 )r �nanr) r* r+ r )r r Z nan_weightsr r r �test_sample_missing_weights� s z&TestSample.test_sample_missing_weightsc C s8 d gd }d|d< t �|jdd|d�|jdd� � d S )Nr g �?� r2 r )r �axisr3 � )r r r �iloc)r r Zweights_with_Noner r r �test_sample_none_weights� s z#TestSample.test_sample_none_weightsz func_str,argznp.arrayr r$ r2 r znp.random.MT19937z NumPy<1.18)�reason)Zmarksznp.random.PCG64r4 c C sl t tdd�tdd�d��}|tkr*|d }|jdt|�|�d�}|jdt�t|�|��d�}t�||� d S )Nr � � )�col1�col2rE r$ )r r ) r r r r �eval�comr r r )r Zfunc_str�arg�frame_or_seriesr �result�expectedr r r �test_sample_random_state� s z#TestSample.test_sample_random_statec C sP t dtd�i�}|tkr |d }d}tjt|d�� |jddd� W d Q R X d S )N�A�abczJReplace has to be set to `True` when upsampling the population `frac` > 1.)r( r F)r r )r �listr r) r* r+ r )r rJ r r, r r r �*test_sample_upsampling_without_replacement� s z5TestSample.test_sample_upsampling_without_replacementN) �__name__� __module__� __qualname__r) Zfixturer r r �markZ parametrizerP r r# r' r- r. r0 r1 r6 r7 r9 r: r< rA r Zskipifr rM rQ r r r r r s2 ( r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestSampleDataFramec C s� dgd }d|d< t tdd�tdd�dgd |d��}|jdd d �}t�||jdd � � ttd��}d }tjt |d �� |jddd � W d Q R X d}tjt |d �� |jdddd� W d Q R X tjt dd �� |jddd � W d Q R X dgd }d|d< t�|jd|d �|jd d� � t td�dgd d��}ddg}t�|jdd|d�|dg � t�|jdd|d�|dg � dgd }d|d< t�|jdd|d�|jdd � � t�|jdd|d�|jdd � � d}tjt |d �� |jdd d � W d Q R X d }tjt |d �� |jdd d � W d Q R X ttd��}tjt d d �� |jddd � W d Q R X d!}tjt |d �� |jdddgd d� W d Q R X d S )"Nr r r2 r= rC rD �a)rE rF � colString� easyweightsrY )r r3 r? z@Strings cannot be passed as weights when sampling from a Series.)r( r$ Z weight_columnzLStrings can only be passed to weights when sampling from rows on a DataFrame)r r3 r> z-'String passed to weights not a valid column'Znot_a_real_column_nameg �?)rE rF )r r> r3 rF �columns�rows�indexz)No axis named 2 for object type DataFramer )r r> z2No axis named not_a_name for object type DataFrameZ not_a_namez&No axis named 1 for object type Seriesz5Weights and axis to be sampled must be of same length) r r r r �assert_frame_equalr@ r r) r* r+ �KeyError) r �easy_weight_list�df�sample1�serr, Zweights_less_than_1Zsecond_column_weight�weightr r r r# � sZ " z TestSampleDataFrame.test_samplec C s| dgd }d|d< t tdd�tdd�dgd d ��}|jdd|d �}t�||d g � t�|jdd d �|jddd d�� d S )Nr r$ r2 r r rC rD rW )rE rF rX )r r> r3 rX �* )r r )r r> r )r r r r r] )r r_ r` ra r r r �test_sample_axis1 s z%TestSampleDataFrame.test_sample_axis1c C s t dddgdddgd�ddd gd �}td d d gd ddgd �}t�|jd g |jd |d �� tdd dgd ddgd �}t�|jd g |jd |d �� tdd gd dgd �}t�|jd g |jd |d �� td d gd dgd �}tjtdd�� |jd |d � W d Q R X d S )Nr= r? � rW �b�c)rE rF � r$ )r\ r2 r )r3 g����MbP?i' r g{�G�z�?r z$Invalid weights: weights sum to zero)r( ) r r r r] �locr r) r* r+ )r r` rb Zser2Zser3Zser4r r r �%test_sample_aligns_weights_with_frame, s $ z9TestSampleDataFrame.test_sample_aligns_weights_with_framec C sH t tj�dd�dddgd�}|�d�}t�d �� d|d< W d Q R X d S ) Nr r$ rW rg rh )rZ r2 �d)r r r r r r Zassert_produces_warning)r r` Zdf2r r r �test_sample_is_copyB s z'TestSampleDataFrame.test_sample_is_copyc C sP t tdd�tdd�dgd d��}|jddd�}td d d g�}t�|j|� d S ) Nr rC rD rW )rE rF rX r$ T)Z ignore_indexr r2 r )r r r r r Zassert_index_equalr\ )r r` rK Zexpected_indexr r r �test_sample_ignore_indexK s z,TestSampleDataFrame.test_sample_ignore_indexN)rR rS rT r# re rk rm rn r r r r rV � s U rV )�numpyr r) Z pandas.compatr Zpandasr r r Zpandas._testingZ_testingr Zpandas.core.common�core�commonrH r rV r r r r �<module> s 5
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка