U
|
ý°dé4 ã @ sb d dl Zd dlZd dlmZmZmZ d dlmZ d dl
|
m m Z G dd dZG dd dZdS )é N)Ú DataFrameÚIndexÚSeriesc @ sÈ e Zd Zejdd Zej dee d¡dd Z
|
dd Z d d
|
Z d d Z d d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zej d d d d d!d"gfd#d$g¡d%d& Zd'd( Zd)d* Zd+S ),Ú
|
TestSamplec C s0 |t krtj d¡}ntj dd¡}||d dS )Né
|
)Zdtype)r ÚnpÚrandomÚrandn)ÚselfÚframe_or_seriesZarr© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_sample.pyÚobj s zTestSample.objÚtestr c C sB 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 }}tdD ]4}t j |¡ | |jdd
|
¡ | |jdd ¡ qô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 s8 ÿ þþþþ
|
|
zTestSample.test_samplec C sL t |jdddkstt |jdddks0tt |jdddksHtd S )Nr r gÃõ(\ÂÕ?r é g
|
×£p=
|
×?)Úlenr ÚAssertionError©r
|
r r r r Útest_sample_lengths> s z TestSample.test_sample_lengthsc C s. d}t jt|d |jdd W 5 Q R X d S )Nzdrandom_state must be an integer, array-like, a BitGenerator, Generator, a numpy RandomState, or None©ÚmatchZa_string)r ©ÚpytestÚraisesÚ
|
ValueErrorr ©r
|
r Úmsgr r r Ú test_sample_invalid_random_stateD s ÿz+TestSample.test_sample_invalid_random_statec C s0 d}t jt|d |jddd W 5 Q R X d S )Nz0Please enter a value for `frac` OR `n`, not bothr( r# g333333Ó?)r r r* r. r r r Ú"test_sample_wont_accept_n_and_fracM s z-TestSample.test_sample_wont_accept_n_and_fracc C sP t jtdd |jdd W 5 Q R X t jtdd |jdd W 5 Q R X d S )Nz<A negative number of rows requested. Please provide `n` >= 0r( éýÿÿÿr z?A negative number of rows requested. Please provide `frac` >= 0g333333Ó¿r r* r& r r r Ú$test_sample_requires_positive_n_fracS s þþz/TestSample.test_sample_requires_positive_n_fracc C s* t jtdd |jdd W 5 Q R X d S )Nz$Only integers accepted as `n` valuesr( g @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 5 Q R X t jt|d dgd }|jd|d W 5 Q R X t jtd d" tddd
|
g}|jd |d W 5 Q R X d S ) Nú5Weights and axis to be sampled must be of same lengthr( r# r é ©r Úweightsç à?é 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_lengthsd s
|
z-TestSample.test_sample_invalid_weight_lengthsc C s: dgd }d}t jt|d |jd|d W 5 Q R X d S )Ng¹¿r z.weight vector many not include negative valuesr( r# r7 r* )r
|
r r; r/ r r r Ú test_sample_negative_weightsr s
|
z'TestSample.test_sample_negative_weightsc C s dgd }t j|d< d}tjt|d |jd|d W 5 Q R X dgd }t j |d< tjt|d |jd|d W 5 Q R X d S )Ng¹?r r z*weight vector may not include `inf` valuesr( r# r7 )r Úinfr+ r, r- r )r
|
r Zweights_with_infr/ Zweights_with_ninfr r r Útest_sample_inf_weightsy s
|
|
|
z"TestSample.test_sample_inf_weightsc C s6 dgd }t jtdd |jd|d W 5 Q R X d S )Nr r ú$Invalid weights: weights sum to zeror( r# r7 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 5 Q R X d S )Nr r@ r( r# r7 )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 r9 é r6 r ©r Úaxisr8 é )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# r6 r )znp.random.MT19937r# )znp.random.PCG64r: c C sh t tddtddd}t ||¡}|jdt||d}|jdt t||¡d}t ||¡ d S )Nr é é ©Úcol1Úcol2r# r ) r r r Úget_objr ÚevalÚcomr r )r
|
Zfunc_strÚargr r ÚresultÚexpectedr r r Útest_sample_random_state s
|
|
z#TestSample.test_sample_random_statec C s |t d¡}t j ¡ }|jd|d}|jd|d}|jj|jjk ¡ rLt|jdt j d¡d}|jdt j d¡d}t
|
||¡ d S )Nr é2 r r: ) r Zaranger Z default_rngr ÚindexÚvaluesÚallr% r r )r
|
r r ÚrngZresult1Zresult2r r r Útest_sample_generator s
|
z TestSample.test_sample_generatorc C sL t dtdi}t ||¡}d}tjt|d |jddd W 5 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 rO r+ r, r- r )r
|
r r r/ r r r Ú*test_sample_upsampling_without_replacement½ s ÿz5TestSample.test_sample_upsampling_without_replacementN)Ú__name__Ú
|
__module__Ú __qualname__r+ Zfixturer ÚmarkZ parametrizer^ r r" r' r0 r1 r3 r4 r< r= r? rA rC rI rU r[ r_ r r r r r s2
|
|
' ýþ
|
r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestSampleDataFramec C s dgd }d|d< t tddtdddgd |d}|jdd d
|
}t ||jdd
¡ ttd}d }tjt |d |jddd
|
W 5 Q R X d}tjt |d |jdddd W 5 Q R X tjt
|
dd |jddd
|
W 5 Q R X dgd }d|d< t |jd|d
|
|jd d
¡ t tddgd 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 5 Q R X d }tjt |d |jdd d W 5 Q R X ttd}tjt d d |jddd W 5 Q R X d!}tjt |d |jdddgd d W 5 Q R X d S )"Nr r r6 rD rJ rK Úa)rM rN Ú colStringÚ easyweightsrg r7 rG 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 r8 rF z-'String passed to weights not a valid column'Znot_a_real_column_namer9 rL rE rN ÚcolumnsÚrowsrW z)No axis named 2 for object type DataFramer )r rF z2No axis named not_a_name for object type DataFrameZ
|
not_a_namez&No axis named 1 for object type Seriesr5 ) r r r r Úassert_frame_equalrH r r+ r, r- ÚKeyError) r
|
Úeasy_weight_listÚdfÚsample1Úserr/ Zweights_less_than_1Zsecond_column_weightZweightr r r r" Ï sp
|
üÿ ÿ ÿ
|
ÿ ÿ
|
" ÿ z TestSampleDataFrame.test_samplec C s| dgd }d|d< t tddtdddgd d }|jdd|d
|
}t ||d g ¡ t |jdd d |jddd d¡ d S )Nr r# r6 r r rJ rK re ©rM rN rf rE rf é* r )r rF r )r r r r rj )r
|
rl rm rn r r r Útest_sample_axis1$ s
|
ÿ ÿz%TestSampleDataFrame.test_sample_axis1c C s t dddgdddgdddd 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 5 Q R X d S )NrD rG é re ÚbÚcrL é r# )rW r6 r ©r8 gü©ñÒMbP?i' r g{®Gáz?r r@ r( ) r r r rj Úlocr r+ r, r- )r
|
rm ro Zser2Zser3Zser4r r r Ú%test_sample_aligns_weights_with_frame4 s $ z9TestSampleDataFrame.test_sample_aligns_weights_with_framec C sH t tj dd¡dddgd}| d¡}t d ¡ d|d< W 5 Q R X d S ) Nr r# re rt ru )rh r6 Úd)r r r r r r Zassert_produces_warning)r
|
rm Zdf2r r r Útest_sample_is_copyI s
|
z'TestSampleDataFrame.test_sample_is_copyc C s t t jdt jg¡}| ¡ }tdddg}|j|d t ||¡ tdddgdt jt jgd}|d ¡ }|jdddd |d }t ||¡ d S )
|
Nr6 r r# rw )rX r8 r8 g ð?T)r r r8 )
|
r ÚarrayrB Úcopyr r r Zassert_numpy_array_equalr Zassert_series_equal)r
|
rS rT ro rm r r r Ú#test_sample_does_not_modify_weightsR s z7TestSampleDataFrame.test_sample_does_not_modify_weightsc C sR t tddtdddgd d}|jddd}ttd}tj|j|dd d S )
|
Nr rJ rK re rp r# T)Z ignore_index)Úexact)r r r r r Zassert_index_equalrW )r
|
rm rS Zexpected_indexr r r Útest_sample_ignore_indexd s ÿ z,TestSampleDataFrame.test_sample_ignore_indexN) r` ra rb r" rr ry r{ r~ r r r r r rd Ë s U rd )Únumpyr r+ Zpandasr r r Zpandas._testingZ_testingr Zpandas.core.commonÚcoreÚcommonrQ r rd r r r r Ú<module> s ?
|
