U
|
ý°d½ ã @ sâ d dl Z d dlmZmZmZ d dlmZ e j dddg¡dd Z
|
dd Z d
|
d Z d d Z dd Ze j dddg¡dd Zdd Zdd Ze j dddd d gddd d gfd d d d!gd d d d gfg¡d"d# Zd$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 úWd:\z\workplace\vscode\pyvenv\venv\Lib\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 Ú+test_groupby_sample_unbalanced_groups_shape s r$ c C s® dgd dgd } t | | dddddddgd}| 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 Ú,test_groupby_sample_index_value_spans_groups( s r& c C s~ t ddgddgd} d}tjt|d | d¡jddd W 5 Q R X tjt|d | d¡d jddd W 5 Q R X d S )
|
Nr r r z0Please enter a value for `frac` OR `n`, not both©Úmatchr
|
g ð?r r ©r ÚpytestZraisesÚ
|
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 5 Q R X tjt|d | d¡d
|
jddd W 5 Q R X d S ) Nr r r zJReplace has to be set to `True` when upsampling the population `frac` > 1.r' r
|
ç ø?F©r Úreplacer r) r, r r r Ú:test_groupby_sample_frac_gt_one_without_replacement_raisesA s r2 r éÿÿÿÿr/ c C s t ddgddgd}| dk r$d}nd}tjt|d | d¡j| d W 5 Q R X tjt|d | d¡d
|
j| d W 5 Q R X d S ) Nr r r r z=A negative number of rows requested. Please provide `n` >= 0.z$Only integers accepted as `n` valuesr' r
|
r" r r) )r r r- r r r Ú$test_groupby_sample_invalid_n_raisesL s r4 c C s¢ dgd dgd } t | | d}| d¡jddd}dgd dgd } t | | d|jd
|
}t ||¡ | d¡d jddd}t| d |jd }t ||¡ d S ) Nr r r r r
|
g @Tr0 r r r r r r# r r r Ú test_groupby_sample_oversample\ s r5 c C s dgd dgd } t | | d}| d¡jd d d}t ddgddgd|jd}t ||¡ | d¡d jd d d}tddgd|jd }t ||¡ d S r r r# r r r Ú%test_groupby_sample_without_n_or_fracj s r6 zindex, expected_indexÚwÚxÚyÚzr% é r! é c C s® dgd dgd }t ||dt| d}| d¡jddddddgd}t ||dt|d}t ||¡ | d¡d jddddddgd}t|d t|d
|
}t ||¡ d S ) Nr r r r r
|
Tr )r r1 Úweightsr r )r r r r r r r r )r Zexpected_indexr r r r r r r Ú test_groupby_sample_with_weightsw s "r> c C sl dgd dgd } t | | | d}| d¡ddg jd d d}t ddgddgd |jd
|
}t ||¡ d S ) Nr r r )r
|
r Úcr
|
r r? r )r r? r )r r r r r r r# r r r Ú#test_groupby_sample_with_selections s
|
r@ c C s4 t g g d} | d¡}| ¡ }| }t ||¡ d S )Nr r
|
)r r r r r )r Z
|
groupby_dfr r r r r Ú%test_groupby_sample_with_empty_inputs s
|
|
rA )r* Zpandasr r r Zpandas._testingZ_testingr ÚmarkZ parametrizer r$ r& r. r2 r4 r5 r6 r> r@ rA r r r r Ú<module> s&
|
|
.þ
|
|
