U
|
ý°d ã 2 @ sL d Z ddlZddlmZ ddlmZmZmZmZ ddlmZ ddl m
|
Z
|
ddl m Z ddl Z ddlmZ ddlZddlZdd lmZmZmZ ddlZddlZddlZdd
|
lmZm Z m Z ddl m! m"Z# dd l$m%Z% dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4 ddl5Z6dd l5m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZB ddlCmDZE ddlFmGZGmHZHmIZI ejJddgeKddd ZLG dd dZMdgddfg ddfddddidd fi dd fddhd d feN d d feOddhd d feO d d!fePdd"gdd#fePg dd$fd%d& d'D dd(fd)d& d*D dd+fe?dgdd,fe?g eQd-dd.fe?dgjKdd/fe?g d0d-jKdd1fe;dgdd2fe;g dd3fe8dggdd4fe8 dd5fe Rd6¡dd7fe Sg ¡dd8fe Rd9¡dd:fe Sg g¡dd;fe Rd<¡dd=fe Sg gg¡dd>fe Rd?¡dd@fe Sg ggg¡ddAfe Sd"¡ddBfeMe Rd6¡ddCfeMe Sg ¡ddDfeMe Rd9¡ddEfeMe Sg g¡ddFfeMe Rd<¡ddGfeMe Sg gg¡ddHfeMe Rd?¡ddIfeMe Sg ggg¡ddJfeMe Sd"¡ddKfdLdMdNdOdPeKddQfeQ ddRfejTddSfdTg1ZUeVeU \ZWZXZYejJeVeWeXeYddUdV ZZdWdX Z[dYdZ Z\d[d\ Z]d]d^ Z^d_d` Z_dadb Z`dcdd Zaejb cdeg dgdfd'ddiddhe?dge?g eQd-e?dgjKdgd& eddhD g
|
¡ejb cdieee?ejSefg¡djdk Zgejb cdldmg dgdfdgddhdd"dnge?dge8dodgidpd& eddhD g ¡dqdr Zhejb cdsi dodie?dge i¡ g¡dtdu Zjejb cdsdvddd"gd'edd"e;dgekejie?g ¡dwdx Zlejb cdyddg¡ejb cdzddg¡ejb cd{ddg¡d|d} Zmd~d Zne ddddg¡Zoejb cdseodd"dng¡dd Zpejb cdsd
de?ddig¡dd Zqdd Zrejb cdse sd¡g¡dd Ztejb cdsdd"dneQ g¡dd Zuejb cdsddde sd¡de sd¡g¡dd Zvejb cdsdg eQ g¡dd ZwG dd dZxG dd dZyG dd dZzG d d¡ d¡Z{ejb cd¢d£d¤d¥g¡d¦d§ Z|d¨d© Z}e#j~ejb dª¡d«d¬ Zdd® ZdS )¯z
|
These the test the public routines exposed in types/common.py
|
related to inference and not otherwise tested in types/test_common.py
|
|
é N)Ú
|
namedtuple)ÚdateÚdatetimeÚtimeÚ timedelta)ÚDecimal)ÚFraction)ÚStringIO)ÚNumber)ÚGenericÚIteratorÚTypeVar)ÚlibÚmissingÚops)Ú inference)Ú ensure_int32Úis_boolÚ
|
is_complexÚis_datetime64_any_dtypeÚis_datetime64_dtypeÚis_datetime64_ns_dtypeÚis_datetime64tz_dtypeÚis_floatÚ
|
is_integerÚ is_numberÚ is_scalarÚis_scipy_sparseÚis_timedelta64_dtypeÚis_timedelta64_ns_dtype) Ú CategoricalÚ DataFrameÚ
|
DateOffsetÚ DatetimeIndexÚIndexÚIntervalÚPeriodÚ PeriodIndexÚSeriesÚ TimedeltaÚTimedeltaIndexÚ Timestamp)Ú BooleanArrayÚ FloatingArrayÚ IntegerArrayTF)ÚparamsÚidsc C s | j S ©N©Úparam©Úrequest© r6 úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/dtypes/test_inference.pyÚcoerceO s r8 c @ st e Zd ZdZddddZedddZeddd Zdd
|
d Z e
|
d d Z e
|
dd Z e
|
dd Z e
|
dd ZdS )ÚMockNumpyLikeArraya·
|
A class which is numpy-like (e.g. Pint's Quantity) but not actually numpy
|
|
The key is that it is not actually a numpy array so
|
``util.is_array(mock_numpy_like_array_instance)`` returns ``False``. Other
|
important properties are that the class defines a :meth:`__iter__` method
|
(so that ``isinstance(abc.Iterable)`` returns ``True``) and has a
|
:meth:`ndim` property, as pandas special-cases 0-dimensional arrays in some
|
cases.
|
|
We expect pandas to behave with respect to such duck arrays exactly as
|
with real numpy arrays. In particular, a 0-dimensional duck array is *NOT*
|
a scalar (`is_scalar(np.array(1)) == False`), but it is not list-like either.
|
N©Úreturnc C s
|
|| _ d S r1 )Ú_values)ÚselfÚvaluesr6 r6 r7 Ú__init__d s zMockNumpyLikeArray.__init__c s t | j fdd}| S )Nc 3 s E d H d S r1 r6 r6 ©Z iter_valuesr6 r7 Úit_outerj s z-MockNumpyLikeArray.__iter__.<locals>.it_outer)Úiterr< )r= rA r6 r@ r7 Ú__iter__g s
|
zMockNumpyLikeArray.__iter__c C s
|
t | jS r1 )Úlenr< ©r= r6 r6 r7 Ú__len__o s zMockNumpyLikeArray.__len__c C s t j| j|dS ©N©Údtype)ÚnpÚasarrayr< )r= Útr6 r6 r7 Ú __array__r s z MockNumpyLikeArray.__array__c C s | j jS r1 )r< ÚndimrE r6 r6 r7 rN u s zMockNumpyLikeArray.ndimc C s | j jS r1 )r< rI rE r6 r6 r7 rI y s zMockNumpyLikeArray.dtypec C s | j jS r1 )r< ÚsizerE r6 r6 r7 rO } s zMockNumpyLikeArray.sizec C s | j jS r1 )r< ÚshaperE r6 r6 r7 rP s zMockNumpyLikeArray.shape)N)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r? r rC ÚintrF rM ÚpropertyrN rI rO rP r6 r6 r6 r7 r9 T s
|
|
|
|
r9 é Úlistz
|
list-empty)©rW TÚtuple)r6 Tz tuple-emptyÚaÚdictz
|
dict-emptyÚsetz set-emptyÚ frozensetzfrozenset-emptyé Úiteratorziterator-emptyc c s | ]
|
}|V qd S r1 r6 ©Ú.0Úxr6 r6 r7 Ú <genexpr> s rd ©rW r_ Ú generatorc c s | ]
|
}|V qd S r1 r6 ©rb Ú_r6 r6 r7 rd s r6 zgenerator-emptyr( rH z Series-emptyZ StringMethodsÚOzStringMethods-emptyr$ z Index-emptyr! zDataFrame-empty)r_ z
|
ndarray-1dzndarray-1d-empty)r_ r_ z
|
ndarray-2dzndarray-2d-empty)r_ r_ r_ z
|
ndarray-3dzndarray-3d-empty)r_ r_ r_ r_ z
|
ndarray-4dzndarray-4d-emptyz
|
ndarray-0dzduck-ndarray-1dzduck-ndarray-1d-emptyzduck-ndarray-2dzduck-ndarray-2d-emptyzduck-ndarray-3dzduck-ndarray-3d-emptyzduck-ndarray-4dzduck-ndarray-4d-emptyzduck-ndarray-0d)rW FrU )s 123FÚbytes)ó Fz bytes-empty)Z123FÚstring)Ú Fz string-emptyz string-typeÚobjectÚNaN)NFÚNonec C s | j S r1 r2 r4 r6 r6 r7 Úmaybe_list_like¾ s rq c C s. | \}}|dkrdn|}t |¡|ks*td S )Nr] T©r Ú is_list_likeÚAssertionError©rq ÚobjÚexpectedr6 r6 r7 Útest_is_list_likeà s rx c C s2 | \}}|dkrdn|}t j|dd|ks.td S )Nr] F)Z
|
allow_setsrr ru r6 r6 r7 Ú test_is_list_like_disallow_setsÉ s ry c sP fdd t ¡ } z*t d¡ t t¡ W 5 Q R X W 5 t | ¡ X d S )Nc s t g ¡ d S r1 )r rs r6 ©Ú list_liker6 r7 r{ Ò s
|
z.test_is_list_like_recursion.<locals>.list_likeéd )ÚsysÚgetrecursionlimitÚsetrecursionlimitÚtmZexternal_error_raisedÚRecursionError)Z rec_limitr6 rz r7 Útest_is_list_like_recursionÏ s
|
r c C s" G dd d} t | ¡r td S )Nc @ s e Zd Zdd ZdZdS )z3test_is_list_like_iter_is_none.<locals>.NotListLikec S s | S r1 r6 ©r= Úitemr6 r6 r7 Ú __getitem__ä s z?test_is_list_like_iter_is_none.<locals>.NotListLike.__getitem__N)rQ rR rS r
rC r6 r6 r6 r7 Ú NotListLikeã s r rr )r r6 r6 r7 Ú test_is_list_like_iter_is_noneà s r c C sj t d} G dd dtt| }|t }|t ddddgi}t |¡rJtt|tsXtt |¡sftd S )NÚTc @ s e Zd ZdS )z.test_is_list_like_generic.<locals>.MyDataFrameN©rQ rR rS r6 r6 r6 r7 Ú MyDataFrameñ s r rc rW r_ é )r r! r rU r rs rt Ú
|
isinstance)r r ZtstcZtstr6 r6 r7 Útest_is_list_like_genericì s r c C s\ t j} | dst| ddgs"t| dr.t| tjr<tG dd d}| | rXtd S )Nre rW r_ Zabcdc @ s e Zd Zdd ZdS )ztest_is_sequence.<locals>.Ac S s dS )NrW r6 r r6 r6 r7 r
s z'test_is_sequence.<locals>.A.__getitem__N)rQ rR rS r
r6 r6 r6 r7 ÚA s r )r Z is_sequencert rJ Úint64)Zis_seqr r6 r6 r7 Útest_is_sequenceü s r c C sî t tg td¡stt tddg¡s,tt t ddg¡¡sDtt tdg¡sXtt t ddg¡¡sptt tt ddg¡¡stG dd d t } t | ¡s¬tt dddg¡rÀtt d
|
¡rÎtt d ¡rÜtt d ¡rêtd S ) NrH rW r_ r[ Úbú
|
2016-01-01r c @ s e Zd ZdZdS )z%test_is_array_like.<locals>.DtypeListÚspecialN)rQ rR rS rI r6 r6 r6 r7 Ú DtypeList s r r6 Úfooé{ )
|
r Z is_array_liker( rn rt rJ Úarrayr$ r9 rX )r r6 r6 r7 Útest_is_array_like
|
s r ÚinnerrY c c s | ]
|
}|V qd S r1 r6 ra r6 r6 r7 rd ) s é Úouterc s, | fddt dD }t |¡s(td S )Nc s g | ]} qS r6 r6 rg ©r r6 r7 Ú
|
<listcomp>. s z3test_is_nested_list_like_passes.<locals>.<listcomp>r )Úranger rs rt )r r Úresultr6 r r7 Ú test_is_nested_list_like_passes s r rv Úabcr r c c s | ]}d dgV qdS )rW r_ Nr6 rg r6 r6 r7 rd ? s c C s t | ¡rtd S r1 )r Zis_nested_list_likert )rv r6 r6 r7 Ú test_is_nested_list_like_fails2 s r¢ Úllc C s t | ¡std S r1 ©r Z is_dict_likert ©r£ r6 r6 r7 Útest_is_dict_like_passesF s r¦ Ú1c C s t | ¡rtd S r1 r¤ r¥ r6 r6 r7 Útest_is_dict_like_failsK s r¨ Úhas_keysÚ has_getitemÚ has_containsc sH G fddd}|ddi}t |¡}o6o6 }||ksDtd S )Nc sD e Zd ZddddZr"dd Zr.dd Z r@edd d
|
ZdS ) z-test_is_dict_like_duck_type.<locals>.DictLikeNr: c S s
|
|| _ d S r1 )Úd)r= r¬ r6 r6 r7 r? b s z6test_is_dict_like_duck_type.<locals>.DictLike.__init__c S s
|
| j ¡ S r1 )r¬ ÚkeysrE r6 r6 r7 r g s z2test_is_dict_like_duck_type.<locals>.DictLike.keysc S s | j |¡S r1 )r¬ r
©r= Úkeyr6 r6 r7 r
l s z9test_is_dict_like_duck_type.<locals>.DictLike.__getitem__c S s | j |¡S r1 )r¬ Ú __contains__r® r6 r6 r7 r° q s z:test_is_dict_like_duck_type.<locals>.DictLike.__contains__)rQ rR rS r? r r
Úboolr° r6 ©r« rª r© r6 r7 ÚDictLikea s r³ rW r_ r¤ )r© rª r« r³ r¬ r rw r6 r² r7 Útest_is_dict_like_duck_type] s
|
|
r´ c C s¬ G dd d} t j}td}||s(t| }||r:tdd | _| }||rVtdd | _| }||srt| `dd | _| }||stdd d
|
g}||r¨td S ) Nc @ s e Zd ZdS )z#test_is_file_like.<locals>.MockFileNr r6 r6 r6 r7 ÚMockFile| s rµ Údatac S s dS ©Nr r6 rE r6 r6 r7 Ú<lambda> rk z#test_is_file_like.<locals>.<lambda>c S s | S r1 r6 rE r6 r6 r7 r¸ rk c S s dS r· r6 rE r6 r6 r7 r¸ rk rW r_ r )r Z is_file_liker rt ÚwriterC Úread)rµ Úis_filer¶ Úmr6 r6 r7 Útest_is_file_like{ s$
|
|
|
|
r½ Ú
|
test_tupler Úcc C s t | ¡std S r1 ©r Zis_named_tuplert r¥ r6 r6 r7 Útest_is_names_tuple_passes¦ s rÁ )rW r_ r Úpiç
ëQ¸ @c C s t | ¡rtd S r1 rÀ r¥ r6 r6 r7 Útest_is_names_tuple_fails« s rÄ c C sº G dd d} G dd d}G dd d}ddt d¡d d
|
d | f}g | f}g f| f}|D ]}t |¡s^tq^|D ]}t |¡rvtqv|D ]}t |¡rtqt t g ¡¡r¶td S ) Nc @ s e Zd ZdS )z'test_is_hashable.<locals>.HashableClassNr r6 r6 r6 r7 Ú HashableClass² s rÅ c @ s e Zd ZdZdS )z*test_is_hashable.<locals>.UnhashableClass1N©rQ rR rS Ú__hash__r6 r6 r6 r7 ÚUnhashableClass1µ s rÈ c @ s e Zd Zdd ZdS )z*test_is_hashable.<locals>.UnhashableClass2c S s t dd S )Nz Not hashable)Ú TypeErrorrE r6 r6 r7 rÇ ¹ s z3test_is_hashable.<locals>.UnhashableClass2.__hash__NrÆ r6 r6 r6 r7 ÚUnhashableClass2¸ s rÊ rW rà r[ r6 rY )rJ Úfloat64r Z is_hashablert r )rÅ rÈ rÊ ZhashableZ not_hashableZ abc_hashable_not_really_hashableÚir6 r6 r7 Útest_is_hashable° s
|
rÍ Úadc C s t | ¡std S r1 ©r Zis_rert r¥ r6 r6 r7 Útest_is_re_passesÍ s rÐ rc c C s t | ¡rtd S r1 rÏ r¥ r6 r6 r7 Útest_is_re_failsÒ s rÑ ZasdfZadsfz \u2233\s*rm c C s t | ¡std S r1 ©r Zis_re_compilablert r¥ r6 r6 r7 Útest_is_recompilable_passes× s rÓ c C s t | ¡rtd S r1 rÒ r¥ r6 r6 r7 Útest_is_recompilable_failsÞ s rÔ c @ sÈ e Zd Zej dejedddejeddd e
|
¡dej dgg¡dd Z ej d e d
|
d fejd fej d fd dg¡dd Zej d e dd fej d fejd fd dg¡dd Zej dd eejd d gdde d d g¡fd ejd ej gddfg¡dd Zej dd d g¡ej dd d g¡ej dd
|
d d d d d d!d"g¡ej d#d$d%d&g¡d'd( Zej dd d g¡d)d* Zej dd d g¡d+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zej dejd5ej ge
|
dejed5ej ge
|
dejej d5ge
|
dejej ed5ge
|
dg¡d6d7 Zej dd d g¡d8d9 Z ej d:ejd5d;ge
|
dejed5d;ge
|
dejed5ed;ge
|
dejd;d5ge
|
dejd;ed5ge
|
dejed;ed5ge
|
dg¡ej dd d g¡d<d= Z ej dd d g¡d>d? Z ej d@dAdBg¡dCdD Z ej dEdFej e dGg¡ej dHdIdJg¡dKdL Z ej dMd5gej!fe !d5¡gej!fdNd;gej"fd5d;ge
|
fe #dO¡gej#fe $dO¡gej$fe %dO¡gej%fe !dO¡gej!fe #dN¡e $dO¡gej$fe %dN¡e $dO¡gej%fe %dN¡d;ge
|
fe %dN¡dOgej!fe %dN¡e &dO¡ge
|
fg ¡dPdQ Z'dRdS Z(dTdU Z)dVdW Z*ej dHdXdYg¡dZd[ Z+d\d] Z,d^d_ Z-ej d`e.ejdNdagdbde d d g¡e.ejdNdagdcde d d g¡g¡ddde Z/ej dfdgdhe 0ej"¡j1dO fg¡didj Z2ej dd e.ejdNdagdbde d d g¡fd ejdNej gdkdfg¡dldm Z3ej dd e4ejdndogdkde d d g¡fd ejdnej gdkdfg¡dpdq Z5drds Z6dtdu Z7ej dEdFej g¡dvdw Z8ej dxd dOdydze 9dO¡e :dO¡e &dO¡e "dO¡e ;dO¡e <dO¡e =dO¡e >dO¡e ?dO¡g ¡ej d{d dOdydze 9dO¡e :dO¡e &dO¡e "dO¡e ;dO¡e <dO¡e =dO¡e >dO¡e ?dO¡g ¡d|d} Z@d~d ZAej deBjCjDeEdddeBjFdd
ddg¡dd ZGdFS )Ú TestInferenceÚarrr¡ ZS1rH ó aó cc C s t j|dd}|dkstd S )NT©Úskipnarj ©r Ú infer_dtypert ©r= rÖ r r6 r6 r7 Útest_infer_dtype_bytesä s z$TestInference.test_infer_dtype_byteszvalue, expectedÚinfTF)rW F)r[ Fc C s t |¡}||kstd S r1 )Ú
|
libmissingZisposinf_scalarrt ©r= Úvaluerw r r6 r6 r7 Útest_isposinf_scalarð s
|
z"TestInference.test_isposinf_scalarú-infc C s t |¡}||kstd S r1 )rà Zisneginf_scalarrt rá r6 r6 r7 Útest_isneginf_scalarÿ s
|
z"TestInference.test_isneginf_scalarz convert_to_masked_nullable, expr± rn c C sT t jdt jgtd}tj|t |d}|r<t t | |¡ n|d }t
|
||¡ d S )NTrH ©Úconvert_to_masked_nullabler ) rJ r ro rn ÚlibopsZmaybe_convert_boolr] r Ú assert_extension_array_equalr, Úassert_numpy_array_equal©r= rç ÚexprÖ r r6 r6 r7 Ú#test_maybe_convert_nullable_boolean s ÿz1TestInference.test_maybe_convert_nullable_booleanrç Úcoerce_numericÚinfinityZinFZiNfZInfZiNFZInFZINfZINFÚprefixrm ú-ú+c C sZ t jtj|| gtddddh||d\}}t |dkr>tjntj g¡}t ||¡ d S )NrH rm ÚNULLÚnan©Ú na_valuesrî rç )rm rò )r Úmaybe_convert_numericrJ r rn rß r rê )r= rî rï rð rç r rh rw r6 r6 r7 Ú%test_maybe_convert_numeric_infinities% s
|
ü
|
z3TestInference.test_maybe_convert_numeric_infinitiesc C sF d}t jt|d* tjtjdgtddddhd|d W 5 Q R X d S )
|
NzUnable to parse string©ÚmatchZfoo_infrH rm ró rô Frõ )ÚpytestÚraisesÚ
|
ValueErrorr r÷ rJ r rn )r= rç Úmsgr6 r6 r7 Ú,test_maybe_convert_numeric_infinities_raises8 s üz:TestInference.test_maybe_convert_numeric_infinities_raisesc C s t jdddgtd}t jdt jdgt jd}ddh}tj||||d }|rlt|t |¡}t
|
|t| ¡ n|d
|
}t ||¡ d S ) Nz1.200z-999.000z4.500rH g333333ó?g @iüÿÿg 8Àræ r ) rJ r rn rô rË r r÷ r- Úisnanr ré rê )r= r8 rç r¶ rw Z
|
nan_valuesÚoutr6 r6 r7 Ú,test_maybe_convert_numeric_post_floatify_nanC s üz:TestInference.test_maybe_convert_numeric_post_floatify_nanc C st t jdddgdd}t |t d¡\}}|jt jks8tt jdddgdd}t |t d¡\}}|jt jksptd S )Nrß ri rH Frä )rJ r r r÷ r] rI rË rt ©r= rÖ r rh r6 r6 r7 Útest_convert_infsY s z TestInference.test_convert_infsc C sD t jddddgdd}t |t dd¡\}}t t |¡¡s@td S ) NZ42EZ2EZ99eZ6eri rH FT)rJ r r r÷ r] Úallr rt r r6 r6 r7 Útest_scientific_no_exponentb s z)TestInference.test_scientific_no_exponentc C sN t jddgddgtd}t |t dd¡\}}t |t t jdt jg¡¡ d S )Ng $@r_ ç ð?ZapplerH FT) rJ r rn r r÷ r] r rê rô r r6 r6 r7 Útest_convert_non_hashableh s z'TestInference.test_convert_non_hashablec C s t jdgtd}t jdgt jd}t t |t ¡d |¡ t jt dgtd}t jdgt jd}t t |t ¡d |¡ t jt d¡gtd}t jdgt jd}t t |t ¡d |¡ d S )Nì rH r )
|
rJ r rn Úuint64r rê r r÷ r] Ústr)r= rÖ rì r6 r6 r7 Útest_convert_numeric_uint64o s z)TestInference.test_convert_numeric_uint64r c C s< |r| t¡n| ¡ }tj|t |d\}}t ||¡ d S )N)rî ©ÚastypeÚfloatÚcopyr r÷ r] r Úassert_almost_equal)r= r8 rÖ rw r rh r6 r6 r7 Ú test_convert_numeric_uint64_nan| s
|
z-TestInference.test_convert_numeric_uint64_nanc C s t jddgtd}dh}|r0t jt jdgtdn| ¡ }tj||||d}|r|rtt jddgddt jddgd d}t| }n|d }t
|
||¡ d S )
|
Nr l rH ©rî rç r Úu8TFr± ) rJ r rn rô r r r r÷ r. r r )r= r8 rç rÖ rö rw r r6 r6 r7 Ú&test_convert_numeric_uint64_nan_values s$ ÿüþ
|
z4TestInference.test_convert_numeric_uint64_nan_valuesÚcaseéÿÿÿÿc C s> |r| t¡n| ¡ }tj|t ||d\}}t ||¡ d S )Nr r )r= r r8 rç rw r rh r6 r6 r7 Ú!test_convert_numeric_int64_uint64¤ s ü
|
z/TestInference.test_convert_numeric_int64_uint64c C sH t jtjdgtdt d|d}|r.t| }n|d }t |¡sDtd S )Nr
|
rH Tr r ) r r÷ rJ r rn r] r- r rt )r= rç r r6 r6 r7 Ú"test_convert_numeric_string_uint64½ s ü
|
z0TestInference.test_convert_numeric_string_uint64râ lûÿÿÿ l c C s* t j|gtd}t |¡}t ||¡ d S rG )rJ r rn r Úmaybe_convert_objectsr rê )r= râ rÖ r r6 r6 r7 Útest_convert_int_overflowÌ s
|
z'TestInference.test_convert_int_overflowÚvalNrô rI zM8[ns]úm8[ns]c C s t |¡}t jtj|gtd}tj|dd|d}|j|ks>tt |¡
|
¡ sPttj|d d d
dd|d}|j|ksztt |¡
|
¡ std S )NrH T©Úconvert_datetimeÚconvert_timedeltaZdtype_if_all_natr ) rJ rI r ÚpdÚNaTrn r r rt Zisnatr )r= r rI Úvalsr r6 r6 r7 Ú(test_maybe_convert_objects_nat_inferenceÓ s$
|
ü üz6TestInference.test_maybe_convert_objects_nat_inferencezvalue, expected_dtyper_ rW c C s2 t j|td}t j||d}t t |¡|¡ d S rG )rJ r rn r rê r r )r= râ Zexpected_dtyperÖ rì r6 r6 r7 Ú test_maybe_convert_objects_uintê s z-TestInference.test_maybe_convert_objects_uintc C sæ t jt d¡t dd¡gtd}| ¡ }tj|ddd}t ||¡ t jt
|
j t dd¡gtd}t jt d¡t dd¡gdd}tj|ddd}t ||¡ t jt dd¡t j gtd}|d d d
}tj|ddd}t ||¡ d S )
|
Nz
|
2000-01-01rW ÚsrH T©r r r" r r ) rJ r Ú
|
datetime64Ú timedelta64rn r r r r rê r! r" rô ©r= rÖ rì r r6 r6 r7 Ú#test_maybe_convert_objects_datetime s6 ÿ ÿ ÿ ÿz1TestInference.test_maybe_convert_objects_datetimec C s¢ t jtjtjgtd}tj|ddd}t ||¡ tj|ddt d¡d}t jddgdd}t ||¡ tj|ddt d¡d}t jddgdd}t ||¡ d S )NrH Tr' útimedelta64[ns]r r" údatetime64[ns])
|
rJ r r! r" rn r r r rê rI )r= rÖ r rì r6 r6 r7 Ú+test_maybe_convert_objects_dtype_if_all_nat s. ÿ ü üz9TestInference.test_maybe_convert_objects_dtype_if_all_natc C sL t jtjtjgtd}tjtdd tj |ddt
|
d¡d W 5 Q R X d S )NrH r rù Tr ) rJ r r! r" rn rû rü rý r r rI ©r= rÖ r6 r6 r7 Ú3test_maybe_convert_objects_dtype_if_all_nat_invalid: s üzATestInference.test_maybe_convert_objects_dtype_if_all_nat_invalidr- r, c C sT t ddd}|dkr$|t ddd }tj|gtd}tj|ddd }t ||¡ d S )
|
Ni; é
|
é r, i² rW rH Tr' )r rJ r rn r r r rê )r= rI ZstamprÖ r r6 r6 r7 Ú1test_maybe_convert_objects_datetime_overflow_safeF s ÿz?TestInference.test_maybe_convert_objects_datetime_overflow_safec C sp t d}|| ¡ | ¡ tjtjd g}t |¡D ]<}tj t
|
|t d}t |j j}tj|dd}t ||¡ q.d S )NÚnowrH T©r )r+ Z to_pydatetimeZ to_datetime64r! r" rJ rô Ú itertoolsÚ permutationsr rX rn r# Ú_dataZ_ndarrayr r r rê )r= Útsr# r¶ rw r r6 r6 r7 Ú*test_maybe_convert_objects_mixed_datetimesS s z8TestInference.test_maybe_convert_objects_mixed_datetimesc C sZ t dd¡}t j|gtd}|d |ks,ttj|dd}t j|gdd}t ||¡ d S )Nr" ÚnsrH r T)r r ) rJ r) r rn rt r r r rê )r= rv rÖ r rw r6 r6 r7 Ú*test_maybe_convert_objects_timedelta64_nat] s z8TestInference.test_maybe_convert_objects_timedelta64_natrì r Úi8r c C s2 t jdt jgtd}tj|dd}t ||¡ d S )Nr_ rH T©Zconvert_to_nullable_dtype)rJ r ro rn r r r ré )r= rì rÖ r r6 r6 r7 Ú+test_maybe_convert_objects_nullable_integerg s z9TestInference.test_maybe_convert_objects_nullable_integerz
|
dtype, val)r rW r
|
c C sX t j|d dgdd}tj|dd}tt j|ddg|dt dddg¡}t ||¡ d S )Nr rn rH Tr> r F)rJ r r r r. r ré )r= rI r rÖ r rw r6 r6 r7 Ú(test_maybe_convert_objects_nullable_noneu s ÿz6TestInference.test_maybe_convert_objects_nullable_nonerË c C sX t jdt jgtd}tj|t |d}|r@t| }t ||¡ n|d }t
|
||¡ d S )Nr_ rH ræ r ) rJ r ro rn r r÷ r] r. r ré rê rë r6 r6 r7 Ú+test_maybe_convert_numeric_nullable_integer s ÿz9TestInference.test_maybe_convert_numeric_nullable_integerç @ç c C sT t jdt jgtd}tj|t |d}|r<t t | |¡ n|d }t
|
||¡ d S )NrB rH ræ r ) rJ r rô rn r r÷ r] r ré r- rê rë r6 r6 r7 Ú)test_maybe_convert_numeric_floating_array s ÿz7TestInference.test_maybe_convert_numeric_floating_arrayc C sJ t ddtjgtd}tjddtjgtd}tj|jdd}t ||¡ d S )NTFrH rW )Úsafe)
|
r$ rJ rô rn r r r r> r rê )r= Úindrì r r6 r6 r7 Ú#test_maybe_convert_objects_bool_nan± s z1TestInference.test_maybe_convert_objects_bool_nanc C s t jddgtd}t ddg¡}tj|dd}t ||¡ t jddtjgtd}t jddtjgtd}tj|dd}t ||¡ d S ©NTFrH r> ) rJ r rn r r r rê r! r" r* r6 r6 r7 Ú+test_maybe_convert_objects_nullable_boolean¸ s z9TestInference.test_maybe_convert_objects_nullable_booleanc C sT t jdd|gtd}tt dddg¡t dddg¡}tj|dd}t ||¡ d S rH )rJ r rn r, r r r ré )r= r rÖ rì r r6 r6 r7 Ú.test_maybe_convert_objects_nullable_boolean_naÄ s ÿz<TestInference.test_maybe_convert_objects_nullable_boolean_naÚdata0r ù ð? ð?Údata1c C s ||g}t j|dd}t jt|t|gg dj}t|ds@dn|jj}t|dsVdn|jj}|dkr|dkr|}t|jj|jj} nPt |st |r´t |rªt |rªdnd}d} n"t
|
|sÄt
|
|rÎ|}d} n|}d } t j|| | d}
|
t |¡} t | |
|
¡ d S )
|
Nrn rH )Z scalar_typesrI Úpythonr± rm é é )rJ r Zfind_common_typeÚtypeÚkindÚhasattrrI ÚmaxÚitemsizer r r r r rê ) r= rK rM r¶ rÖ Z common_kindZkind0Zkind1rR rU rw r r6 r6 r7 Ú#test_maybe_convert_objects_itemsizeÎ s, & ÿ
|
z1TestInference.test_maybe_convert_objects_itemsizec C s> t jtdddtjddgtd}tj|dd}t ||¡ d S )Niß rW )ÚtzinforH Tr5 )
|
rJ r r ÚpytzÚutcrn r r r rê rÝ r6 r6 r7 Ú%test_mixed_dtypes_remain_object_array s z3TestInference.test_mixed_dtypes_remain_object_arrayÚidxr Úboth©Úclosedr r ÚD)ÚperiodsÚfreqc C s, t jtj|tdddd}t ||j¡ d S )NrH T)Zconvert_periodZconvert_interval)r r rJ r rn r ré r8 )r= r[ r r6 r6 r7 Ú test_maybe_convert_objects_ea s ýz+TestInference.test_maybe_convert_objects_ea)HrQ rR rS rû ÚmarkÚ parametrizerJ r rX r rn rô rÞ r rß rã rå r, rí rø rÿ r r r r r r r r r r r r$ r
|
r Zuint8Zuint16Zuint32Úint32r% r+ r. r0 r3 r: r< r. r? ZiinforT r@ rA r- rD rG rI rJ Zint8Úint16Úfloat16Úfloat32rË Ú complex64Ú
|
complex128rV rZ r! Ú IntervalIndexÚ from_breaksr Z period_rangerb r6 r6 r6 r7 rÕ ã sz
|
ýþ
|
|
|
ûþ
|
|
|
|
ûþ
|
|
ÿþùþ
|
ÿ
|
|
|
üþ
|
|
úþ
|
|
|
|
îþ
|
|
|
|
þþ
|
ÿ
|
$þþ
|
ÿþùþ
|
|
óþóþ þþrÕ c @ s e Zd ZG dd dZdd Zej dddg¡dd Zd
|
d Z ej d e
|
j d de
|
j e
|
j dgdddfe
|
j d de
|
j e
|
j dgdddfe
|
j d dde
|
d¡e
|
d¡e
|
j gdddfe
|
j d dde
|
d¡e
|
d¡e
|
j gdddfg¡dd Zdd Zdd Zdd Zd d Zej dddg¡d d Zd d! Zd"d# Zej d$ee
|
j dd%fee
|
j dd%fee
|
j dd%fee
|
j dd&fed'dd(fed'dd&fg¡ej d)ee
|
j g¡d*d+ Zd,d- Zd.d/ Z ej d0e j e
|
j g¡d1d2 Z ej d3e
|
j e
|
d4¡e
|
!d5¡gede
|
j e
|
!d5¡e
|
d4¡gede
|
e
|
!d6¡e"d5g¡e
|
e"d5e
|
!d6¡g¡e
|
e
|
j e"d5d7g¡e
|
j e
|
j d6e"d5gede
|
j e
|
!d4¡e
|
d d8¡gede
|
j e
|
d d8¡e
|
!d4¡gedg¡d9d: Z#d;d< Z$ej d3e
|
e"d6e"d5g¡e
|
e%d=d d e%d>dd g¡e
|
e%d=d d e"d5g¡g¡d?d@ Z&ej d0e j e
|
j g¡ej dAe"d6e%d=d d g¡dBdC Z'ej d3e
|
e(dDe(dEg¡e
|
j e
|
d d8¡e
|
dd8¡gede
|
e)d e)dg¡g¡dFdG Z*ej d0e j e
|
j g¡ej dHe(dDe
|
d d8¡e)d g¡dIdJ Z+dKdL Z,ej dMe j ee-g¡ej dddg¡dNdO Z.dPdQ Z/ej d0e j e
|
j g¡dRdS Z0dTdU Z1dVdW Z2ej dXdYdZd[d\d]d^d_g¡d`da Z3dbdc Z4ej dde5ded d e"dfge"dfe5ded d ge5ded d e j ge j e5ded d gg¡ej dddg¡dgdh Z6didj Z7dkdl Z8dmdn Z9dodp Z:dqdr Z;dsdt Z<dudv Z=ej dwddg¡dxdy Z>ej dze"d{e(d{d{d|g¡d}d~ Z?dd Z@ej dMe j eg¡ej dddg¡ej ddddgdde jAgg¡d
d ZBej dMe j eg¡ej dddg¡ej ddddgdde jAgg¡dd ZCd'S )ÚTestTypeInferencec @ s e Zd ZdS )zTestTypeInference.DummyNr r6 r6 r6 r7 ÚDummy% s rn c C s" |\}}|t j|ddks td S )NTrÙ rÛ )r= Zany_skipna_inferred_dtypeZinferred_dtyper> r6 r6 r7 Útest_inferred_dtype_fixture( s z-TestTypeInference.test_inferred_dtype_fixturerÚ TFc C s~ t jtjg dd|d}|dks$tt jg |d}|dks>tt tjg tdtjg tdg¡}t j||d}|dksztd S )NÚi4rH rÙ ÚintegerÚempty)r rÜ rJ r rt rn )r= rÚ r rÖ r6 r6 r7 Útest_length_zero/ s "z"TestTypeInference.test_length_zeroc C s´ t jdddt d¡t d¡gdd}tj|dd }|d
|
ks>tt jdddt d¡t d¡d gdd}tj|dd }|d ks~tt jdddddgd d}tj|dd }|d
|
ks°td S )NrW r_ r r2 r ri rH TrÙ rq r ú mixed-integerrp )rJ r r re r rÜ rt rÝ r6 r6 r7 Ú test_integers< s $ & z TestTypeInference.test_integersz arr, skipnarW r_ r ri rH r2 r c C s* t j||d}|rdnd}||ks&td S )NrÙ rq z
|
integer-narÛ )r= rÖ rÚ r rw r6 r6 r7 Útest_integer_naI s z!TestTypeInference.test_integer_nac C s2 t jdddt jgtd}t |¡}|dks.td S )NrW r_ r rH rq )rJ r rô rn r rÜ rt rÝ r6 r6 r7 Ú test_infer_dtype_skipna_defaultX s
|
z1TestTypeInference.test_infer_dtype_skipna_defaultc C s t jdddddgdd}tj|dd}|dks2tt jt d¡t d¡gdd}tj|dd}|dksjtt jddddgdd}tj|dd}|dkstt jdddgtd}tj|dd}|dksÈtt jdt jdgdd}tj|dd}|dksøttj|dd}|dkstd S ) NTFri rH rÙ Úbooleanr Úmixed)rJ r r rÜ rt Úbool_r± rô rÝ r6 r6 r7 Ú
|
test_bools` s" z TestTypeInference.test_boolsc C sæ t jdddt d¡t d¡gdd}tj|dd }|d
|
ks>tt jd d d t d¡t d¡dgdd}tj|dd }|dks~tt jd d d ddgdd}tj|dd }|d
|
ks°tt jd d d ddgdd}tj|dd }|d
|
ksâtd S )Nr rB g @r2 r ri rH TrÙ ÚfloatingrW r_ r r rt Zf4Zf8)rJ r rË rh r rÜ rt rÝ r6 r6 r7 Ú test_floatsx s $ & z TestTypeInference.test_floatsc C sú t tdtdtdg¡}tj|dd}|dks6tt ddtdg¡}tj|dd}|d ksdttj|d d d
|
dd}|d kstt tdtd tdg¡}tj|dd}|dks¾tt jtdt jtdgd d }tj|dd}|dksötd S )NrW r_ r TrÙ Údecimalr rB ry r ro ri rH )rJ r r r rÜ rt rô rÝ r6 r6 r7 Ú test_decimals s z TestTypeInference.test_decimalsc C sB t dddg¡}tj||d}|dks*tt jdddgdd}tj||d}|dksXttj|d d d
|d}|dks|tt d
|
t jdg¡}tj||d}|dks¨tt jdt jdgdd}tj||d}|dksØtt jdt jd gdd}tj||d}|dks
|
tt jdt jd gt jd}tj||d}|dks>td S ) Nr rB rL rÙ Úcomplexri rH ry r rW y @ @)rJ r r rÜ rt rô ri )r= rÚ rÖ r r6 r6 r7 Ú test_complex s( z TestTypeInference.test_complexc C s d S r1 r6 rE r6 r6 r7 Ú test_stringÀ s z TestTypeInference.test_stringc C s¾ dt jdg}tj|dd}|dks&tdt jdg}tj|dd}|dksLtdtjdg}tj|dd}|dksrtdtjdg}tj|dd}|dkstddg}tj|dd}|dksºtd S )Nr[ r¿ FrÙ ry Trl )rJ rô r rÜ rt r! ÚNAr" rÝ r6 r6 r7 Ú test_unicodeà s z TestTypeInference.test_unicodez dtype, missing, skipna, expectedr| rr Nry Úboxc C s. |||g|d}t j||d}||ks*td S )NrH rÙ rÛ )r= r
r rI rÚ rw rÖ r r6 r6 r7 Útest_object_emptyÜ s z#TestTypeInference.test_object_emptyc C s. dd t ddD }t|}|jdks*td S )Nc S s g | ]}t d d|qS ©éÜ rW )r ra r6 r6 r7 r ð s z3TestTypeInference.test_datetime.<locals>.<listcomp>rW é r( )r r$ Ú inferred_typert )r= ÚdatesÚindexr6 r6 r7 Ú test_datetimeï s z TestTypeInference.test_datetimec C s8 t jt d¡t d¡gtd}tj|dddks4td S )Nú
|
2011-01-01rH TrÙ r( )rJ r r( rn r rÜ rt r/ r6 r6 r7 Útest_infer_dtype_datetime64ô s
|
ÿz-TestTypeInference.test_infer_dtype_datetime64Úna_valuec C sZ t |t d¡g¡}tj|dddks*tt |t d¡|g¡}tj|dddksVtd S )Nú
|
2011-01-02TrÙ r( )rJ r r( r rÜ rt ©r= r rÖ r6 r6 r7 Ú#test_infer_dtype_datetime64_with_naú s z5TestTypeInference.test_infer_dtype_datetime64_with_narÖ Únatr r çñ?r_ c C s t j|dddkstd S )NFrÙ ry rÛ r/ r6 r6 r7 Ú#test_infer_datetimelike_dtype_mixed s z5TestTypeInference.test_infer_datetimelike_dtype_mixedc C s0 t t jtddg¡}tj|dddks,td S )Nr rW TrÙ rt )rJ r rô r+ r rÜ rt r/ r6 r6 r7 Ú test_infer_dtype_mixed_integer s z0TestTypeInference.test_infer_dtype_mixed_integeréÛ r c C s t j|dddkstd S ©NTrÙ r rÛ r/ r6 r6 r7 Útest_infer_dtype_datetime s z+TestTypeInference.test_infer_dtype_datetimeÚ
|
time_stampc C sN t ||g¡}tj|dddks$tt |||g¡}tj|dddksJtd S r ©rJ r r rÜ rt )r= r r rÖ r6 r6 r7 Ú!test_infer_dtype_datetime_with_na& s z3TestTypeInference.test_infer_dtype_datetime_with_naú1 daysz2 daysc C s t j|dddkstd S ©NTrÙ r rÛ r/ r6 r6 r7 Útest_infer_dtype_timedelta2 s z,TestTypeInference.test_infer_dtype_timedeltaÚdeltac C sN t ||g¡}tj|dddks$tt |||g¡}tj|dddksJtd S r r )r= r r¡ rÖ r6 r6 r7 Ú"test_infer_dtype_timedelta_with_na= s z4TestTypeInference.test_infer_dtype_timedelta_with_nac C sl t tdddtdddg¡}tj|dddks4tt tdddtdddg¡}tj|ddd kshtd S )
|
Nú2011-01r_ ©ra z2011-02TrÙ ÚperiodÚMry ©rJ r r& r rÜ rt r/ r6 r6 r7 Útest_infer_dtype_periodI s z)TestTypeInference.test_infer_dtype_periodÚklassc C s |t dddt dddtjg}tj||ddks6t|t dddt dddtjg}|tjkrddnd }tj||d|ks~td S )
|
Nr r_ r¤ r rÙ r¥ r¦ z unknown-arrayry )r& r! r" r rÜ rt r )r= r© rÚ r> rì r6 r6 r7 Ú test_infer_dtype_period_arrayR s
|
|
ýÿ
|
|
ýÿz/TestTypeInference.test_infer_dtype_period_arrayc C sp t jtdddt d¡gtd}tj|dddks6tt jt d¡tdddgtd}tj|dddksltd S ) Nr£ r¦ r¤ r rH FrÙ ry )rJ r r& r( rn r rÜ rt r/ r6 r6 r7 Ú test_infer_dtype_period_mixedk s ÿ ÿz/TestTypeInference.test_infer_dtype_period_mixedc C s^ t |tdddg¡}tj|dddks,tt |tddd|g¡}tj|dddksZtd S )Nr£ r_ r¤ TrÙ r¥ r§ r r6 r6 r7 Ú test_infer_dtype_period_with_nav s z1TestTypeInference.test_infer_dtype_period_with_nac C s. t t jt jg¡}tj|dddks(tt t jt jd g¡}tj|dddksRttj|dddkshtt d t jt jg¡}tj|dddksttj|dddks¨tt tjg¡}tj|dddksÌtt tjt jg¡}tj|dddksôtt t jtjg¡}tj|dddks tt t jtjt jg¡}tj|dddksLtt d tjd g¡}tj|dddksvtt t d¡g¡}tj|ddd ks tt jtjd fD ]f}t |t d¡|g¡}tj|ddd ksàtt tj|t d¡|g¡}tj|ddd ks®tq®t jt d¡gt
|
d
|
}tj|ddd ksDtt jtjd fD ]f}t |t d¡|g¡}tj|ddd kstt tj|t d¡|g¡}tj|ddd ksRtqRt tjt d¡t d¡t jg¡}tj|dddksôtt jt d¡t d¡gt
|
d
|
}tj|dddks*td S ) NTrÙ r| rr Fry r r r( rH r ) rJ r rô r rÜ rt r! r" r( r) rn )r= rÖ Únr6 r6 r7 Ú!test_infer_dtype_all_nan_nat_like sH " z3TestTypeInference.test_infer_dtype_all_nan_nat_likec C sº t t jtjt d¡g¡}t |¡s(tt |¡s6tt
|
|¡rDtt t jtjt d¡g¡}t |¡rltt |¡rztt
|
|¡stt t jtjt d¡t d¡g¡}t |¡r¸tt |¡rÆtt
|
|¡rÔtt t jtjg¡}t |¡sôtt |¡stt
|
|¡stt jt jt jgt d}t |¡r:tt |¡rJtt
|
|¡rZtt t jtdddtdddgt d¡stt t jtdddtdddgt d¡r¶td S )Nr rH Ú20130101ú
|
US/Eastern©ÚtzZ20130102ZCET)rJ r rô r! r" r( r Úis_datetime_arrayrt Úis_datetime64_arrayÚ!is_timedelta_or_timedelta64_arrayr) rn Z is_datetime_with_singletz_arrayr+ r/ r6 r6 r7 Ú+test_is_datetimelike_array_all_nan_nat_like¸ sH "
|
|
þûÿ
|
|
|
þûÿz=TestTypeInference.test_is_datetimelike_array_all_nan_nat_likeÚfuncr³ r´ Z is_bool_arrayrµ Z is_date_arrayZ is_time_arrayZis_interval_arrayc C sj t t|}t ddg¡}||r$t|| dd¡r8tt ddg¡}||rRt|| dd¡rftd S )Nr Úbarr_ rW )Úgetattrr rJ r rt Zreshape)r= r· rÖ r6 r6 r7 Útest_other_dtypes_for_arrayå s
|
z-TestTypeInference.test_other_dtypes_for_arrayc C s~ dd t ddD }t|}|jdks*tdd t ddD tjg }tj|dd}|d ks`ttj|d
|
d}|dksztd S ) Nc S s g | ]}t d d|qS r ©r ©rb Údayr6 r6 r7 r ü s z/TestTypeInference.test_date.<locals>.<listcomp>rW r r c S s g | ]}t d d|qS r r» r¼ r6 r6 r7 r s FrÙ ry T)r r$ r rt rJ rô r rÜ )r= r r r r6 r6 r7 Ú test_dateû s zTestTypeInference.test_dater> iä z
|
2020-01-01c C s t j||d}|dkstd S )NrÙ r rÛ )r= r> rÚ r r6 r6 r7 Ú%test_infer_dtype_date_order_invariant s z7TestTypeInference.test_infer_dtype_date_order_invariantc C s t t ddg¡¡stt t ddtjg¡¡s4tt t ddg¡¡rLtt t ddg¡¡sdtt t ddg¡¡r|td S )NrW rB r_ )r Zis_float_arrayrJ r rt rô Zis_integer_arrayrE r6 r6 r7 Útest_is_numeric_array s
|
z'TestTypeInference.test_is_numeric_arrayc C s t t ddg¡¡stt jtjddtjgtdddr<tt jtjddtjgtddds`tt jtjddd gtdddstt jtjddtjgtddds¦tt jtjddtj gtdddrÊtt jtjddt
|
d¡gtdddròtt jtjddt dgtdddrtt jtjddd gtdddr>tt jtjddtjgtdddrdtt t d d
|
g¡¡r~td S ) Nr r¸ rH FrÙ Tr" ro rW r_ ) r Zis_string_arrayrJ r rt r! r rn rô r" r( r rE r6 r6 r7 Útest_is_string_array sL ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ ÿ ÿ z&TestTypeInference.test_is_string_arrayc C s< d}|g}t |¡ tdd}|dd}|g}t |¡ d S )N)r é Úrecordzx yr r )r Zto_object_array_tuplesr )r= Úrr> rà r6 r6 r7 Útest_to_object_array_tuplesC s
|
|
|
z-TestTypeInference.test_to_object_array_tuplesc C sH t jd gdd}tj|dd}|dks*ttj|dd}|dksDtd S )Nri rH FrÙ ry Trr r rÝ r6 r6 r7 Ú test_objectN s
|
z TestTypeInference.test_objectc C s¤ dddgdddgg}t j|td}t |¡}t ||¡ t j|td}tj|dd}t ||¡ t jdddd d gdddd d ggtd}tj|dd}t ||¡ d S ) NrW r_ r r2 r r rH )Z min_width)rJ r rn r Zto_object_arrayr rê )r= Úrowsrw r r6 r6 r7 Útest_to_object_array_widthW s
|
ÿz,TestTypeInference.test_to_object_array_widthc C sb t tddd¡stt tdgdd¡r.tt td¡r@tt d¡rNtt tj¡r^td S )Nr£ r¦ r¤ rW )r Z is_periodr& rt r' r+ rJ rô rE r6 r6 r7 Útest_is_periodi s
|
z TestTypeInference.test_is_periodc C s t td}tj|dd}|dks&ttjt|dd}|dksDtt tddgdd}tj|dd}|dksrttjt|dd}|dkstd S )Nr¡ TrÙ Z categoricalZcegfab)Ú
|
categoriesZordered)r rX r rÜ rt r( rÝ r6 r6 r7 Útest_categoricalp s z"TestTypeInference.test_categoricalÚasobjectc C s t jjtddd}|r"| t¡}tj|dd}|dks<ttj|j dd}|dksXttjt
|
||j ddd}|dks|td S )Nr r\ r] FrÙ ÚintervalrH ) r! rk rl r r rn r rÜ rt r8 r( rI )r= rÌ r[ Úinferredr6 r6 r7 Ú test_interval s
|
z TestTypeInference.test_intervalrâ r rC c C sp t ||dd}t ||dd}tj||gtd}tj|dddksDttj||gtd}tj|dddksltd S ) NÚleftr] ÚrightrH FrÙ rÍ ry )r% rJ r rn r rÜ rt )r= râ ÚfirstÚsecondrÖ Zarr2r6 r6 r7 Ú test_interval_mismatched_closed s z1TestTypeInference.test_interval_mismatched_closedc C sà t dddd}t tdtddd}t tdtddd}t ||g¡}tj|dddks^tt ||g¡}tj|dddkstt ||g¡}tj|dddks¦tt dd dd}tj||gtd
|
}tj|ddd ksÜtd S ) Nr rW rÐ r] FrÙ ry g ø?g @rH rÍ ) r% r+ r) rJ r r rÜ rt rn )r= rÒ rÓ ÚthirdrÖ Z flt_intervalr6 r6 r7 Ú test_interval_mismatched_subtype s z2TestTypeInference.test_interval_mismatched_subtyper¶ r[ r r¿ c C s* |||d}t j||d}|dks&td S )NrH rÙ rl rÛ )r= r¶ rÚ r© Znullable_string_dtyper rÎ r6 r6 r7 Útest_string_dtype¯ s z#TestTypeInference.test_string_dtypec C s* ||dd}t j||d}|dks&td S )Nrx rH rÙ rÛ )r= r¶ rÚ r© r rÎ r6 r6 r7 Útest_boolean_dtype¸ s z$TestTypeInference.test_boolean_dtype)DrQ rR rS rn ro rû rc rd rs ru rJ r rô r re rv rw r{ r} r r r r r rn 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º r¾ r r¿ rÀ rÁ rÅ rÆ rÈ rÉ rË rÏ rÔ rÖ r r× rØ r6 r6 r6 r7 rm # s
|
**üþ
|
|
|
|
úþ
|
ÿ ÿ ôþ
|
ýþ
|
ÿ ýþ
|
ÿ
|
9-ùþ
|
|
üþ %
|
|
rm c @ sV e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d d g¡dd Z dd Z dS )ÚTestNumberScalarc C s$ t ds tt dstt ds$tt ds0tt t d¡sBtt t d¡sTtt t d¡sftt tjsttt d rtt drtt tdddr tt t d¡r²tt t drÂtt t ddd rÖtt t
|
d
|
rætt t d rött t d ¡r
|
tt t dd ¡s td S )NTrW r ù ð? @rc r r r° r± éè r Fr_ )r rt rJ r rË rj rô r r( r+ r r) rz r) rE r6 r6 r7 Útest_is_numberà s$ z TestNumberScalar.test_is_numberc C s2 t ds tt dstt t d¡s*tt dr6tt drBtt drNtt t d¡r`tt t d¡rrtt t d¡rtt tjrtt d rtt drªtt tdddr¾tt t d¡rÐtt t
|
dràtt t
|
dd d
|
rôtt t d rtt t dd ¡r tt t d r.td S )NTFrW r rÚ rc r r r° r± rÛ r_ r )r rt rJ rz r rË rj rô r r( r+ r r) r) rE r6 r6 r7 Ú test_is_boolÚ s& z TestNumberScalar.test_is_boolc C s2 t ds tt t d¡s tt dr*tt dr6tt drBtt drNtt t d¡r`tt t d¡rrtt t d¡rtt tjrtt d rtt drªtt tdddr¾tt t d¡rÐtt t
|
dràtt t
|
dd d
|
rôtt t d rtt t d rtt t dd ¡r.td S )NrW Tr rÚ Frc r r r° r± rÛ r r_ )r rt rJ r rz rË rj rô r r( r+ r r) r) rE r6 r6 r7 Útest_is_integerð s& z TestNumberScalar.test_is_integerc C s2 t ds tt t d¡s tt tjs,tt dr8tt drDtt drPtt dr\tt t d¡rntt t d¡rtt t d¡rtt d rtt drªtt tdddr¾tt t d¡rÐtt t
|
dràtt t
|
dd d
|
rôtt t d rtt t dd ¡r tt t d r.td S )Nr TrW rÚ Frc r r r° r± rÛ r_ r )r rt rJ rË rô rz r rj r r( r+ r r) r) rE r6 r6 r7 Ú test_is_float s& z TestNumberScalar.test_is_floatc C sâ t jddd}t jdddd}tds*ttds6tt|sBtt|rNttdrZttdsftt|srtt|s~ttdsttdstt|s¢tt|s®ttdrºttdrÆtt|rÒtt|sÞtd S )Nr¯ r )r` r° )r` r² r( r- )r! Z
|
date_ranger rt r r r )r= r9 Ztsar6 r6 r7 Útest_is_datetime_dtypes s$ z(TestNumberScalar.test_is_datetime_dtypesr² r° ÚUTCc C s@ d| d}t |rtt|s$tt|s0tt|s<td S )Nzdatetime64[ns, ú])r rt r r r )r= r² rI r6 r6 r7 Ú test_is_datetime_dtypes_with_tz4 s
|
z0TestNumberScalar.test_is_datetime_dtypes_with_tzc C s t ds tt dsttdr$ttds0ttddgdd}t |sLtt|sXtt| d¡sjtttg tjdrtttg tjdrtd S )Nr) r, g ļÖBg ļæBrH ) r rt r r* r r$ rJ rË r )r= Ztdir6 r6 r7 Útest_is_timedelta< s z"TestNumberScalar.test_is_timedeltaN) rQ rR rS rÜ rÝ rÞ rß rà rû rc rd rã rä r6 r6 r6 r7 rÙ Â s
|
rÙ c @ s¶ e Zd Zdd Zdd Zdd Zej de
|
d¡e
|
d ¡e
|
e d
|
¡¡e
|
e dd ¡¡e
|
e d ¡¡g¡d d Z ej de
|
g ¡e
|
g g¡g¡dd Zdd Zdd Zdd ZdS )Ú TestIsScalarc C sü t d s tt dstt ds$tt t s2tt ds>tt dsJtt tdsZtt tdsjtt tjsxtt dstt dstt td dds¤tt td dds¸tt t d
|
d sÊtt t
|
dd sÜtt t j sêtt t j søtd S ) NTFrC rW r_ ro Úfoobaró foobariÞ é r ©Úhours)r rt r r r rJ rô r r r r r! r" r rE r6 r6 r7 Ú test_is_scalar_builtin_scalarsL s" z+TestIsScalar.test_is_scalar_builtin_scalarsc C s^ t i r tt g rtt dgr&tt dr2tt dr>tt td rNtt trZtd S )NrW r6 rY )r rt ÚsliceÚEllipsisrE r6 r6 r7 Ú!test_is_scalar_builtin_nonscalars_ s z.TestIsScalar.test_is_scalar_builtin_nonscalarsc C sº t t d¡stt t d¡s$tt t d¡s6tt t d¡sHtt t d¡sZtt t d¡sltt t d¡s~tt t
|
d¡stt t d¡s¢tt t dd¡s¶td S )NrW r r_ ræ rç ú
|
2014-01-01Úh) r rJ r rt rË re ri Zobject_Zstr_Zunicode_Úbytes_r( r) rE r6 r6 r7 Ú"test_is_scalar_numpy_array_scalarsh s z/TestIsScalar.test_is_scalar_numpy_array_scalarsÚzerodimrW ræ rï rð r" c C s" t |r tt t |¡s td S r1 )r rt r Zitem_from_zerodim)r= ró r6 r6 r7 Ú#test_is_scalar_numpy_zerodim_arrayst s z0TestIsScalar.test_is_scalar_numpy_zerodim_arraysrÖ c C s t |r tt t|r td S r1 )r rt r9 r/ r6 r6 r7 Útest_is_scalar_numpy_arrays s z(TestIsScalar.test_is_scalar_numpy_arraysc C sp t tdstt tdds"tt tds2tt tdddsFtt tddsXtt tj d¡sltd S )Nrï rW ré r )rÐ rÑ )Údaysr )
|
r r+ rt r) r& r% r" r! ÚoffsetsZMinuterE r6 r6 r7 Ú test_is_scalar_pandas_scalars s z*TestIsScalar.test_is_scalar_pandas_scalarsc C sÐ t ttdrtt tdgr$tt t r2tt tdggrFtt tg rVtt tdgrhtt tg rxtt tg jrtt t g jrtt tg j
|
d¡r´tt t dddg¡rÌtd S )NrH rW r_ r_ r ) r r( rn rt r! r$ r r# r8 r* Z to_periodr! r rE r6 r6 r7 Ú test_is_scalar_pandas_containers s z-TestIsScalar.test_is_scalar_pandas_containersc C s( G dd dt }|d}t|s$td S )Nc @ s( e Zd ZddddZedddZdS )z3TestIsScalar.test_is_scalar_number.<locals>.NumericNr: c S s
|
|| _ d S r1 ©râ )r= râ r6 r6 r7 r? ¢ s z<TestIsScalar.test_is_scalar_number.<locals>.Numeric.__init__c S s | j S r1 rú rE r6 r6 r7 Ú__int__¥ s z;TestIsScalar.test_is_scalar_number.<locals>.Numeric.__int__)rQ rR rS r? rU rû r6 r6 r6 r7 ÚNumeric¡ s rü rW )r
|
r rt )r= rü Únumr6 r6 r7 Útest_is_scalar_number s z"TestIsScalar.test_is_scalar_numberN)rQ rR rS rë rî rò rû rc rd rJ r r( r) rô rõ rø rù rþ r6 r6 r6 r7 rå K s$ ûþ
|
|
|
rå ÚunitÚmsÚusr; c C s. t tjg d| dd}t|dks*td S )Nz datetime64[râ rH r )r# rJ r rD rt )rÿ r[ r6 r6 r7 Ú.test_datetimeindex_from_empty_datetime64_array¬ s r c C s t tjtdddtdd} tj| jdd
d d
f< | jd }|tj ksPt
|
| d ¡ }tj|d
|
d
< |d
|
tj ks|t
|
d S ) Nr1 rË rH Z20010101)r ÚBr r )r2 r r rP é ) r! rJ rK r r+ rô ZilocÚlocr! r" rt r )Zdfr r& r6 r6 r7 Útest_nan_to_nat_conversions² s ÿ
|
r z!ignore::PendingDeprecationWarningc C s. t | ddggstt t dg¡r*td S )Nr rW )r rt rJ r )Zspmatrixr6 r6 r7 Útest_is_scipy_sparse¿ s r c C sT t jdt jd} t| }|jt jks(tt jdt jd} t| }|jt jksPtd S )Nr1 rH )rJ Zarangere r rI rt r )r> r r6 r6 r7 Útest_ensure_int32Æ s r )rT Ú collectionsr r r r r r~ r Z fractionsr Úior r6 Znumbersr
|
Úrer} Útypingr r r ÚnumpyrJ rû rX Z pandas._libsr r rà r rè Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandas.core.dtypesr Zpandas.core.dtypes.commonr r r r r r r r r r r r r r Zpandasr! r r! r" r# r$ r% r& r' r( r) r* r+ Zpandas._testingZ_testingr Zpandas.core.arraysr, r- r. Zfixturer r8 r9 r] r^ rB rn Zndarrayr rô Z ll_paramsÚzipÚobjsrw r0 rq rx ry r r r r r rc rd r rX rZ r r¢ Ú defaultdictr¦ r\ r¨ r´ r½ r¾ rÁ rÄ rÍ ÚcompilerÐ rÑ rÓ rÔ rÕ rm rÙ rå r r Zskip_if_no_scipyÚfilterwarningsr r r6 r6 r6 r7 Ú<module> s` @8
|
5
|
|
|
|
|
|
Ï3
|
|
|
öþ õþ
|
"
|
÷þ
|
(
|
|
|
|
ÿ
|
|
D $
|
a
|
|
|
