U
|
ý°dbB ã @ s~ d dl Zd dlZd dlmZ d dlZd dlmZmZ d dl m
|
Z d dl m Z G dd deZG dd de ZG d d
|
d
|
ZdS ) é N)ÚExtensionDtype)Ú DataFrameÚ Timestamp)ÚExtensionArrayc @ s6 e Zd ZeZddddZedd Zedd ZdS ) Ú
|
DummyDtypeN©Úreturnc C s
|
|| _ d S ©N©Z_numeric)ÚselfÚnumeric© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_select_dtypes.pyÚ__init__ s zDummyDtype.__init__c C s dS )NZDummyr ©r r r r Úname s zDummyDtype.namec C s | j S r r
|
r r r r Ú _is_numeric s zDummyDtype._is_numeric) Ú__name__Ú
|
__module__Ú __qualname__ÚintÚtyper Úpropertyr r r r r r r s
|
r c @ sL e Zd ZddddZdd Zedd Zedd d
|
Zd d Z d d Z
|
dS )Ú
|
DummyArrayNr c C s || _ || _d S r )ÚdataÚ_dtype)r r Údtyper r r r s zDummyArray.__init__c C s | j S r )r )r r r r r Ú __array__# s zDummyArray.__array__c C s | j S r )r r r r r r & s zDummyArray.dtypec C s
|
t | jS r )Úlenr r r r r Ú__len__* s zDummyArray.__len__c C s d S r r )r Úitemr r r Ú __getitem__- s zDummyArray.__getitem__c C s | S r r r r r r Úcopy0 s zDummyArray.copy) r r r r r r r r r r! r" r r r r r s
|
r c @ s: e Zd Zdd Zdd Zdd Zej de j
|
dfe j
|
d fd
|
e fg¡d d Z d d Z dd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zej d ed e jd!d"e jd#g¡ej d$dd%g¡d&d' Zd(d) Zd*d+ Zej d,e j d-d.ge j d/d0fe j d-d.gd1d/d0fe d-d.ge d0d2d/d0fe d-d.ge d3d2d/d3ff¡d4d5 Z!d6d7 Z"ej d8e#e$d9e$d:d;e$d<d=d>d?j%e&e j'e j(d?d/e&ge#e$d9e$d:d;e$d<d=d>d?j%e&e j'e j(d?d/d@ge#dAe$d<d=d>ie j(d/e j(ge#e$d9e$d:d;dBj%e&e j'dBd/e j'gg¡dCdD Z)dEdF Z*dGdH Z+dIS )JÚTestSelectDtypesc C s¶ t tdttddt dd¡ d¡tjddd d
|
d d d gt td¡tjd ddtjd dddtjd dddtj ddddtj
|
dddd }|j tj gd}|ddddg }t ||¡ |j tj gd gd }|dddg }t ||¡ |j tj d gd gd }|dddd g }t ||¡ |j d gd}|d!g }t ||¡ |j d"gd}|d!g }t ||¡ |j d#gd}|d$d%g }t ||¡ tjtd&d' |j d(gd W 5 Q R X d S ))NÚabcé é é é Úu1ç @ç @Úfloat64©r TFÚ20130101©Úperiodsú
|
US/Eastern©r0 ÚtzÚCETú2013-01ÚM©r0 Úfreqú1 day© ÚaÚbÚcÚdÚeÚfÚgÚhÚiÚjÚk©Úincluder< r= r> rE Ú timedelta©rG ÚexcludeÚcategoryr@ ÚdatetimerA Ú
|
datetime64Z
|
datetimetzrB rC ú^$©ÚmatchÚperiod©r ÚlistÚrangeÚnpÚarangeÚastypeÚpdÚ CategoricalÚ
|
date_rangeÚ period_rangeÚtimedelta_rangeÚ select_dtypesÚnumberÚtmÚassert_frame_equalÚpytestÚraisesÚNotImplementedError©r ÚdfÚriÚeir r r Ú*test_select_dtypes_include_using_list_like5 sD õÿ
|
|
z;TestSelectDtypes.test_select_dtypes_include_using_list_likec C sn t tdttddt dd¡ d¡tjddd d
|
d d d gd }|jtjgd}|ddg }t ||¡ d S )Nr$ r% r& r' r( r) r* r+ r, r- TF©r; r< r= r> r? ©rJ r; r? )
|
r rS rT rU rV rW r] r^ r_ r` )r re ÚreÚeer r r Ú*test_select_dtypes_exclude_using_list_likea s ûÿ z;TestSelectDtypes.test_select_dtypes_exclude_using_list_likec
|
C sº t tdttddtjddddtjdd d
|
dd d d gtjd ddjd}tjf}tj df}|j
|
||d}|dddg }t ||¡ d}d}|j
|
||d}|ddg }t ||¡ d S )Nr$ r% r& r' r( r) r- r* r+ r, TFÚnowr/ ©r; r< r= r> r? r@ ÚintegerrI r< r= r? )rL )ÚboolÚint64Úint32) r rS rT rU rV rX rZ ÚvaluesrM Úbool_r] r_ r` )r re rJ rG Úrr? r r r Ú2test_select_dtypes_exclude_include_using_list_likeo s& úÿ
|
|
zCTestSelectDtypes.test_select_dtypes_exclude_include_using_list_likerG r rp rq c
|
C s t tdttddtjddddtjdd d
|
dd d d gtjd ddjd}tjf}|j ||d}|dddg }t
|
||¡ d S )Nr$ r% r& r' r( rs r- r* r+ r, TFrn r/ ro rI r< r= r? ) r rS rT rU rV rX rZ rt rM r] r_ r` )r rG re rJ ÚresultÚexpectedr r r Ú&test_select_dtypes_exclude_include_int s úÿ
|
z7TestSelectDtypes.test_select_dtypes_exclude_include_intc C sJ t tdttddt dd¡ d¡tjddd d
|
d d d gt td¡tjd ddtjd dddtjd dddtj ddddtj
|
dddd }|j tj d}|ddddg }t ||¡ |j d d}|d g }t ||¡ |j d d}|d g }t ||¡ |j d d}|d g }t ||¡ tjtd!d" |j d#d W 5 Q R X d S )$Nr$ r% r& r' r( r) r* r+ r, r- TFr. r/ r1 r2 r4 r5 r6 r7 r9 r: rF r< r= r> rE rL rA rM rK r@ rN rO rQ rR rd r r r Ú(test_select_dtypes_include_using_scalars s8 õÿ
|
|
|
z9TestSelectDtypes.test_select_dtypes_include_using_scalarsc C s t tdttddt dd¡ d¡tjddd d
|
d d d gt td¡tjd ddtjd dddtjd dddtj ddddtj
|
dddd }|j tj d}|ddddd d d g }t ||¡ |j d d}|dd d!d"ddd d d d#g
|
}t ||¡ tjtd$d% |j d&d W 5 Q R X d S )'Nr$ r% r& r' r( r) r* r+ r, r- TFr. r/ r1 r2 r4 r5 r6 r7 r9 r: rj r; r? r@ rA rB rC rD rK r< r= r> rE rN rO rQ rR rd r r r Ú(test_select_dtypes_exclude_using_scalars¾ s, õÿ z9TestSelectDtypes.test_select_dtypes_exclude_using_scalarsc C s¾ t tdttddt dd¡ d¡tjddd d
|
d d d gt td¡tjd ddtjd dddtjd dddtj ddddtj
|
dddd }|j tj dd}|dddg }t ||¡ d S ) Nr$ r% r& r' r( r) r* r+ r, r- TFr. r/ r1 r2 r4 r5 r6 r7 r9 r: ÚfloatingrI r< r= rE ©r rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` rd r r r Ú0test_select_dtypes_include_exclude_using_scalarsÚ s" õÿzATestSelectDtypes.test_select_dtypes_include_exclude_using_scalarsc C sð t tdttddt dd¡ d¡tjddd d
|
d d d gt td¡tjd ddtjd dddtjd dddtj ddddtj
|
dddd }|j tj ddgd}|ddg }t ||¡ |j tj d gdd}|ddd d g }t ||¡ d S ) Nr$ r% r& r' r( r) r* r+ r, r- TFr. r/ r1 r2 r4 r5 r6 r7 r9 r: r} rH rI r< r= rK r@ rE r~ rd r r r Ú6test_select_dtypes_include_exclude_mixed_scalars_listsï s( õÿ zGTestSelectDtypes.test_select_dtypes_include_exclude_mixed_scalars_listsc
|
C sª t dddgdddgt dd¡ d¡tjd d
|
d d d dd gtjdddjd}ddddddg|_t tt ddt dd¡ d¡d}|j
|
tj gdgd}t ||¡ d S )Nr; r< r= r% é r' r( r) r* r+ r, r- TFrn r/ ro r& ©r; r< r} rI )r rU rV rW rX rZ rt ÚcolumnsrS rT r] r^ r_ r` )r re ry rx r r r Ú$test_select_dtypes_duplicate_columns s úÿ
|
ÿz5TestSelectDtypes.test_select_dtypes_duplicate_columnsc
|
C sÈ t tdttddt dd¡ d¡tjddd d
|
d d d gtjd ddjd}|j
|
¡ |d< t tdrlt |j ddgdgd}|ddg }t ||¡ |j dddgd}|dddg }t ||¡ d S )Nr$ r% r& r' r( r) r* r+ r, r- TFrn r/ ro rA Úu8Úi8ÚOrH rI r; r< ztimedelta64[ns]rF )r rS rT rU rV rW rX rZ rt r@ ZdiffÚhasattrÚAssertionErrorr] r_ r` )r re rv r? r r r Ú4test_select_dtypes_not_an_attr_but_still_valid_dtype s" úÿ
|
zETestSelectDtypes.test_select_dtypes_not_an_attr_but_still_valid_dtypec C sF t tdttddd}d}tjt|d | ¡ W 5 Q R X d S )Nr$ r% r& r z3at least one of include or exclude must be nonemptyrO )r rS rT ra rb Ú
|
ValueErrorr] ©r re Úmsgr r r Útest_select_dtypes_empty2 s z)TestSelectDtypes.test_select_dtypes_emptyc
|
C s¤ t tdttddt dd¡ d¡tjddd d
|
d d d gtjd ddjd}t j
|
t dd |j dgd W 5 Q R X t j
|
t dd |j dgd W 5 Q R X d S )Nr$ r% r& r' r( r) r* r+ r, r- TFrn r/ ro z.+ is too specificrO z datetime64[D]rF zdatetime64[as]rj ) r rS rT rU rV rW rX rZ rt ra rb r r] )r re r r r Ú!test_select_dtypes_bad_datetime648 s úÿ
|
z2TestSelectDtypes.test_select_dtypes_bad_datetime64c C sn t tdddtddddtdd}tj|j ¡ |j ¡ gd d
|
}|jd gd }|j g d }t
|
||¡ d S )NZ20130102r1 )r3 Z20130603r4 ©ÚAÚBé )Úindexr% )Zaxiszdatetime64[ns]rF )r ) r r rT rX Úconcatr Zto_framer r] Zreindexr_ r` )r Zdf2Zdf3rx ry r r r Ú#test_select_dtypes_datetime_with_tzI s
|
|
þû z4TestSelectDtypes.test_select_dtypes_datetime_with_tzr ÚstrZS1ÚunicodeZU1ÚargrJ c C s t tdtdttddt dd¡ d¡tjddd d
|
d d d gtjd ddjd}d}||gi}t j
|
t |d |j f | W 5 Q R X d S )Nr$ r% r& r' r( r) r* r+ r, r- TFrn r/ ©r; rA r< r= r> r? r@ z string dtypes are not allowedrO © r rS rT rU rV rW rX rZ rt ra rb Ú TypeErrorr] )r r r re r Úkwargsr r r Ú test_select_dtypes_str_raisesV s ùÿ
|
z.TestSelectDtypes.test_select_dtypes_str_raisesc C s t tdtdttddt dd¡ d¡tjddd d
|
d d d gtjd ddjd}d}t j
|
t |d | dg¡ W 5 Q R X d S )Nr$ r% r& r' r( r) r* r+ r, r- TFrn r/ r zdata type.*not understoodrO zblargy, blarg, blargr r r r r Ú!test_select_dtypes_bad_arg_raisesl s ùÿ z2TestSelectDtypes.test_select_dtypes_bad_arg_raisesc C s< t jdddd d}|}ttjd }t | |¡|¡ d S )Né r' c S s
|
t j ¡ S r )rU Úrandom)ÚxÚyr r r Ú<lambda> ó z?TestSelectDtypes.test_select_dtypes_typecodes.<locals>.<lambda>)Z
|
data_gen_fZAllFloat)r_ ZmakeCustomDataframerS rU Z typecodesr` r] )r re ry Z FLOAT_TYPESr r r Ú test_select_dtypes_typecodes} s z-TestSelectDtypes.test_select_dtypes_typecodesz arr,expectedr% r r- TZInt32)r Fc C s, t |}| tj¡j|jk}||ks(td S r )r r] rU r^ Úshaper )r Úarrry re Ú is_selectedr r r Útest_select_dtypes_numeric s z+TestSelectDtypes.test_select_dtypes_numericc C s: t jddg|d}t|}| tj¡j|jk}|r6td S )Nr; r< r- )rX Úarrayr r] rU r^ r§ r )r Znullable_string_dtyper¨ re r© r r r Ú*test_select_dtypes_numeric_nullable_string s z;TestSelectDtypes.test_select_dtypes_numeric_nullable_stringzexpected, float_dtypesr' r é é
|
é éÿÿÿÿ©r r ÚCÚfloatr² r c C sX t tjtjd}ttdtddtdddd}| |¡}|j|d}t ||¡ d S ) Nr± r' r r r® r¯ r° rF )
|
r³ rU r, Úfloat32r rT rW r] r_ r` )r ry Z float_dtypesZ
|
dtype_dictre rx r r r Ú test_select_dtypes_float_dtype s ÿ
|
z/TestSelectDtypes.test_select_dtypes_float_dtypec
|
C s~ t dddgtjdddgddt dddg¡t dddg¡tj dddg¡d}|jd d
|
}t d dddgi}t
|
||¡ d S ) Nr% r r' TFÚbooleanr- ri r^ rF r; ) r rX ZSeriesrU r« rY ZarraysZ SparseArrayr] r_ r` )r re rx ry r r r Ú&test_np_bool_ea_boolean_include_number¼ s ûÿ z7TestSelectDtypes.test_np_bool_ea_boolean_include_numberc C sJ t dddgdddgd}| ¡ }|jdgd }d
|
|jd < t ||¡ d S ) Nr% r r' r& r r( r r^ rF r )r r )r r" r] Zilocr_ r` )r re Zdf_origrx r r r Útest_select_dtypes_no_viewË s
|
|
z+TestSelectDtypes.test_select_dtypes_no_viewN),r r r rh rm rw ra ÚmarkZ parametrizerU ru r rz r{ r| r r r r r r r r Zstring_Zunicode_r r r¦ r« rs rX r r rª r¬ r rT rW r³ r, r´ rµ r· r¸ r r r r r# 4 s| , ÿ
|
$ ÿ üþ
|
ÿþü ÿþü
|
ÿüòþ
|
|
r# )ÚnumpyrU ra Zpandas.core.dtypes.dtypesr ZpandasrX r r Zpandas._testingZ_testingr_ Zpandas.core.arraysr r r r# r r r r Ú<module> s
|
