U
|
ý°d ã @ sn d dl m Z d dlmZ d dlZd dlZd dlmZ d dlZ d dlm
|
Z
|
m Z m Z m Z mZmZmZmZ d dlmZ d dlmZ ejddd d
|
gd d d Zdd ZG dd dZdd Zdd Z dd Z dd Z dd Z d d Z d d Z!ej" #d d!d"g¡d#d$ Z$d%d& Z%eed'd(d) Z&d*d+ Z'd,d- Z(d.d/ Z)d0d1 Z*ej" #d2d3d!d4de d5d6d7g¡d8d9 Z+d:d; Z,dS )<é )Údatetime)ÚgivenN)Ú is_scalar)Ú DataFrameÚ DatetimeIndexÚIndexÚSeriesÚ StringDtypeÚ TimestampÚ
|
date_rangeÚisna)ÚOPTIONAL_ONE_OF_ALLÚdefaultÚ float_stringÚ mixed_floatÚ mixed_int)Úparamsc C sT | j dkr&ttj dd¡dddgdS | j dkr4|S | j d krB|S | j d
|
krP|S d S ) Nr é é ÚAÚBÚC©Úcolumnsr r r )Úparamr ÚnpÚrandomÚrandn)ÚrequestÚfloat_string_frameZmixed_float_frameZmixed_int_frame© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/indexing/test_where.pyÚ where_frame s
|
|
|
|
r" c s& dd t t fdd| ¡ D S )Nc S s t | jjtjtjfo | jdkS )NZuint8)Ú
|
issubclassÚdtypeÚtyper ÚintegerZfloating)Úsr r r! Úis_ok& s ÿz_safe_add.<locals>.is_okc 3 s. | ]&\}} |r ||d fn||fV qdS )é Nr )Ú.0Úcr' ©r( r r! Ú <genexpr>+ s z _safe_add.<locals>.<genexpr>)r ÚdictÚitems©Údfr r, r! Ú _safe_add$ s r2 c @ s8 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d e e ejg¡dd Ze j
|
ddgdgdggedgdgdggeddddgidgdgdggedgejgedggg¡dd Ze j
|
dddgddgddggeddgddgd dggddgddgddggedddgd d d gd ejedgedejged!ed!ggg¡d"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Z d.d/ Z d0d1 Z d2d3 Z d4d5 Z d6d7 Z!d8d9 Z"d:d; Z#d<d= Z$d>d? Z%d@dA Z&dBdC Z'e j
|
dDi dEdFig¡dGdH Z(dIdJ Z)dKdL Z*dMdN Z+dOdP Z,e j
|
dQdRdSdTdUg¡dVdW Z-dFS )XÚTestDataFrameIndexingWherec C sV ddd}|}||kr@d}t jt|d |dk W 5 Q R X d S |dk}||| d S )NTc S s t | }| ||¡}| |j|¡}| ¡ D ]<\}}tt || | | || ¡|jd}tj||dd q*t ||¡ |r|j
|
| j
|
k ¡ st d S )N©ÚindexF)Z check_names) r2 ÚwhereÚvaluesr/ r r r5 ÚtmÚassert_series_equalÚassert_frame_equalÚdtypesÚallÚAssertionError) r1 ÚcondÚ check_dtypesÚother1ÚrsZrs2ÚkÚvÚexpr r r! Ú
|
_check_get0 s $ z=TestDataFrameIndexingWhere.test_where_get.<locals>._check_getú6'>' not supported between instances of 'str' and 'int'©Úmatchr )T©ÚpytestÚraisesÚ TypeError)Úselfr" r rE r1 Úmsgr> r r r! Útest_where_get/ s
|
z)TestDataFrameIndexingWhere.test_where_getc C sn t dd dD }d|jdd d
f< |j}tt d¡t d¡t d¡t d ¡gdddd gd
|
}t ||¡ d S ) Nc S s i | ]}|t d gd |dqS )r) r ©r$ )r )r* r+ r r r! Ú
|
<dictcomp>J s ÿ zCTestDataFrameIndexingWhere.test_where_upcasting.<locals>.<dictcomp>)Úfloat32Úfloat64Úint32Úint64r r) rR rS rT rU r4 )r Úilocr; r r r$ r8 r9 ©rM r1 ÚresultÚexpectedr r r! Útest_where_upcastingG s þÿü
|
ùz/TestDataFrameIndexingWhere.test_where_upcastingc C s¬ d dd}|}||kr@d}t jt|d |dk W 5 Q R X d S |dkdd
}|||t| |dk}|||t|j |dk}tdd |jD }|||tj|d
|
d S ) NTc S sò | ||¡}t|jD ]²\}}|| }| | j}|| | | j¡ d¡j} t|rZ|}
|
n4t|t j
|
rt |d d
|f |jdj}
|
n
|
|| j}
|
| ¡ r|n t | ||
|
¡} t | |j|d} t j|| dd q|rît|t j
|
sî|j| jk ¡ sîtd S )NFr4 )r5 Úname)Z check_dtype)r6 Ú enumerater r7 Úreindexr5 Úfillnar Ú
|
isinstancer Zndarrayr r< r8 r9 r; r= ) r1 r> Úotherr? rA ÚirB rX Údr+ ÚoZ
|
new_valuesrY r r r! Ú _check_alignc s
|
|
zETestDataFrameIndexingWhere.test_where_alignment.<locals>._check_alignrF rG r r) c s s | ]}t |jtj V qd S )N)r# r% r r& )r* r' r r r! r- s zBTestDataFrameIndexingWhere.test_where_alignment.<locals>.<genexpr>)r? )T) rJ rK rL r2 r7 r< r; r Únan)rM r" r rd r1 rN r> r? r r r! Útest_where_alignmenta s
|
z/TestDataFrameIndexingWhere.test_where_alignmentc C sþ t tj dd¡dddgd}|dk}|d jdd
d d
f }d
|
}tjt|d | ||¡ W 5 Q R X |j d d
d d
f j}t
|
|}d }tjt|d | ||¡ W 5 Q R X tjt|d | d ¡ W 5 Q R X tjt|d | d¡ W 5 Q R X d S )Nr r r r r r r r) é z4other must be the same shape as self when an ndarrayrG ú,Array conditional must be same shape as selfT) r r r r r7 rJ rK Ú
|
ValueErrorr6 rV r2 Úmask)rM r1 r> Zerr1rN Zerr2r@ r r r! Útest_where_invalid s z-TestDataFrameIndexingWhere.test_where_invalidc C s ddd}|}||kr@d}t jt|d |dk W 5 Q R X d S |dk}||| |dk}||| |dkdd
}||| d S ) NTc S s | ¡ }| | ¡ d¡}| | ¡}|j|tjdd}|d ksBtt ||¡ |r| j
|
¡ D ]>\}}t |j tjr|| ¡ st d¡}|| j|ks\tq\d S )NT©ÚinplacerS )ÚcopyZ reindex_liker^ rj r6 r re r= r8 r: r; r/ r# r% r& r< r$ ) r1 r> r? ZdfiZecondrY Ú return_valuerB rC r r r! Ú
|
_check_set« s
|
z=TestDataFrameIndexingWhere.test_where_set.<locals>._check_setrF rG r r) )TrI )rM r" r rp r1 rN r> r r r! Útest_where_set¨ s
|
|
|
z)TestDataFrameIndexingWhere.test_where_setc C sR t tdtddd}| |d dk¡}||d dk |j¡}t ||¡ d S )Nr é é ©ÚaÚbru r) )r Úranger6 r] r5 r8 r: rW r r r! Útest_where_series_slicingÌ s z4TestDataFrameIndexingWhere.test_where_series_slicingÚklassc C s t ddddgi}dgdgdgg}t dtjddgi}| ||¡}t ||¡ d|d< dtjdg|d< ddgddgddgg}| ||¡}t ||¡ d S )Nru r) rg r FTrv )r r re r6 r8 r: )rM ry r1 r> rY rX r r r! Útest_where_array_likeÔ s z0TestDataFrameIndexingWhere.test_where_array_liker> r) r rg r rs ru ÚTrueÚFalsez
|
2017-01-01z
|
2017-01-02c C s> t ddddgi}d}tjt|d | |¡ W 5 Q R X d S )Nru r) rg r ú(Boolean array expected for the conditionrG ©r rJ rK ri r6 ©rM r> r1 rN r r r! Ú test_where_invalid_input_singleå s z:TestDataFrameIndexingWhere.test_where_invalid_input_singlerr é é rt z
|
2017-01-03c C sF t dddgdddgd}d}tjt|d | |¡ W 5 Q R X d S )Nr) rg r rt r} rG r~ r r r r! Ú!test_where_invalid_input_multiple÷ s z<TestDataFrameIndexingWhere.test_where_invalid_input_multiplec C s t dddgdddgg}t dddgdddgg}| |¡}t d tjdgtjtjdgg}t ||¡ d
|
d d g|_| |¡}t tj|j|jd }t ||¡ d S )Nr) rg r rr r é TFç ð?ru rv r+ ©r5 r )r r6 r re r8 r: r r5 )rM r1 r> rX rY r r r! Ú test_where_dataframe_col_match s
|
|
z9TestDataFrameIndexingWhere.test_where_dataframe_col_matchc C sð d}t dddgdddgg}dg}tjt|d | |¡ W 5 Q R X t dddgtjtjtjgg}| t|¡}t ||¡ t
|
d
|
dd
|
dg¡}tjt|d | |¡ W 5 Q R X t tjtjtjgdddgg}| t|¡}t ||¡ d S ) Nrh r) rg r rr r r TrG F) r rJ rK ri r6 r re r r8 r: Úarray)rM rN r1 r> rY Úoutr r r! Útest_where_ndframe_align s z3TestDataFrameIndexingWhere.test_where_ndframe_alignc C s¤ t ddddgddddgddd}t tjtjddgddtjtjgddd}| |dktj¡}t ||¡ | ¡ }|j|dktjd d
|
}|d kstt ||¡ d S ) Nr
ç @ç @ç @rt rS rP rg Trl )r r re r6 r8 r: rn r= )rM r1 rY rX ro r r r! Útest_where_bug1 s ÿ þ z)TestDataFrameIndexingWhere.test_where_bugc C s´ t tjddddg|dtjdddd gd
|
dd }t tjtjddgddtjtjgd d
|
d}| |dktj¡}t ||¡ | ¡ }|j|dktjd d }|d ks¤tt ||¡ d S )Nr) rg r rr rP r r r r
rS rt Trl ) r r r re r6 r8 r: rn r= )rM Zany_signed_int_numpy_dtyper1 rY rX ro r r r! Útest_where_bug_mixedB s þÿ þ z/TestDataFrameIndexingWhere.test_where_bug_mixedc C sè t ddgddgddgd}t tjdgd tjgtjtjgd}| ¡ ||kB }| ¡ }||| < | ||¡}t ||¡ t ddgdd
|
gd }t tjdgdtjgd }| ¡ ||kB }| ¡ }||| < | ||¡}t ||¡ d S ) Nr) rg r rr r r )r r) rg r r r ©r r) )r r re r rn r6 r8 r: )rM ru rv Zdo_not_replacerY rX r r r! Ú test_where_bug_transpositionX s $
|
|
z7TestDataFrameIndexingWhere.test_where_bug_transpositionc C s¸ t tdddtdddtj d¡d}tddd}d }tjt|d
|
||k W 5 Q R X ||j d d
d d
f |k }|
|
¡ }tj |j d dgd f< tj |j d d
df< t ||¡ d S )NZ20130102r ©ÚperiodsZ20130104©r r r iÝ r) r zF'>' not supported between instances of 'float' and 'datetime.datetime'rG éÿÿÿÿr r r )r r r r r r rJ rK rL rV rn re Úlocr8 r: )rM r1 ZstamprN rX rY r r r! Útest_where_datetimen s
|
|
|
ýÿ z.TestDataFrameIndexingWhere.test_where_datetimec C s´ t dttdi t¡}d ||dk< t dtdddddd d
|
dtjtjg
|
i}t ||¡ t dtjd d tjd tjd g}d }t j
|
t |d |j t | d dd W 5 Q R X d S )NZseriesé
|
rs r r) rg r rr r r ZTestr z boolean setting on mixed-typerG Trl )r r rw ÚastypeÚfloatr re r8 r: rJ rK rL r6 r )rM r1 rY rN r r r! Útest_where_none
s "ÿ þÿz*TestDataFrameIndexingWhere.test_where_nonec C s< t dgd}|}|jtk ¡ s"t| |¡}t ||¡ d S )Nru r )r r; Úobjectr< r= r6 r8 r: )rM r1 r> rX r r r! Ú9test_where_empty_df_and_empty_cond_having_non_bool_dtypes s
|
|
zTTestDataFrameIndexingWhere.test_where_empty_df_and_empty_cond_having_non_bool_dtypesc C s dd }| }| | ¡ ¡}|jt |¡| ¡ dd}t ||¡ |jt |¡| ¡ ddd}|d ksjtt ||¡ | d¡}|jdd |d d
|
}|j|dk|d d d}t ||¡ |j|dk|d d d}t ||¡ | }| d ¡}| t |¡t d |j
|
|j d¡}t ||¡ d S )Nc S sR t tj dd¡} tj| jdd
df< tj| jdd
df< tj| jdd
d f< | S )
|
Nr r r r rr r r) r rg )r r r r re rV r0 r r r! Úcreate¥ s
|
z;TestDataFrameIndexingWhere.test_where_align.<locals>.creater ©ÚaxisT©rm r r c S s | | dk|¡S ©Nr )r6 )ÚxÚyr r r! Ú<lambda>· ó z=TestDataFrameIndexingWhere.test_where_align.<locals>.<lambda>©r¤ r5 Úrowsr) r ) r^ Zmeanr6 ÚpdÚnotnar8 r: r= Úapplyr r5 r )rM r r1 rY rX ro r r r! Útest_where_align¤ s*
|
ÿz+TestDataFrameIndexingWhere.test_where_alignc C s\ t ddgtjdggddgd}t ddgddggddgd}tj|| ¡ dk< t ||¡ d S ) Ny ð? ð?rg y @ ð?ru rv r y @ ð?r )r r re Úabsr8 r: )rM rY r1 r r r! Útest_where_complexÅ s z-TestDataFrameIndexingWhere.test_where_complexc C s t tj dd¡}t ddgddgg}tddg}t ddgddggdd}|j||dd}t ||¡ | ¡ }|j||dd d
|
}|d kst t ||¡ t ddgddggdd}|j||d d}t ||¡ | ¡ }|j||d d d
|
}|d ksòt t ||¡ d S ) Nrg Fr r) rS rP r5 r T©r rm r )
|
r r r r r r6 r8 r: rn r= ©rM r1 rj ÚserrY rX ro r r r! Útest_where_axisÌ s" z*TestDataFrameIndexingWhere.test_where_axisc C s> t ddgddggdd}t ddgddgg}tdtjg}t ddgtjtjggd d}|j||d
|
d }t ||¡ | ¡ }|j||d
|
d d }|d kstt ||¡ t dtjgdtjgg}|j||dd }t ||¡ t tj ddgddtj tjtjgd dd}| ¡ }|j||dd d }|d ks.tt ||¡ d S )Nr) rg r rr rU rP Fr rS r5 r Tr¯ r r )
|
r r r re r6 r8 r: rn r= r r° r r r! Útest_where_axis_with_upcastä s, þÿz6TestDataFrameIndexingWhere.test_where_axis_with_upcastc
|
C s
|
t jttj dd¡ttjjddddddgdd d
|
}td |j|jd }t d |jd }t d|jd }|j
|
||dd}td|j|jd }|d d¡|d< |d d¡|d< t ||¡ | ¡ }|j
|
||ddd}|d ksÞtt ||¡ |j
|
||dd}td|j|jd }|d d¡|d< |d d¡|d< t ||¡ | ¡ }|j
|
||ddd}|d ksdtt ||¡ | ¡ jd dd}| ¡ d¡}tj|jd d d
f< |
|
||¡}t ||¡ |j
|
||dd}t ||¡ | ¡ }|j
|
||dd}|d ksþtt ||¡ | ¡ }|j
|
||ddd}|d ks2tt ||¡ | ¡ jd d d} | ¡ }tj|jd d
d f< |
|
|| ¡}t ||¡ |j
|
|| dd}t ||¡ | ¡ }|j
|
|| dd}|d ksÆtt ||¡ | ¡ }|j
|
|| ddd}|d ksútt ||¡ d S )Nr rg r )r rg )ÚsizerU rP Tr) )Z ignore_indexr F)r r5 r4 r r r
r r¯ r5 r r rl r¡ )r© Úconcatr r r r Úrandintr r5 r r6 r r8 r: rn r= Zdropre r )
|
rM r1 rj Ús1Ús2rX rY ro Úd1Zd2r r r! Ú test_where_axis_multiple_dtypes st þú z:TestDataFrameIndexingWhere.test_where_axis_multiple_dtypesc C sH t dddgdddgddd gg}| d
|
d d d ¡}t dddgdddgddd gg}t ||¡ t || |dk|d ¡¡ | d d dd ¡}t dddgdddgdddgg}t ||¡ t || |d dkd¡¡ |d dd dd ¡}t dddgdddgd ddgg}t ||¡ t ||d |d dk|d d ¡¡ d S )Nr) rg r rr r r rs r r c S s | dkS )Nrr r ©r£ r r r! r¥ M r¦ z@TestDataFrameIndexingWhere.test_where_callable.<locals>.<lambda>c S s | d S ©Nr) r r» r r r! r¥ M r¦ c S s | d dkj S )Nrg r )r7 r» r r r! r¥ S r¦ c S s dS )Néc r r» r r r! r¥ S r¦ r½ r c S s | dkS )Nr r r» r r r! r¥ Y r¦ c S s | d S )Nr r r» r r r! r¥ Y r¦ é é é é é é r é )r r6 r8 r: )rM r1 rX rD r r r! Útest_where_callableJ s z.TestDataFrameIndexingWhere.test_where_callablec C s¬ t tdddg|ddgd}t tdddg|ddgd}t d d d
|
gdgd}t tdddg|ddgd}|tkr|d }|d }|d }|d }| ||¡}t ||¡ d S ) NZ20150101Z20150102Z20150103)ÚtzÚdater Z20150104Z20150105TF)r r r r6 r8 Ú assert_equal)rM Ztz_naive_fixtureÚframe_or_seriesZobj1Úobj2rj rD rX r r r! Útest_where_tz_values^ s( þþþ z/TestDataFrameIndexingWhere.test_where_tz_valuesc C sz t t d¡ dd¡tdd}t dddgdddgg¡}| |¡}t dtjtjgtjtjd ggtdd}t ||¡ d S )
|
Nr rg r ÚABCr TFr r )
|
r r ÚarangeÚreshapeÚlistr r6 re r8 r: )rM r1 rj rX rY r r r! Útest_df_where_change_dtypeu s
|
ÿz5TestDataFrameIndexingWhere.test_df_where_change_dtypeÚkwargsr` Nc
|
C s t jdt jd dd¡}t|tdd}t dddgdddgg¡}|j d ¡|_|j d ¡|_ |j
|
d ¡|_
|
|j |f|}t j d
|
t jgd
|
dgd }t j t jt jgd d gd }t j t jdgddgd }t|||d} t || ¡ |jj |d d
d
|
f f|}t|dd} t || ¡ d S )Nr rP rg r rÌ r TFÚcategoryr )Ú
|
categoriesr) rr r r r )r[ )r rÍ rU rÎ r rÏ r r r r r r6 r© Z Categoricalre r8 r: r r9 )
|
rM rÑ Údatar1 rj rX r r r rY r r r! Útest_df_where_with_category s z6TestDataFrameIndexingWhere.test_df_where_with_categoryc C sv t ddgddggddgd}|d d¡|d< | |d dk¡}| ¡ ddi¡}tj|jdd d
f< t ||¡ d S )Nr r) ru rv ©rÔ r rÒ r ) r r r6 rn r re r r8 rÈ rW r r r! Ú test_where_categorical_filtering s z;TestDataFrameIndexingWhere.test_where_categorical_filteringc C sú t dddgdddgd}t dtjd g¡}t|}tj|jtd
|
}d |dd d
f< |j ||d d }t dtjdgdtjdgd
|
t ¡}t ||¡ t|d d
ddgd}t dddgdtjdgd}|d
|
t ¡|d< |j ||dd }t ||¡ d S )Nr) rg r rr r r ©r r rs r rP Fr r r r r4 )r r© r ÚNAr r ZonesÚshapeÚboolr6 r r r8 r: )rM r1 Úarrr± rj rX rY Zser2r r r! Útest_where_ea_other§ s $ z.TestDataFrameIndexingWhere.test_where_ea_otherc C sR t t dd¡g}| | ¡ ¡}t ||¡ |d }| | ¡ ¡}t ||¡ d S r¢ )r r© ÚIntervalr6 rª r8 r: r9 )rM r1 Úresr± r r r! Útest_where_interval_noopº s z3TestDataFrameIndexingWhere.test_where_interval_noopc C sr |t dd¡gd }|ddg}| | ¡ |¡}t || tj¡¡ |j | ¡ |dd t || t
|
¡¡ d S )Nr rg r
r Trl ) r© rÞ r6 rª r8 rÈ r r rU rj r )rM rÉ Úobjr` rß r r r! Ú#test_where_interval_fullop_downcastÄ s z>TestDataFrameIndexingWhere.test_where_interval_fullop_downcastr$ ztimedelta64[ns]zdatetime64[ns]zdatetime64[ns, Asia/Tokyo]z Period[D]c C sø t t d¡d tjd |¡}| ¡ }t dddg¡}| | d¡}t ||¡ |
|
dd¡}| | d¡}t ||¡ | |d¡}t ||¡ | |d¡} t | |¡ | |d¡}
|
t d|j|jd } t |
|
| ¡ |j | dd
|
d t || t¡¡ d S ) Nr i Ê;rP FZfoor r) rr r Trl )r r rÍ rU ÚviewZto_framer r6 r8 r9 rÎ r: rj r r5 r r r ) rM r$ r± r1 rj rß Zmask2Zres2Zres3Zres4Zres5rY r r r! Ú test_where_datetimelike_noopÒ s" z7TestDataFrameIndexingWhere.test_where_datetimelike_noop).Ú__name__Ú
|
__module__Ú __qualname__rO rZ rf rk rq rx rJ ÚmarkÚ parametrizerÏ Útupler r rz r r r
|
r© ÚNaTr r r r r r r r r r r¬ r® r² r³ rº rÅ rË rÐ rÕ r× rÝ rà râ rä r r r r! r3 . s| 2$
|
ûþ
|
|
ýûþ
|
! H
|
|
üþ r3 c C s t d¡ t j¡ dd¡} t| }t j| jtd}d|d d
df< | |d¡}t| d d
df t j
|
dgd t j dd}t ||¡ d S ) Nr r rg rP Tr i r )r rÍ r Úint16rÎ r ÚzerosrÚ rÛ r6 r rT r8 r: )rÜ r1 rj rß rY r r r! Ú%test_where_int_downcasting_deprecatedû s ,rî c C s~ | ddddg}| ¡ }| tkr(|d n|}| |dk ¡}|d9 }t ||¡ | |dkddddg¡}|d9 }t ||¡ d S )Nr) rg r rr r r )rn r r6 r8 rÈ )rÉ rX rY ÚcolZ where_resr r r! Útest_where_copies_with_noop s rð c C sÀ | ddddgddddgt d }| ddgddgt d }td
|
d d d
|
g}| ||¡}| tjddtjgddddgt d }t ||¡ | | |¡}t ||¡ |j| |d d t ||¡ d S ) Nru rv r+ rb Zid1Zid2Zid3Zid4)r5 r$ FTrl )r r r6 r© rÙ r8 rÈ rj )rÉ rá Z filtered_objZ
|
filter_serrX rY r r r! Útest_where_string_dtype s,
|
|
ÿ ÿ
|
ý rñ c C sv t dgd dgd ddddgd} | | dk¡}t tjtjgd tddgd tjdtjdgd}t ||¡ d S )NTrr F)ZAAAZBBBZCCCrP )r r6 r r re r r8 r: )Zdf_maskrX rY r r r! Útest_where_bool_comparison3 s ÿýÿrò c C sF t tdtjtdgddtjgd} | | ¡ d ¡}t || ¡ d S )NZ20130101Z20130103r) rg rØ )
|
r r
|
r© rë r re r6 Únotnullr8 r: )rY rX r r r! Útest_where_none_nan_coerceC s
|
þÿrô c C sÀ t dtjggtddgd} | j\}}t ddgg||d}| t¡ |¡}| d¡ |¡}| j |j¡j}| |¡}t
|
| d¡| d¡¡ t
|
| d¡| d¡¡ t
|
| d¡| d¡¡ d S )Nr r rÖ T)rÔ r r5 Zf8) r r re r Zaxesr r r6 ÚTr8 r: )rX r5 r rj ru rv r+ rb r r r! Ú&test_where_duplicate_axes_mixed_dtypesO s
|
|
rö c C sB t ddgdtjgd} | ¡ }| t | ¡d ¡}t ||¡ d S )Nr
r r rt ) r r re rn r6 r© ró r8 r: )r1 rY rX r r r! Útest_where_columns_casting^ s r÷ Úas_catTFc
|
C sH t jdddd}|r | d¡}| |}t j d¡}tjdddg|jd j}|rTd
|
}nd }|rÔt j
|
t |d | ||¡ W 5 Q R X t j
|
t |d | ||¡ W 5 Q R X t j
|
t |d |j ||dd W 5 Q R X np| t¡ ||¡}| ||¡} t | |¡ | t¡ ||¡}| ||¡} t | |¡ |j ||dd t ||¡ d S )Nú
|
2016-01-01r ÚD)r ÚfreqrÒ úm8[ns]TF©ZndminzUCannot setitem on a Categorical with a new category \(NaT\), set the categories firstzvalue should be a 'Period'rG rl )r© Z period_ranger rë Zto_numpyr r Úndimrõ rJ rK rL r6 rj r r8 rÈ )
|
rÉ rø r Úidxrá Ztdnatrj rN rY rX r r r! Ú test_where_period_invalid_nah s2
|
ÿ r c
|
C s t jdddg|d}| |}tjdddg|jdj}d}tjt jg D ]P}tj t
|
|d | ||¡ W 5 Q R X tj t
|
|d | ||¡ W 5 Q R X qFd S )
|
Nr) rg r rP TFrý z1Invalid value '.*' for dtype (U?Int|Float)\d{1,2}rG ) r© r r rþ rõ r8 ZNP_NAT_OBJECTSrë rJ rK rL r6 rj )rÉ Zany_numeric_ea_dtyperÜ rá rj rN Únullr r r! Ú test_where_nullable_invalid_na s r )rÔ c C sH t d| i}| t |¡d ¡ ¡ }|jt |¡d dd t ||¡ d S )Nru Trl )r r6 r© ró rn r8 rÈ )rÔ r1 Zdf_copyr r r! Útest_where_inplace_casting¨ s r c C sX t dddg} t dddg¡}tjdd}| ||¡}t |||gdd}t ||¡ d S )Nr) rg r F)Údaysrü rP )r r r r© Z Timedeltar6 r8 r9 )r± rj Útdrß rY r r r! Útest_where_downcast_to_td64± s r c C s~ | ||¡}t ||¡ | | |¡}t ||¡ | ¡ } | j| |dd | ¡ sn| ¡ }|d t¡|d< t | |¡ d S )NTrl r )r6 r8 r: rj rn r< r r )r1 rj r` rY rß r r r! Ú_check_where_equivalences½ s r c C sú t ddd} | j dd¡}||d }t|ddgd }t | ¡ ¡ ¡ }d
|
|d d
d f< t|d d
d f |d d
d f d }t|||| d
|
|d< ttj |d |d |d gt
|
d|d d
d f d }t|||| d
|
|d d
< |}t|||| d S )Nrù r r r rg )r r r r r Tr) r rØ )r) r )rg r rP ) r Ú_datarÎ r r Zasarrayr rn r r r )ZdtiZdtar` r1 rj rY r r r! Útest_where_dt64_2dÒ s$ & þÿ r c C sj t tdtjdgdddddgd} | dd ¡}t ttjtjdgddtjddgd}t ||¡ d S ) Nr) rg ZInt64rP r rt c S s | j dd ddS )Nc S s | dkS r¼ r r§ r r r! r¥ ó r¦ zMtest_where_producing_ea_cond_for_np_dtype.<locals>.<lambda>.<locals>.<lambda>r) r )r« r» r r r! r¥ ó r¦ z;test_where_producing_ea_cond_for_np_dtype.<locals>.<lambda>) r r r© rÙ r6 r re r8 r: )r1 rX rY r r r! Ú)test_where_producing_ea_cond_for_np_dtypeð s $"ÿr
|
Ú replacementgü©ñÒMbP?Zsnakeiæ r rr c C sT t dddgtjdd gg}| t |¡| ¡}t dddg| d| gg}t ||¡ d S )Nr
g(,*0EZnineg¹?)r r re r6 r© ró r8 r: )r r1 rX rY r r r! Útest_where_int_overflowú s r c C sd t ddgddgd} t ddgddgd}| j|dd t dtjgtjdgd}t | |¡ d S ) Nr) rg r£ r¤ rt TFrl )r r6 r re r8 r: )r1 r> rY r r r! Útest_where_inplace_no_other s
|
r )-r Z
|
hypothesisr Únumpyr rJ Zpandas.core.dtypes.commonr Zpandasr© r r r r r r
|
r r Zpandas._testingZ_testingr8 Zpandas._testing._hypothesisr Zfixturer" r2 r3 rî rð rñ rò rô rö r÷ rè ré r r r r r r r
|
r r r r r r! Ú<module> sN (
|
|
|
R
|
|
-
|
|
ÿ
|
|
