Ë
|
oñúh ã ó d dl Zd dlZd dlmZmZmZmZ d dl m
|
Z ej d « Z d Zd Zd Zd Zd Zd Zd
|
Zd Zd Zd Zd Zy)é N)Ú DataFrameÚSeriesÚarrayÚ
|
date_rangec ó& t g d¢g d¢d« S )z$
|
base dataframe for testing
|
©é é é )é é é )ÚaÚb)r © ó ú]H:\Change_password\venv_build\Lib\site-packages\pandas/tests/series/methods/test_case_when.pyÚdfr s ô
|
9ª9Ñ5Ó 6Ð6r c ó d}|dz }t j t |¬« 5 | d j d¬« ddd« y# 1 sw Y yxY w)z5
|
Raise ValueError if caselist is not a list.
|
z(The caselist argument should be a list; z instead got.+©Úmatchr r ©ÚcaselistN)ÚpytestÚraisesÚ TypeErrorÚ case_when©r Úmsgs r Ú%test_case_when_caselist_is_not_a_listr sK ð 5CØ?ÑCÜ y¨Ô ,ñ 'Ø
|
3×Ñ 2ÐÔ&÷'÷ 'ñ 'ús £AÁA c ó d}|dz }t j t |¬« 5 | d j g « ddd« y# 1 sw Y yxY w)z6
|
Raise ValueError if no caselist is provided.
|
z(provide at least one boolean condition, z!with a corresponding replacement.r r N)r r Ú
|
ValueErrorr r s r Útest_case_when_no_caselistr# sI ð 5CØÐ .Ñ.CÜ z¨Ô -ñ Ø
|
3×Ñ"Ô ÷ ÷ ñ ús £AÁA
|
c óø d}|dz }t j t |¬« 5 | d j | d j d« d| j
|
j d« fg« ddd« y# 1 sw Y yxY w)z4
|
Raise ValueError if no of caselist is odd.
|
z Argument 0 must have length 2; z2a condition and replacement; instead got length 3.r r r N)r r r" r Úeqr Úgtr s r Útest_case_when_odd_caselistr' ) sk ð ,CØÐ ?Ñ?Cä z¨Ô -ñ <Ø
|
3×ÑBsGJJ qM¨1¨b¯d©d¯g©g°a«jÐ9Ð:Ô;÷<÷ <ñ <ús £AA0Á0A9c ó¾ d}t j t |¬« 5 | d j | d j d« ddgfg« ddd« y# 1 sw Y yxY w)z-
|
Raise Error from within Series.mask
|
z,Failed to apply condition0 and replacement0.r r r r
|
N)r r r" r r% r s r Ú$test_case_when_raise_error_from_maskr) 4 sV ð 9CÜ z¨Ô -ñ 5Ø
|
3×ÑBsGJJ qM¨A¨q¨6Ð2Ð3Ô4÷5÷ 5ñ 5ús ,AÁA c óJ t t j t j t j g« j | j j d« dfg« }t dt j t j g« }t j ||« y)z,
|
Test output on a single condition.
|
r N©r ÚnpÚnanr r r% ÚtmÚassert_series_equal©r ÚresultÚexpecteds r Ú test_case_when_single_conditionr3 = sh ô RVVRVV¤R§V¡VÐ,Ó -× 7Ñ 7¸"¿$¹$¿'¹'À!»*Àa¸ Ð8IÓ JFÜq"&&¤"§&¡&Ð)Ó*HÜ×Ñ6 8Õ,r c óJ t t j t j t j g« j | j j d« dft g d¢« dfg« }t ddt j g« }t j ||« y)zB
|
Test output when booleans are derived from a computation
|
r )FTFr
|
Nr+ r0 s r Ú"test_case_when_multiple_conditionsr5 F su ô RVVRVV¤R§V¡VÐ,Ó -× 7Ñ 7Ø
|
$$''!*a 6Ò"6Ó7¸Ð;Ð<óFô q!RVVnÓ%HÜ×Ñ6 8Õ,r c óX t t j t j t j g« j g d¢df| d j d« | d j d« z g d¢fg« }t ddt j g« }t j ||« y) z0
|
Test output when replacement is a list
|
©TFFr r r r r r
|
N)r r, r- r r& r% r. r/ r0 s r Ú3test_case_when_multiple_conditions_replacement_listr8 Q s ô RVVRVV¤R§V¡VÐ,Ó -× 7Ñ 7Ú
|
Ð " R¨¡W§Z¡Z°£]°R¸±W·Z±ZÀ³]Ñ%BÂIÐ$NÐOóFô q!RVVnÓ%HÜ×Ñ6 8Õ,r c ór t t j t j t j g« j g d¢df| d j d« | d j d« z t g d¢d¬« fg« }t dd t j gd
|
¬« }t j ||« y ) z=
|
Test output when replacement has an extension dtype
|
r7 r r r r r ÚInt64)Údtyper
|
ÚFloat64N) r r, r- r r& r% Úpd_arrayr. r/ r0 s r Ú>test_case_when_multiple_conditions_replacement_extension_dtyper> \ s ô RVVRVV¤R§V¡VÐ,Ó -× 7Ñ 7â ! 1Ð %Ø WZZ]R WZZ¨]Ñ *¬H²YÀgÔ,NÐ Oð
|
óFô q!RVVn¨IÔ6HÜ×Ñ6 8Õ,r c ó t t j t j t j g« j t j g d¢« df| d j d« | d j d« z t g d¢« fg« }t ddt j g« }t j ||« y) z2
|
Test output when replacement is a Series
|
r7 r r r r r r
|
N) r r, r- r r r& r% r. r/ r0 s r Ú5test_case_when_multiple_conditions_replacement_seriesr@ j s ô RVVRVV¤R§V¡VÐ,Ó -× 7Ñ 7ä XXÒ*Ó +¨QÐ /Ø WZZ]R WZZ¨]Ñ *¬F²9Ó,=Ð >ð
|
óFô q!RVVnÓ%HÜ×Ñ6 8Õ,r c ó4 t j j d¬« } t dd¬« }t | j d¬« |g d¢¬ « }t d
|
|j d ¬ « j |j j d « |j fg« }|j j |j j d « |j « j |j j d « d
|
« }t j ||« y)z0
|
Test output if index is not RangeIndex
|
é{ )Úseedz1/1/2000é )Úperiods)rD r )Úsize)ÚAÚBÚCÚD)ÚindexÚcolumnsr rG )rK Únamer N)r, ÚrandomÚ default_rngr r Ústandard_normalr rK r rG r& rH ÚmaskÚwherer. r/ )ÚrngÚdatesr r1 r2 s r Ú test_case_when_non_range_indexrU x sË ô ))×
|
Ñ
|
SÐ
|
Ó
|
)CÜ z¨1Ô -EÜ Ø ×Ñ ÐÓ(°Ò?Sô
|
Bô ARXX¨CÔ 0× :Ñ :¸R¿T¹T¿W¹WÀQ»ZÈÏÉÐ<NÐ;OÓ PFØttyy R§T¡TÓ*×0Ñ0°·±·±¸³¸QÓ?HÜ×Ñ6 8Õ,r c ó t j ddd« } t | « }|j d d fd d fg¬« }t j | | d k | d k\ gd
|
d g« }t j |t |« « y ) z#
|
Test output on a callable
|
g Àg @r c ó | dk S ©Nr r ©r s r ú<lambda>z)test_case_when_callable.<locals>.<lambda> s
|
Q r c ó | S ©Nr rY s r rZ z)test_case_when_callable.<locals>.<lambda> s ¨B¨3 r c ó | dk\ S rX r rY s r rZ z)test_case_when_callable.<locals>.<lambda> s
|
a r c ó | S r\ r rY s r rZ z)test_case_when_callable.<locals>.<lambda> s ¨B r r r c ó | S r\ r ©Úxs r rZ z)test_case_when_callable.<locals>.<lambda> s ¸A¸2 r c ó | S r\ r r` s r rZ z)test_case_when_callable.<locals>.<lambda> s È r N)r, Úlinspacer r Ú piecewiser. r/ )ra Úserr1 r2 s r Útest_case_when_callablerf s ô
|
D#qÓ!AÜ
|
)CØ ]]á ¡Ð /Ù ¡ Ð /ð
|
ð ó Fô ||A A¡ q¨A¡v± ¹{Ð0KÓLHÜ×Ñ6¤6¨(Ó#3Õ4r )Únumpyr, r Úpandasr r r r= r Úpandas._testingÚ_testingr. Úfixturer r r# r' r) r3 r5 r8 r> r@ rU rf r r r ú<module>rl si ðÛ Û ÷ó õ ð ñ7ó ð7ò'ò ò<ò5ò-ò-ò-ò -ò -ò -ó 5r
|
