U
|
ý°d/ ã @ sê d dl Zd dlZd dlmZmZmZmZ d dlm Z
|
ej dd Z ej dd Z dd Zd d
|
Zd d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Z d%d& Z d'd( Z d)d* Z d+d, Z d-d. Z!d/d0 Z"d1d2 Z#d3d4 Z$d5d6 Z%ej& 'd7ej(d8ed9d:d;d8gid<d=ej(d>ed9d:d;d>gid?d=ej(d@ed9d:d;d@gidAd=ej(d ed9d:d;d gidBd=ej(dCed9d:d;dCgidDd=ej(dEed9d:d;dEgidFd=g¡dGdH Z)dIdJ Z*dKdL Z+ej& 'd7ej(dMed:d;dMgdMd:d8gdNd<d=ej(d ed:d;d gd d:d8gdNdBd=ej(dCed:d;dCgdCd:d8gdNdOd=ej(d>dPdQed:d;dPgd>d:d8gdNdRd=ej(ddCdQed:d;dCgdd:d8gdNdSd=ej(dEd@dCgdQed:d;d@dCggdEd:d8gdNdTd=g¡dUdV Z,dS )Wé N)Ú DataFrameÚSeriesÚ from_dummiesÚ get_dummiesc C s2 t dddgdddgdddgdddgdddgdS ©Né r ©Úcol1_aÚcol1_bÚcol2_aÚcol2_bÚcol2_c©r © r r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/test_from_dummies.pyÚ dummies_basic s ûÿr c C s2 t dddgdddgdddgdddgdddgdS r r r r r r Údummies_with_unassigned s ûÿr c C s0 dddg} t jtdd t| W 5 Q R X d S )Nr r zBExpected 'data' to be a 'DataFrame'; Received 'data' of type: list©Úmatch©ÚpytestÚraisesÚ TypeErrorr ©Údummiesr r r Útest_error_wrong_data_type' s
|
þr c C s@ t dddgdddgd} tjtdd t| W 5 Q R X d S )Nr r ©ÚaÚbúHDummy DataFrame contains unassigned value\(s\); First instance in row: 2r ©r r r Ú
|
ValueErrorr r r r r Ú(test_error_no_prefix_contains_unassigned0 s ýr" c C sH t dddgdddgd} tjtdd t| ddgd W 5 Q R X d S ) Nr r r úrExpected 'default_category' to be of type 'None', 'Hashable', or 'dict'; Received 'default_category' of type: listr ÚcÚd©Údefault_category©r r r r r r r r r Ú0test_error_no_prefix_wrong_default_category_type< s ýr) c C s@ t dddgdddgd} tjtdd t| W 5 Q R X d S )Nr r r zHDummy DataFrame contains multi-assignment\(s\); First instance in row: 2r r r r r r Ú%test_error_no_prefix_multi_assignmentH s ýr* c C sB t dddgddtjgd} tjtdd t| W 5 Q R X d S )Nr r r z0Dummy DataFrame contains NA value in column: 'b'r )r ÚnpÚnanr r r! r r r r r Ú!test_error_no_prefix_contains_nanT s ÿr- c C sN t ddddgddddgdddd gd
|
} tjtd d t| W 5 Q R X d S ) Nr é é r é Zc1Úc2Úc3Zc4©r r r$ ú(Passed DataFrame contains non-dummy datar r( r r r r Ú test_error_contains_non_dummies\ s "ÿþr5 c C sT t dddgdddgdddgdddgd} tjtdd t| dd W 5 Q R X d S )Nr r )r r
|
zcol2-azcol2-bz*Separator not specified for column: col2-ar Ú_©Úsepr r r r r Ú*test_error_with_prefix_multiple_seperatorsg s üÿþr9 c C s, t jtdd t| dgd W 5 Q R X d S )NzJExpected 'sep' to be of type 'str' or 'None'; Received 'sep' of type: listr r6 r7 r ©r r r r Ú%test_error_with_prefix_sep_wrong_typew s
|
ýr; c C s* t jtdd t| dd W 5 Q R X d S )Nr r r6 r7 ©r r r! r ©r r r r Ú*test_error_with_prefix_contains_unassigned s
|
ýr> c C s0 t jtdd t| dddgd W 5 Q R X d S )Nr# r r6 ÚxÚy©r8 r' r r= r r r Ú2test_error_with_prefix_default_category_wrong_type s
|
ýrB c C s0 t jtdd t| dddid W 5 Q R X d S )Nz^Length of 'default_category' \(1\) did not match the length of the columns being encoded \(2\)r r6 Úcol1r? rA r< r= r r r Ú9test_error_with_prefix_default_category_dict_not_complete s
|
ýrD c C s6 t j| jd< tjtdd t| dd W 5 Q R X d S )N©r0 r z5Dummy DataFrame contains NA value in column: 'col2_c'r r6 r7 )r+ r, Úlocr r r! r r: r r r Ú#test_error_with_prefix_contains_nan¥ s ÿrG c C s4 d| j d< tjtdd t| dd W 5 Q R X d S )NÚstrrE r4 r r6 r7 )rF r r r r r: r r r Ú+test_error_with_prefix_contains_non_dummies s
|
rI c C s\ t dddgdddgdddgdddgdddgd} tjtdd t| dd W 5 Q R X d S )Nr r r zHDummy DataFrame contains multi-assignment\(s\); First instance in row: 0r r6 r7 r r r r r Ú(test_error_with_prefix_double_assignment³ s ûÿ ýrJ c C sD t ddddg} t| }t|}tdddddgi}t ||¡ d S )Nr r r$ Ú )r r r r ÚtmÚassert_frame_equal©Ú
|
categoriesr ÚresultÚexpectedr r r Ú"test_roundtrip_series_to_dataframeÇ s
|
rR c C s< t dddddgi} t| }t|dd}| }t ||¡ d S )NrK r r r$ r6 r7 ©r r r rL rM rN r r r Ú&test_roundtrip_single_column_dataframeÏ s
|
rT c C sB t dddgdddgd} t| }t|dd}| }t ||¡ d S ©Nr r r$ ©rC Úcol2r6 r7 rS rN r r r Ú test_roundtrip_with_prefixes× s
|
rX c C sT t ddddgddddgddddgd} t dddddgi}t| }t ||¡ d S )Nr r r3 rK r r r$ ©r r rL rM ©r rQ rP r r r Ú test_no_prefix_string_cats_basicß s (r[ c C sT t ddddgddddgddddgd} t dddddgi}t| }t ||¡ d S )NTFr3 rK r r r$ rY rZ r r r Ú,test_no_prefix_string_cats_basic_bool_valuesæ s
|
|
|
ýÿr\ c C sT t ddddgddddgddddgd} t dddd dgi}t| }t ||¡ d S )
|
Nr r FTr3 rK r r r$ rY rZ r r r Ú2test_no_prefix_string_cats_basic_mixed_bool_valuesó s "ÿr] c C sb t ddddgddddgddddgddddgd} t dddddgidd }t| }t ||¡ d S )
|
Nr r )r é r0 é rK r^ r0 r_ Úobject©ZdtyperY rZ r r r Ú test_no_prefix_int_cats_basicü s ,ÿrb c C sb t ddddgddddgddddgddddgd} t dddddgid d
|
}t| }t ||¡ d S ) Nr r )ç ð?ç 9@ç @ç\Âõ(\@rK rc rd re rf r` ra rY rZ r r r Ú test_no_prefix_float_cats_basic s ,ÿrg c
|
C sx t dddddgdddddgdddddgdddddgdddddgd} t dddddd gid d
|
}t| }t ||¡ d S ) Nr r )ç®Gáz®ó?r$ r0 FNrK rh r$ r0 Fr` ra rY rZ r r r Ú test_no_prefix_mixed_cats_basic s ûÿ ri c C sL t dddgdddgdddgd} t ddddgi}t| }t ||¡ d S )Nr r )r r ÚNaNrK r r rj rY rZ r r r Ú:test_no_prefix_string_cats_contains_get_dummies_NaN_column s "rk zdefault_category, expectedr$ rK r r zdefault_category is a str)Úidr zdefault_category is a intg ô?zdefault_category is a floatzdefault_category is a 0Fzdefault_category is a bool)r r0 zdefault_category is a tuplec C s6 t dddgdddgd}t|| d}t ||¡ d S )Nr r r r& rY )r' rQ r rP r r r Ú+test_no_prefix_string_cats_default_category$ s $ rm c C s6 t dddgdddgd}t| dd}t ||¡ d S rU rY )r rQ rP r r r Útest_with_prefix_basicM s rn c
|
C sx t dddgdddgdddgdddgdddgdddgdddgd} t dddgdddgd}t| d d
|
}t ||¡ d S ) Nr r )r r
|
Zcol1_NaNr r r Zcol2_NaNr r rj r$ rV r6 r7 rY rZ r r r Ú0test_with_prefix_contains_get_dummies_NaN_columnS s ùÿ ro r? rV zdefault_category is a Falsere )rW rC z4default_category is a dict with int and float valuesz4default_category is a dict with bool and None valuesz5default_category is a dict with list and tuple valuesc C s t | d|d}t ||¡ d S )Nr6 rA )r rL rM )r r' rQ rP r r r Ú!test_with_prefix_default_categoryd s & ÿrp )-Únumpyr+ r Zpandasr r r r Zpandas._testingZ_testingrL Zfixturer r r r" r) r* r- r5 r9 r; r> rB rD rG rI rJ rR rT rX r[ r\ r] rb rg ri rk ÚmarkZ parametrizeÚparamrm rn ro rp r r r r Ú<module> sÎ
|
|
ýýýýýýæþ#
|
ýýýýý ýæþ#
|
