U
|
®ý°d3 ã @ s d dl Zd dlZd dlmZ d dlZd dlmZmZm Z m
|
Z
|
d dl m Z dd Zdd Zej dejejfejejfejejfejejfejejfejejfg¡d d
|
Zd d Zd d Zdd Zej de e ej eg¡dd Z ej ddd dgedddgdddgedej ed gg¡d d Z!d d Z"d!d" Z#ej d#e$dd$¡ej d%d&d'd'd'd'gd&d'gd'gg¡ej d(d)ej%e &e'¡j(e &e'¡j)g¡ej d*d+d, d-d, d.d, g¡d/d0 Z*d1d2 Z+d3d4 Z,d5d6 Z-ej d7d8d9g¡d:d; Z.d<d= Z/d>d? Z0d@dA Z1dBdC Z2dDdE Z3dS )Fé N)Ú
|
is_integer)ÚSeriesÚ TimestampÚ
|
date_rangeÚisnac C s\ t t d¡| d}|dk }tdd||< t ttddttdd | d}t ||¡ d S ©Né
|
©Údtypeé é é ©r ÚnpÚarangeÚrangeÚlistÚtmÚassert_series_equal)Zany_signed_int_numpy_dtypeÚsÚmaskÚexpected© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/indexing/test_where.pyÚtest_where_unsafe_int s þr c C s` t t d¡| d}|dk }tdd||< ttddttdd }t || d}t ||¡ d S r r )Zfloat_numpy_dtyper r Údatar r r r Útest_where_unsafe_float s r zdtype,expected_dtypec C sZ t t d¡| d}dddddg}|dk }t |ttdd |d}|||< t ||¡ d S ) Nr r ç @ç @ç @ç @g @r )r r r r r r r )r
|
Zexpected_dtyper Úvaluesr r r r r Útest_where_unsafe_upcast( s r" c C s: t t d¡dd} ddddg}| dk}t ttd | d
|
d}|| |< t | |¡ t t d¡dd} | dk }td d | |< t ttd d ttdd dd}t | |¡ | j|jksºtt t d¡dd} | dk}d gd | |< t d dd dddgd gd dd}t | |¡ t t d¡} | dk}d}t j
|
t |d dddd dg| |< W 5 Q R X t j
|
t |d d gd | |< W 5 Q R X t dd ddg} | | d ktj ¡}t tj tj ddg}t ||¡ t td t¡} d | d< | d }t|sðtt td t¡} d | | dk< | t| }t tj dgd}t ||¡ d S )Nr Úint64r r r r r r é Úfloat64r r r é é é úKcannot set using a list-like indexer with a different length than the value©Úmatché é ©Úindex)r r r r r r r r
|
ÚAssertionErrorÚpytestÚraisesÚ
|
ValueErrorÚwhereÚnanÚastypeÚfloatr )r r! r r ÚmsgÚresultr r r Útest_where_unsafe= sJ $ " r: c C s t tj d¡} | dk}| |¡ ¡ }| | }t ||¡ | || ¡}t || ¡ ¡ | |¡}| j |j ksrt
|
|| k s~t
|
t dddddg| j d}| ¡ }|| |j d d
¡ |j ¡}| |d d
¡}t ||¡ | ¡ }|d |j d< | |d d
| ¡}t ||¡ d S )Nr r TFr. r( )r r ÚrandomÚrandnr4 Údropnar r ÚabsÚshaper0 r/ ZreindexZiloc)r ÚcondÚrsZrs2Ús2r r r r Ú
|
test_wheret s&
|
|
rC c C s t tj d¡} | dk}d}tjt|d | d¡ W 5 Q R X tjt|d | |d d
j| ¡ W 5 Q R X t ddg} ddg| dd g< t ddg}t
|
| |¡ d
|
}tjt|d dddg| dd g< W 5 Q R X tjt|d g | dd g< W 5 Q R X d S ) Nr r ú,Array conditional must be same shape as selfr* r' r( r TFr) ) r r r; r< r1 r2 r3 r4 r! r r )r r@ r8 r r r r Útest_where_error s " rE Úklassc C sF t dddg}dddg}t tjddg}| | |¡}t ||¡ d S )Nr' r r( FT)r r r5 r4 r r )rF r r@ r r9 r r r Útest_where_array_likeª s
|
|
rG r@ r' r r r ÚTrueÚFalsez
|
2017-01-01z
|
2017-01-02c C sd t dddg}d}tjt|d | | ¡ W 5 Q R X d}tjt|d | dg¡ W 5 Q R X d S )Nr' r r( z(Boolean array expected for the conditionr* rD T)r r1 r2 r3 r4 )r@ r r8 r r r Útest_where_invalid_inputµ s rJ c C sÎ d} t dddg}dg}tjt| d | |¡ W 5 Q R X t dtjtjg}| t |¡}t ||¡ t ddddg¡}tjt| d | |¡ W 5 Q R X t tjdtjg}| t |¡}t ||¡ d S )NrD r' r r( Tr* F)
|
r r1 r2 r3 r4 r r5 r r Úarray)r8 r r@ r Úoutr r r Útest_where_ndframe_alignË s rM c C s2 dd } t td}tjt| dd ttd|dd
< W 5 Q R X ttd|dd
< t dd d
|
g}t | t j
|
¡|¡ t td }tjt| dd ttd|dd d
|
< W 5 Q R X t td }ttd
|
|dd d
|
< t dd d dddg}t ||¡ t td }tjt| dd ttd|d d
< W 5 Q R X ttd
|
|dd
< t dd ddd dg}t ||¡ t td}tjt| dd ttd|dd d
|
g< W 5 Q R X t td}tjt| dd ttd
|
|dd d
|
g< W 5 Q R X t td}ttd|d< t ttdd dg}t ||¡ d S )Nc S s d| dS )Nzcannot set using a z/ indexer with a different length than the valuer ©Úxr r r Ú<lambda>ç ó z,test_where_setitem_invalid.<locals>.<lambda>ÚabcÚslicer* é r r( r' r Zabcdefr& ÚbÚdÚeÚféÿÿÿÿéýÿÿÿÚaÚcz list-liker ) r r r1 r2 r3 r r r r6 r r# )r8 r r r r r Útest_where_setitem_invalidâ s> ÿ r] Úsizer$ r TFÚitemg @Úboxc C s t | g¡S ©N)r rK rN r r r rP % rQ rP c C s | gS ra r rN r r r rP % rQ c C s | fS ra r rN r r r rP % rQ c s t || ¡}t j| td t fddt|D }t }||< t ||¡ t }| | |¡}t ||¡ t }| ||¡}t ||¡ d S )Nr c s g | ]\}}|rn | qS r r )Ú.0ÚiZuse_item©r r_ r r Ú
|
<listcomp>0 s z"test_broadcast.<locals>.<listcomp>)
|
r Úresizer r7 r Ú enumerater r r4 r )r^ r r_ r` Z selectionr r r9 r rd r Útest_broadcast s ÿ rh c C s t tj d¡} | dk}| ¡ }|j|dd t | ¡ | | ¡ t || |¡¡ | ¡ }|j|| dd t || || ¡¡ d S )Nr r T)Zinplace) r r r; r< Úcopyr4 r r r= )r r@ rA r r r Útest_where_inplaceA s rj c C sô t ttd} t ttd}t | |g¡}| |dk ¡}t ddtjddtjgddddddgd}t ||¡ d||dk < t ddddddgddddddgd}t ||¡ ||dk d7 < t ddddddgddddddgd}t ||¡ d S ) Nr( r r r' r. r r é )
|
r r r ÚpdÚconcatr4 r r5 r r )Ús1rB Úcombr9 r r r r Útest_where_dupsP s ( $ $rp c C sN t dddg} | | dkd¡}t|d r.tt|d s>tt|d sNtt|d ts`t|jdksnt| | dkdddg¡}t|d rtt|d s¤tt|d s´tt|d tsÆt|jdksÔt| | dkt dddg¡¡}t|d rtt|d stt|d s&tt|d ts:t|jdksJtd S ) Nr' r r( ÚXr ÚobjectÚYÚZ) r r4 r r0 Ú
|
isinstanceÚstrr
|
r rK )r Úwr r r Ú test_where_numeric_with_stringe s& rx r
|
ztimedelta64[ns]zdatetime64[ns]c C sÄ t ddg| d}t ddg}t ddg¡}| |ddg¡}t ||¡ | |d¡}t ||¡ | |d¡}t ||¡ | |ddg¡}t ||¡ | |dtjg¡}t dd gdd}t ||¡ d S )Nr' r r r Fg $@rr )r r rK r4 r r r5 )r
|
Úserr r rA r r r Ú test_where_datetimelike_coerce s rz c C sV ddg} t dd | D dd}| t ddg¡}t tj|d gdd}t ||¡ d S )
|
Nz2016-12-31 12:00:04+00:00z 2016-12-31 12:00:04.010000+00:00c S s g | ] }t |qS r )r )rb Útr r r re s z)test_where_datetimetz.<locals>.<listcomp>zdatetime64[ns, UTC]r FTr' )r r4 rl ÚNaTr r )Z
|
timestampsry rA r r r r Útest_where_datetimetz s
|
r} c C sH t tj ddg¡} | | dkd¡}t tj ddg¡}t ||¡ d S )Nr' r r )r rl ZarraysZ SparseArrayr4 r r )ry r9 r r r r Útest_where_sparse s r~ c C s& t g td} | g ¡}t || ¡ d S )Nr )r r7 r4 r r )ry r9 r r r Ú=test_where_empty_series_and_empty_cond_having_non_bool_dtypes§ s
|
r c C s\ | t jddddtjgdddgddd}| dddddgdd}| |dk¡}t ||¡ d S )NÚAÚBÚC)Ú
|
categoriesÚcategoryr )rl Ú Categoricalr r5 r4 r Z assert_equal)Zframe_or_seriesÚexpZdfÚresr r r Útest_where_categorical® s þr c C sê | }t dd|d d ¡}t |d |d tjg¡}t |d tj|d g¡}t dddg¡}| ||¡}t
|
||¡ |j ||¡}t ||j ¡ t| ||¡}t |t|¡ t |¡ |d d
d f t |¡¡}t |t |¡¡ d S ) Nz
|
2001-01-01r( )ZperiodsÚtzr r' r TF)r Z
|
_with_freqrl Z DatetimeIndexr| r
r rK r4 r Zassert_index_equalÚ_dataZ_whereZassert_datetime_array_equalr r Z DataFrameZassert_frame_equal)Ztz_naive_fixturer ZdrZlvalsZrvalsr r r r r Ú#test_where_datetimelike_categorical¹ s $r )4Únumpyr r1 Zpandas.core.dtypes.commonr Zpandasrl r r r r Zpandas._testingZ_testingr r r ÚmarkZ parametrizeZint8r% Úint16Úint32r# Úfloat32r" r: rC rE r ÚtuplerK rG r| rJ rM r] r r5 Zfinfor7 ÚmaxÚminrh rj rp rx rz r} r~ r r r r r r r Ú<module> st
|
|
|
|
|
|
úþ
|
|
7
|
|
üþ
|
9 ÿ ÿ ÿ
|
|
